HDU5514 Frogs 容斥原理

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

题意：n只青蛙，m个编号为(0~m-1)的石头，所有青蛙开始站在编号为0的石头上，给定每只青蛙一次可以跳多少步，问最终所有被跳过的石头编号之和

第一道容斥原理，看了别人的博客才懂了点。

每只青蛙只能走到gcd(m,每次跳步)的倍数上，首先用divisor记录m的因子（不含本身），vis[i]表示该因子是否能被经过，num[i]表示该因子被累加的次数，vis[i]-num[i]则表示该因子应该被累加的次数（加多了减去，少加了加上）。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e4+3;
int divisor[100];
int vis[N];
int num[N];
int gcd(int a,int b)
{
    return b==0?a:gcd(b,a%b);
}
int main()
{
    int T,cas=1;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        ll ans=0;
        scanf("%d%d",&n,&m);
        int temp=sqrt(m);
        divisor[0]=1;
        int cur=1;
        for(int i=2;i<=temp;i++)
        {
            if(m%i==0)
            {
                divisor[cur++]=i;
                if(i*i!=m)
                    divisor[cur++]=m/i;
            }
        }
        sort(divisor,divisor+cur);
        memset(vis,0,sizeof(vis));
        memset(num,0,sizeof(num));
        for(int i=0;i<n;i++)
        {
            int t;
            scanf("%d",&t);
            temp=gcd(t,m);
            for(int j=0;j<cur;j++)
            {
                if(divisor[j]%temp==0)
                    vis[j]=1;
            }
        }
        for(int i=0;i<cur;i++)
        {
            if(vis[i]!=num[i])
            {
                ll t=m/divisor[i];
                ans+=divisor[i]*t*(t-1)/2*(vis[i]-num[i]);
                for(int j=i+1;j<cur;j++)
                {
                    if(divisor[j]%divisor[i]==0)//j的因子i被累加，j一定也被累加
                        num[j]+=vis[i]-num[i];
                }
            }
        }
        printf("Case #%d: %lld\n",cas++,ans);
    }
    return 0;
}