/*
找规律 每个ai 对应 li= gcd(M,ai) k*li k>0 为 ai 走到的位置。
ai 能走到的位置 之和 等于一个等差数列,
根据容斥 计算出所有答案 ,
常规容斥 是 2(N) 的效率,
ANS = a1 +a2+ a3 -a1a2 -a2a3-....+a1a2a3;
每一项都是 m 的因子,那直接计算 每个因子对应的常数
a1 1 a2 1 a3 1 a1a2 -1 a2a3 -1...
a1a2 = (1) - 1+1
a1a2a3 = (1) -1+1+1-1-1-1
可以这样构造
*/
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
using namespace std;
#define For(i,a) for(int i=0;i<a;++ i)
#define ms(str) memset(str,0,sizeof(str))
typedef long long ll;
const int M= 1e4+1;
int g,vis[M],fac[M],num[M];
int main()
{
int T,n,m,cas=0;
scanf("%d",&T);
while(T--)
{
int cnt=0;
scanf("%d%d",&n,&m);
for(int i=2;i<sqrt(m+1);++ i)
{
if(i*i==m) fac[++cnt]=i;
else if(m%i==0) fac[++cnt] =i,fac[++cnt] =m/i;
}
ms(num);ms(vis);
sort(fac+1,fac+1+cnt);
int flag=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&g);
g =__gcd(g,m);
if(g==1) flag=1;
for(int i=1;i<=cnt;++ i)
if(fac[i]%g==0)
{
vis[i]=1;
}
}
printf ("Case #%d: ",++cas);
ll ans=0;
if(flag)
{
ans =(ll)m*(m-1)/2;
printf("%I64d\n",ans);
continue;
}
for(int i=1;i<=cnt;++ i)
{
if(vis[i]!=num[i])
{
int k =m/fac[i];
ans +=(ll)k*(k-1)/2*fac[i]*(vis[i]-num[i]);
for(int j=i+1;j<=cnt; ++ j)
num[j] +=(fac[j]%fac[i])?0:vis[i]-num[i];
}
}
printf ("%I64d\n",ans);
}
return 0;
}
HDU 5514 Frogs 容斥
最新推荐文章于 2021-02-01 19:10:04 发布