#include #include #include #include #include #include #include constexpr size_t length = 1048576; constexpr size_t sample_length = 1000; #define CALC_HEIGHT #define CALC_CDF_DIFF #ifdef CALC_HEIGHT constexpr int block_size = 2; unsigned trailing_zeroes(size_t index) { unsigned bits = 0; unsigned x = index / block_size; if (x) { while (x % block_size == 0) { ++bits; x /= block_size; } } return bits; } unsigned int getHeight(long loc) { if (loc % block_size == 0) { auto level = trailing_zeroes(loc) + 1; return level - 1; } return 0; } #else unsigned int getHeight(long loc) { return loc; } #endif std::vector population_vector, sample_vector; void initialize(const std::vector &input_population, const std::vector &input_sample) { population_vector = input_population; std::sort(population_vector.begin(), population_vector.end()); sample_vector = input_sample; std::sort(sample_vector.begin(), sample_vector.end()); } long double sample_cdf(long double x) { auto it = lower_bound(sample_vector.begin(), sample_vector.end(), x); if (it == sample_vector.end()) { return 1; } if (it == sample_vector.begin()) { return 0; } auto it_prev = it - 1; return ((long double)(it_prev - sample_vector.begin()) + (long double)(x - *it_prev) / (*it - *it_prev)) / (long double)(sample_vector.size() - 1); } int infer_offset(long double x) { return int(sample_cdf(x) * double(population_vector.size())); } #ifdef CALC_CDF_DIFF int infer_offset_from_cdf(long double cdf) { return int(cdf * double(population_vector.size())); } // const double M_SQRT1_2 = sqrt(0.5); constexpr long long s_avg = 2147483648; constexpr long long s_stddev = 2147483648; long double goal_cdf(long double x) { return std::erfc(-x * M_SQRT1_2) * 0.5; } long double stat_cdf(long double x) { return goal_cdf((x - s_avg) / s_stddev); } long double finite_population_cdf(long double x) { auto it = upper_bound(population_vector.begin(), population_vector.end(), x); return (long double)(it - population_vector.begin()) / population_vector.size(); } #endif int main() { std::vector original_vector; std::set set1, set2; typedef std::tuple element_type; std::vector location; FILE *file = fopen("normal_distribution.txt", "r"); for (long long i; fscanf(file, "%lld ", &i) != EOF; original_vector.push_back(i), set1.insert(i)) ; fclose(file); assert(original_vector.size() == set1.size()); assert(original_vector.size() >= length); auto it = original_vector.begin(); for (int i = 1; i <= sample_length; i++) { auto element = *it; auto ret = set2.insert(element); if (ret.second) { sample_vector.push_back(element); // printf("%lld\n", element); } it++; } assert(sample_vector.size() == sample_length); initialize(original_vector, sample_vector); for (auto iter = population_vector.begin(); iter != population_vector.end(); iter++) { #ifdef CALC_CDF_DIFF auto cdf = sample_cdf(*iter); printf("%d,%d,%.6Lf ", getHeight(infer_offset_from_cdf(cdf)), getHeight(iter - population_vector.begin()), std::abs(cdf - stat_cdf(*iter))); #else printf("%d,%d ", getHeight(infer_offset(*iter)), getHeight(iter - population_vector.begin())); #endif fflush(stdout); } puts(""); return 0; }