Repository navigation
Expand file tree
/
Copy pathFFT_AND.cpp
More file actions
56 lines (51 loc) · 1.45 KB
/
Copy pathFFT_AND.cpp
File metadata and controls
56 lines (51 loc) · 1.45 KB
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
51
52
53
54
55
56
/*
* CS Academy: And Closure
* given an array of N integers. choose any subset of numbers and compute
* their binary and. find the number of distinct results you can get.
*/
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const ll MOD = 5767169;
ll bigmod(ll b, ll p) {
ll ret = 1LL;
while( p ) {
if( p & 1 )
ret = (ret * b) % MOD;
b = (b * b) % MOD;
p >>= 1;
}
return ret;
}
void andfft(vector<ll> &P, bool inv) {
int n = (int)P.size();
for(int l = 1; (l << 1) <= n; l <<= 1) {
for(int i = 0; i < n; i += (l << 1)) {
for(int j = 0; j < l; j += 1) {
ll u = P[ i + j ], v = P[ i + l + j ];
if( !inv )
P[ i + j ] = v, P[ i + l + j ] = (u + v) % MOD;
// P[ i + j ] = (u + v) % MOD, P[ i + l + j ] = u; // for orfft
else
P[ i + j ] = (-u + v) % MOD, P[ i + l + j ] = u;
// P[ i + j ] = v, P[ i + l + j ] = (u - v) % MOD; // for orfft
}
}
}
}
int main() {
int n; cin >> n;
vector<ll> A(1 << 20, 0);
for(int i = 1; i <= n; i += 1) {
int x; cin >> x; A[ x ] = 1;
}
andfft(A, 0);
for(auto &x : A)
x = ( bigmod(x, n) + MOD ) % MOD;
andfft(A, 1);
int ans = 1;
for(int i = 1; i < (1 << 20); i += 1)
if( A[ i ] ) ans += 1;
cout << ans << '\n';
return 0;
}