-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCountingNecklaces.cpp
More file actions
59 lines (46 loc) · 1020 Bytes
/
CountingNecklaces.cpp
File metadata and controls
59 lines (46 loc) · 1020 Bytes
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
57
58
59
//Counting Necklaces - https://cses.fi/problemset/task/2209
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MOD = 1e9 + 7;
const ll INF = 1e18;
ll expo(ll a, ll b, ll m=MOD) {
ll res = 1;
if (b == 0) return res;
while (b > 0) {
if (b & 1) {
res *= a;
res %= m;
}
a *= a;
a %= m;
b >>= 1;
}
return res;
}
ll inverse(ll a, ll mod=MOD) {
return expo(a , mod-2 , mod);
}
//burnside's lemma
void solve() {
ll n, m;
cin >> n >> m;
ll group_size = n;
ll sum_of_fixed_pts = 0;
//counting rotations
for (int rotation = 0; rotation < n; rotation++) {
sum_of_fixed_pts += expo(m, gcd(rotation, n));
sum_of_fixed_pts %= MOD;
}
cout << (sum_of_fixed_pts * inverse(group_size)) % MOD << endl;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
int T = 1;
// cin >> T;
while (T--) {
solve();
}
return 0;
}