test/math/binomial3.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

#include "../util.h"
#include <math/extendedEuclid.cpp>
#include <math/multInv.cpp>
#include <math/binomial3.cpp>


constexpr ll mod = 503;

void stress_test() {
	vector<ll> last = {1};
	ll queries = 0;
	for (ll i = 0; i < mod; i++) {
		for (ll j = 0; j <= i; j++) {
			ll got = calc_binom(i, j, mod);
			ll expected = last[j];
			if (got != expected) cerr << "calc_binom(" << i << ", " << j << "), got: " << got << ", expected: " << expected << FAIL;
		}
		queries += ssize(last);

		last.push_back(1);
		for (ll j = i; j > 0; j--) {
			last[j] = (last[j] + last[j - 1]) % mod;
		}
	}
	cerr << "tested queries: " << queries << endl;
}

int main() {
	stress_test();
}