【bzoj4197】[Noi2015]寿司晚宴 dp

因为每个数只有一个大于根号n的质因子，所以我们把每个数拆成一个大于根号n的质因子乘以一个数的形式，对于大于根号n的质因子相同的数，我们放到一起处理
dp[0/1][i][x][y]表示A/B选了当前的大质数，现在枚举到具有当前大质数的第i个数，之前A选中的集合为x，B选中的集合为y的方案数
dp[0/1][0][x][y]=f[i-1][x][y]
dp[0][i][x][y]=dp[0][i-1][x][y]+dp[0][i-1][x-S][y]
dp[1][i][x][y]=dp[1][i-1][x][y]+dp[1][i-1][x][y-S]
f[i][x][y]表示考虑了前i个大质数，A选中的集合为x，B选中的集合为y的方案数
f[i][x][y]=dp[0][num[i]][x][y]+dp[1][num[i]][x][y]-f[i-1][x][y]
由于不选当前大质数的方案被计算了2次，所以要-1
空间可用01背包的方法优化掉一维，但要注意枚举顺序

注：这道题指数只需要存到19，因为23*29>500

能想出这道题的人都好厉害呀

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>

using namespace std;

struct yts
{
	int p,a;
}q[510];

bool cmp(yts x,yts y)
{
	return x.p<y.p;
}

int pri[10]={2,3,5,7,11,13,17,19};
int dp[2][1<<8][1<<8],f[1<<8][1<<8];
int n,m,mod,tot;

void calc(int x)
{
	tot++;
	for (int i=0;i<=7;i++)
	  if (x%pri[i]==0)
	  {
	  	while (x%pri[i]==0) x/=pri[i];
	  	q[tot].a|=(1<<i);
	  }
	q[tot].p=x;
}

int main()
{
	scanf("%d%d",&n,&mod);
	for (int i=2;i<=n;i++) calc(i);
	sort(q+1,q+tot+1,cmp);
	f[0][0]=1;
	for (int i=1;i<=tot;i++)
	{
		if (q[i].p==1 || q[i].p!=q[i-1].p)
		  for (int x=0;x<(1<<8);x++)
		    for (int y=0;y<(1<<8);y++)
		      dp[0][x][y]=dp[1][x][y]=f[x][y];
		for (int x=(1<<8)-1;x>=0;x--)
		  for (int y=(1<<8)-1;y>=0;y--)
		  {
		  	if ((y&q[i].a)==0) dp[0][x|q[i].a][y]=(dp[0][x|q[i].a][y]+dp[0][x][y])%mod;
		  	if ((x&q[i].a)==0) dp[1][x][y|q[i].a]=(dp[1][x][y|q[i].a]+dp[1][x][y])%mod;
		  }
		if (q[i].p==1 || i==n || q[i].p!=q[i+1].p)
		  for (int x=0;x<(1<<8);x++)
		    for (int y=0;y<(1<<8);y++)
		      f[x][y]=((long long)dp[0][x][y]+dp[1][x][y]-f[x][y]+mod)%mod;
	}
	int ans=0;
	for (int i=0;i<(1<<8);i++)
	  for (int j=0;j<(1<<8);j++)
	    if ((i&j)==0) ans=(ans+f[i][j])%mod;
	printf("%d\n",ans);
	return 0;
}