HDU 5514 Frogs

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5514

题意：有m个石头围成一圈，编号分别为0到m-1，现在有n只青蛙，都在0号石头上，第i只青蛙会从当前编号为pos的石头跳到编号为（pos+ai）%m的石头上。
但凡被青蛙经过的石头都会被占领，求这m块石头中所有被占领过的石头的编号和。

 思路：容斥定理，gcd(ai,m)是第i只青蛙可以跳到的石头的最小编号（0除外），那么(m-1)/gcd(ai,m)就是这只青蛙可以跳到的石头数量。因为所有的gcd
(ai,m)都是m的因数，所以先把m分解因数，然后再用容斥去统计答案。如果出现了一个x = gcd(ai,m)，那么所有x的倍数的因数都算是出现过了，在计算的时
候如果把x的倍数都加了k遍，那么就要把pk的倍数减去k遍。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>

using namespace std;

#define rep(i,j,k) for(int i = j;i<=k;i++)
#define Rep(i,j,k) for(int i = j;i>=k;i--
#define Clean(x,y) memset(x,y,sizeof(x))
#define LL long long

int T,n,m;
int fact[1009];
int sum;
int appear[1009];
int cnt[1009];

void init(int x)
{
    Clean(appear,0);
    Clean(cnt,0);
	sum = 0;
	for(int i = 1;i*i<=x; i++)
	{
		if ( x % i == 0 )
		{
			fact[++sum] = i;
			if ( i * i != x ) fact[++sum] = x/i;
		}
	}
}


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

int main()
{

	cin>>T;
	rep(kase,1,T)
	{
		LL ans = 0;
		scanf("%d %d",&n,&m);
		init(m);
		sort(fact+1,fact+1+sum);
		sum--; //去掉因数m
		int a;
		rep(i,1,n)
		{
			scanf("%d",&a);
			int x = gcd(m,a);
            rep(i,1,sum)
                if ( fact[i] % x == 0 ) appear[i] = 1; //因数fact[i]贡献的值也在答案之中。
		}

        rep(i,1,sum)
            if ( cnt[i] != appear[i] ) //算多了减，算少了加
            {
                LL t = (m-1)/fact[i];  // fact[i](1+2+3+...+ (m-1)/fact[i] )
                ans += t*(t+1) / 2 * fact[i] * ( appear[i] - cnt[i] ); 
                rep(j,i+1,sum)
                    if ( fact[j] % fact[i] == 0 ) cnt[j] += appear[i] - cnt[i]; //所有为fact[i]的倍数的因数在计算fact[i]的答案时均被计算了

一遍。
            }
        printf("Case #%d: %I64d\n",kase,ans);
	}

	return 0;
}