HDU 5514 Frogs 容斥

/*
   找规律 每个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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值