Frogs HDU - 5514 (容斥+gcd）（注意容斥操作）

Frogs

HDU - 5514

;

题意

有一堆青蛙，一开始都在0点，然后有一堆圈成一圈的石子，石子的编号是从0-m-1的

然后青蛙只能顺时针跳，每个青蛙可以一次跳a[i]格，然后所有青蛙都这样一直跳下去

然后问你，这些青蛙踩过的石子的编号和是多少？

思路

规律发现，对于第i只青蛙，他跳过的格子，一定是m中gcd(a[i],m)的倍数，

后来我们觉得是容斥寻找每个数的倍数，然后发现根本不可写；

后来：

网上的思路：

我们就分解m的因数之后，因为青蛙可以去的点是gcd（a[i],m）,所以他可以去的点一定包含m的因子，然后对于m的因子进行处理;

容斥进行：

这个容斥的方法是：对于每个元素，可以求出他的目标值（也就是最终应该贡献数，比如这个题目应该是1，只加一次），将这个数存在vis中

后来在进行答案求和过程中：

  初始化一个num【i】数组用来存，(表示到现在为止的操作中，i已经被加存了多少次），如果多了那就减去，如果少了那就加上；（vis[i] - num[i]），

num【i】数组的更新通过他前面与他有关系的i进行更新；

具体实现过程看代码

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

using namespace std;

typedef long long LL;

const int maxn = 1e4+10;
int vis[maxn];//应该加个数
int num[maxn];//真实加的个数
int a[maxn];//因子

int gcd(int a,int b)
{
    if(b==0)
      return a;
    else
        return gcd(b,a%b);
}
int main()
{
    int t;
    int n, m;
    scanf("%d",&t);
    for(int ka=1;ka<=t;ka++)
    {
        scanf("%d%d",&n,&m);
        int cnt=0;//质因子个数
        memset(vis,0,sizeof(vis));
        memset(num,0,sizeof(num));
           //寻找m的因子
        for(int i=1;i*i<=m;i++)
        {
            if(m%i==0)
             {
                 a[cnt++]=i;
                 if(m/i!=i)
                    a[cnt++]=m/i;
             }
        }
        sort(a,a+cnt);//一定别忘了
       //确定每个因子的应该出现数
       for(int i=0;i<n;i++)
       {
           int x;
           scanf("%d",&x);
           int y=gcd(x,m);
           for(int j=0;j<cnt;j++)
           {
               if(a[j]%y==0)
                  vis[j]=1;
           }
       }
       vis[cnt-1]=0;//m是不存在的 0，根本不需要加
       long long ans=0;
       //vis[i]-num[i]是每个数应再次加出现次数，
       //如果相等说明这个数正好加上vis[i];
       for(int i=0;i<cnt;i++)
       {
           if(vis[i]!=num[i])//如果多加，或者多减
           {
               long long x=m/a[i];
               //计算这个m中有多少个a[i],，这a[i]的倍数全部加和，
               ans+=((x+1)*x/2*a[i]*(vis[i]-num[i]));
               //更新后面的数的现在已加数目
               for(int j=i+1;j<cnt;j++)
               {
                   if(a[j]%a[i]==0)
                    num[j]+=vis[i]-num[i];
               }
           }
       }
       printf("Case #%d: %lld\n",ka,ans);
    }


    return 0;
}