[poj3904]Sky Code_状态压缩_容斥原理

Posted JZYshuraK_彧

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[poj3904]Sky Code_状态压缩_容斥原理相关的知识,希望对你有一定的参考价值。

Sky Code poj-3904

    题目大意:给你n个数,问能选出多少满足题意的组数。

    注释:如果一个组数满足题意当且仅当这个组中有且只有4个数,且这4个数的最大公约数是1,$1\le n\le 10^4$。

      想法:我们显然可以知道4个数是可以不用两两互质的,所以正面计算难度较大,我们考虑从反面考虑。我们通过计算所有gcd不为1的组数,用总组数相减即可。然后,我们发现一个不为0的gcd显然可以被组中的任意一个数整除,所以我们可以进行容斥。只需要枚举gcd的约数个即可。计算的过程我们用状态压缩实现。

    最后,附上丑陋的代码... ...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 10005 
using namespace std;
typedef long long ll;
ll a[10010];//记录单个数的质因数
ll cnt;//记录单个数的质因数个数
ll ans[10010][2];//ans[i][0]表示包含i这个因子的数的个数,ans[i][1]表示i的质因子个数
ll Calc(ll x)//计算C[n][4]
{
	return x*(x-1)*(x-2)*(x-3)/24;
}
void separate(ll x)//分解质因数,由于我们在后面需要用cnt进行状态压缩,所以a数组从0开始记录
{
	for(int i=2;i*i<=x;i++)
	{
		if(x%i==0)
		{
			a[cnt]=i;
			cnt++;
			while(x%i==0)
			{
				x/=i;
			}
		}
	}
	if(x>1) a[cnt++]=x;
}
void dispose(ll x)
{
	cnt=0;
	separate(x);
	for(int i=1;i<(1<<cnt);i++)//通过枚举当前全集来统计桶
	{
		ll flag=0,middle=1;
		for(int j=0;j<cnt;j++)
		{
			if(i&(1<<j))
			{
				flag++;
				middle*=a[j];
			}
		}
		ans[middle][0]++;
		ans[middle][1]=flag;
	}
}
int main()
{
	ll n;
	while(~scanf("%lld",&n))
	{
		memset(ans,0,sizeof ans);
		ll x;
		for(int i=1;i<=n;i++)
		{
			scanf("%lld",&x);
			dispose(x);
		}
		ll answer=Calc(n);
		for(int i=2;i<=maxn/4;i++)
		{
			if(ans[i][0])//Important
			{
				if(1&ans[i][1]) answer-=Calc(ans[i][0]);//如果是偶数个质因子
				else answer+=Calc(ans[i][0]);//如果是奇数个质因子
			}
		}
		// puts("Fuck");
		printf("%lld\n",answer);//输出答案即可
	}
	return 0;
}

     小结:如果一个问题极其复杂,我们不妨反其道而行之。容斥原理就是一例。

以上是关于[poj3904]Sky Code_状态压缩_容斥原理的主要内容,如果未能解决你的问题,请参考以下文章

POJ3904 Sky Code容斥原理

poj3904-Sky Code-一起揭露莫比乌斯反演和容斥的一致性

[POJ3904]Sky Code

POJ 3904 Sky Code 莫比乌斯反演入门

[bzoj2287][poj Challenge]消失之物_背包dp_容斥原理

[ An Ac a Day ^_^ ] POJ 3254 Corn Fields 状压dp