POJ 2409 Pólya 计数定理

题意

传送门 POJ 2409

题解

要求将手镯旋转、翻转看做相同的状态，考虑 $P\acute{o}lya$ 计数定理。设颜色有 $m$ 种。对于旋转，有旋转 $[0,n)$ 个 $2\pi /n$ 的角度的置换，即 $n$ 个置换群。考虑旋转 $k$ 个位置后和原来相同的染色方案个数，将颜色与第 $i$ 个石头相同的石头的集合叫做 $i$ 的轨迹，个数为 $n/gcd(k,n)$。因此，旋转 $k$ 个位置后和原来相同的染色方案数就是 $m^{gcd(k,n)}$。

翻转同理，若 $n$ 为奇数，对称轴必经过某一个点，共 $n$ 种置换；若 $n$ 为偶数，对称轴经过两点或两点连线的中点，各 $n/2$ 种置换。同时考虑旋转与翻转，共 $2n$ 个置换群。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
using namespace std;

int gcd(int a, int b)
{
    return b == 0 ? a : gcd(b, a % b);
}

int _pow(int x, int n)
{
    int res = 1;
    while (n)
    {
        if (n & 1)
            res *= x;
        x *= x;
        n >>= 1;
    }
    return res;
}

int main()
{
    int n, m;
    while (~scanf("%d%d", &m, &n) && (n | m))
    {
        int res = 0;
        for (int i = 0; i < n; ++i)
            res += _pow(m, gcd(i, n));
        if (n & 1)
            res += n * _pow(m, (n + 1) / 2);
        else
            res += n / 2 * (_pow(m, (n + 2) / 2) + _pow(m, n / 2));
        res /= 2 * n;
        printf("%d\n", res);
    }
    return 0;
}