summaryrefslogtreecommitdiff
path: root/content/math/linearSieve.cpp
blob: 64440dd2d2bb144268114872fa78ac066771323e (plain)
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
constexpr ll N = 10'000'000;
ll small[N], power[N], sieved[N];
vector<ll> primes;

//wird aufgerufen mit (p^k, p, k) für prime p und k > 0
ll mu(ll pk, ll p, ll k) {return -(k == 1);}
ll phi(ll pk, ll p, ll k) {return pk - pk / p;}
ll div(ll pk, ll p, ll k) {return k+1;}
ll divSum(ll pk, ll p, ll k) {return (pk*p-1) / (p - 1);}
ll square(ll pk, ll p, ll k) {return k % 2 ? pk / p : pk;}
ll squareFree(ll pk, ll p, ll k) {return p;}

void sieve() { // O(N)
	small[1] = power[1] = sieved[1] = 1;
	for (ll i = 2; i < N; i++) {
		if (small[i] == 0) {
			primes.push_back(i);
			for (ll pk = i, k = 1; pk < N; pk *= i, k++) {
				small[pk] = i;
				power[pk] = pk;
				sieved[pk] = mu(pk, i, k); // Aufruf ändern!
		}}
		for (ll j=0; i*primes[j] < N && primes[j] < small[i]; j++) {
			ll k = i * primes[j];
			small[k] = power[k] = primes[j];
			sieved[k] = sieved[i] * sieved[primes[j]];
		}
		if (i * small[i] < N && power[i] != i) {
			ll k = i * small[i];
			small[k] = small[i];
			power[k] = power[i] * small[i];
			sieved[k] = sieved[power[k]] * sieved[k / power[k]];
}}}

ll naive(ll n) { // O(sqrt(n))
	ll res = 1;
	for (ll p = 2; p * p <= n; p++) {
		if (n % p == 0) {
			ll pk = 1;
			ll k = 0;
			do {
				n /= p;
				pk *= p;
				k++;
			} while (n % p == 0);
			res *= mu(pk, p, k); // Aufruf ändern!
	}}
	if (n > 1) res *= mu(n, n, 1);
	return res;
}