diff --git a/histogram.cpp b/histogram.cpp new file mode 100644 index 0000000..4d33c05 --- /dev/null +++ b/histogram.cpp @@ -0,0 +1,50 @@ +#include "histogram.h" + +using namespace std; + +static void +find_minmax(const vector& numbers, double& min, double& max) +{ + min = numbers[0]; + max = numbers[0]; + for (double x : numbers) + { + if (x > max ) + { + max = x; + } + if (x < min) + { + min = x; + } + } +} + +vector +make_histogram(const vector& numbers, size_t& bin_count) +{ + double min = 0; + double max = 0; + find_minmax(numbers, min, max); + auto bin_size = (max - min)/bin_count; + vector bins(bin_count); + for (double number : numbers) + { + bool found = false; + for (size_t j = 0; (j < bin_count - 1) && !found; j++) + { + auto lo = min + j * bin_size; + auto hi = min + (j + 1) * bin_size; + if ((lo <= number) && (number < hi)) + { + bins[j]++; + found = true; + } + } + if (!found) + { + bins[bin_count - 1]++; + } + } + return bins; +} diff --git a/histogram.h b/histogram.h new file mode 100644 index 0000000..cb3c148 --- /dev/null +++ b/histogram.h @@ -0,0 +1,9 @@ +#ifndef HISTOGRAM_H_INCLUDED +#define HISTOGRAM_H_INCLUDED + +#include + +std::vector +make_histogram(const std::vector& numbers, size_t& bin_count); + +#endif // HISTOGRAM_H_INCLUDED