Pólya 计数板子

Description

一个手链上有 $n$ 个珠子，有 $n$ 种颜色，给每个珠子都染上颜色。有多少种不同的染色方法。由于方案数很多，请对 $998244353$ 取模。

Input

一个非负整数 $n(0\le n\le 10^9)$

Output

一个整数，表示染色方案数对 $998244353$ 取模的结果

Sample Input

2

Sample Output

3

ploya计数的裸题，详细题解见洛谷P4980

代码：

#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long
const int N = 1e4+5;
const ll mod = 998244353;

ll qpow(ll a, ll b, ll c){
    a %= c;
    ll ans = 1;
    while(b){
        if(b & 1)
            ans = (ans * a) % c;
        a = (a * a) % c;
        b >>= 1;
    }
    return ans;
}

ll phi(ll n){
    ll ans = n;
    for(int i = 2; i*i <= n; ++i){
        if(n % i == 0){
            ans = ans / i * (i-1);
            while(n % i == 0)
                n /= i;
        }
    }
    if(n != 1)
        ans = ans / n * (n-1);
    return ans;
}

void solve(ll n){
    if(n == 1)
        cout << 1 << '\n';
    else
    {
        ll ans = 0;
        for(int i = 1; i * i <= n; ++i){
            if(n % i == 0){
                if(i * i == n)
                    ans = (ans + (phi(n/i) * qpow(n, i, mod) % mod)) % mod;
                else{
                    ans = (ans + (phi(i) * qpow(n, n/i, mod) % mod)) % mod;
                    ans = (ans + (phi(n/i) * qpow(n, i, mod) % mod)) % mod;
                }
            }
        }
        cout << (ans * qpow(n, mod-2, mod)) % mod << '\n';
    }
}

int main()
{
	std::ios::sync_with_stdio(false);
    ll n;
    cin >> n;
    solve(n);
	return 0;
}