Polya计数原理小记

Polya计数原理小记

Polya计数原理：

证明：emmm我不会
1)只有旋转（平面内旋转最小单位角度 360° / n, 一共n种旋转方案，第i种方案的旋转角度 i * 360° / n, 其循环节为 gcd(n,i) ）

使用欧拉函数的原因:
对于大多数i,gcd（n，i）值相同都是n的约数，当n比较大的时候转而用sqrt(n)的复杂度枚举n的约数。
For(d | n){
temp += pow(m,d);
phid = getphi(d);
ans += phid * temp;
}

2)考虑翻转，旋转的情况：
旋转：
N个点顺时针转动360/n 个角度，第i个旋转置换的循环数gcd(n,i);旋转总方案：sigm m^gcd(n,i) / n（i=1，2，…n）
翻转：
注意一下 / 号 代表向下取整的意思
N为偶数(分两种情况讨论)：
对称轴不过顶点：循环节：n / 2, 方案数：m ^ (n / 2)
对称轴过顶点：循环节：(n / 2 +1)， 方案数 ： m ^ (n / 2 + 1)
N为奇数：循环节：( (n + 1) / 2 )， 方案数 : m ^ ((n+1) /2)

模板：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn = 36000;
bool isprime[maxn];
int prime[maxn],phi[maxn],countp;
void init(){
    memset(isprime,1,sizeof(isprime));
    isprime[0] = isprime[1] = false;
    countp = 0;
    phi[1] = 1;
    for(int i = 2; i < maxn; ++i){
        if(isprime[i]){
            prime[++countp] = i;
            phi[i] = i - 1;
        }
        for(int j = 1; j <= countp && i * prime[j] < maxn; ++j){
            isprime[i*prime[j]] = 0;
            if(!(i%prime[j])){
                phi[i*prime[j]] = phi[i] * prime[j];
                break ;
            }
            phi[i*prime[j]] = phi[i] * (prime[j] - 1);
        }
    }
}

int fast_pow(int a,int b,int mod){
    int res = 1;
    while(b > 0){
        if(b&1) res = (ll)res * a % mod;
        a = (ll)a * a % mod;
        b >>= 1;
    }
    return res;
}

int exgcd(int a,int b,int &x,int &y){
    if(!b){
        return x = 1, y = 0, a;
    }
    int d = exgcd(b, a % b, x, y);
    int z = x; x = y, y = z - (a / b) * y;
    return d;
}

int get_inv(int a,int mod){
    int x,y;
    exgcd(a, mod, x, y);
    /*
    int d =  exgcd(a, mod, x, y);
    if(d > 1) return -1;
    */
    return (x % mod +  mod) % mod;
}

int getphi(int n){
    if(n < maxn) return phi[n];
    int ans = n;
    for(int i = 1; prime[i] * prime[i] <= n; ++i){
        if(!(n % prime[i])){
            ans = ans / prime[i] * (prime[i] - 1);
            while(!(n % prime[i])) n /= prime[i];
        }
    }
    if(n > 1) ans = ans / n * (n - 1);
    return ans;
}

int main(){
    init();
    int n,m,p;
    while(~scanf("%d%d%d",&n,&m,&p)){
        int ans = 0;
        for(int i = 1; (ll)i * i <= n; ++i){
            if(i * i == n){
                ans = (ans + (ll)getphi(i) * fast_pow(m, i, p) % p) % p;
                continue ;
            }
            else if(!(n % i)){
                ans = (ans + (ll)getphi(n / i) * fast_pow(m, i, p) % p) % p;
                ans = (ans + (ll)getphi(i) * fast_pow(m, n / i, p) % p) % p;
            }
        }
        int inv = get_inv(n,p); //excgd, p is not a prime;
        ans = (ll)ans*inv;
        /* case 2:
        int inv2 = get_inv(2,p);
        ans = (ans + (ll)inv2 * (fast_pow(m, n / 2 + 1, p) + fast_pow(m, (n + 1) / 2, p)) % p) % p;
        ans = (ll)ans * inv2 % p;
        */
        printf("%d\n",ans);
    }
}

题目：POJ 2154
题意：n个点，n中颜色，不考虑翻转方案数模P输出，P不是素数且不能保证(n,p)互素。1 <= n, p <= 1e9;
思路：gcd(n,p)>=1, 即n 模 p 的逆元可能不存在，但注意到公式m = n，m ^ i = n ^ i，可以统一消去个n次方。
注意：不要开太多long long运算，会TLE，可以int在中间过程强制转换类型

AC code:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn = 36000;
bool isprime[maxn];
int prime[maxn],phi[maxn],countp;
void init(){
    memset(isprime,1,sizeof(isprime));
    isprime[0] = isprime[1] = false;
    countp = 0;
    phi[1] = 1;
    for(int i = 2; i < maxn; ++i){
        if(isprime[i]){
            prime[++countp] = i;
            phi[i] = i - 1;
        }
        for(int j = 1; j <= countp && i * prime[j] < maxn; ++j){
            isprime[i*prime[j]] = 0;
            if(!(i%prime[j])){
                phi[i*prime[j]] = phi[i] * prime[j];
                break ;
            }
            phi[i*prime[j]] = phi[i] * (prime[j] - 1);
        }
    }
}

int fast_pow(int a,int b,int mod){
    int res = 1;
    while(b > 0){
        if(b&1) res = (ll)res * a % mod;
        a = (ll)a * a % mod;
        b >>= 1;
    }
    return res;
}

int exgcd(int a,int b,int &x,int &y){
    if(!b){
        return x = 1, y = 0, a;
    }
    int d = exgcd(b, a % b, x, y);
    int z = x; x = y, y = z - (a / b) * y;
    return d;
}

int get_inv(int a,int mod){
    int x,y;
    exgcd(a, mod, x, y);
    /*
    int d =  exgcd(a, mod, x, y);
    if(d > 1) return -1;
    */
    return (x % mod +  mod) % mod;
}

int getphi(int n){
    if(n < maxn) return phi[n];
    int ans = n;
    for(int i = 1; prime[i] * prime[i] <= n; ++i){
        if(!(n % prime[i])){
            ans = ans / prime[i] * (prime[i] - 1);
            while(!(n % prime[i])) n /= prime[i];
        }
    }
    if(n > 1) ans = ans / n * (n - 1);
    return ans;
}


int main(){
    init();
    int T,n,m,p;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&p);
        int ans = 0;
        for(int i = 1; i * i <= n; ++i){
            if(i * i == n){
                ans = (ans + (ll)getphi(i) * fast_pow(n, i - 1, p) % p) % p;
                continue ;
            }
            else if(!(n % i)){
                ans = (ans + (ll)getphi(n / i) * fast_pow(n, i - 1, p) % p) % p;
                ans = (ans + (ll)getphi(i) * fast_pow(n, (n / i) - 1, p) % p) % p;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}