4197: [Noi2015]寿司晚宴 思路题 状压DP

毕竟是NOI的T3，自我感觉思维量还是挺大的。。虽然代码量并不大。。

---

我们从数据范围入手，因为我们知道一个数n的大于sqrt(n)的因数只可能含有1个或0个，而本题n<=500,所以小于sqrt(500)的质因子只可能有8个。由于我们要求的是互质，那么我们选择了一个数就代表着选择了这个数含有的质因子。我们可以根据这个进行分类。s[i].first代表>sqrt(n)的因子，如果没有默认是1，s[i].second是一个[0,1<<8)的数，二进制表示是否含有这8个< sqrt(500)的素因子。我们按照s[i].first分类（注意如果s[i].first==1则单独为一类），同一类之间的不能同时选。然后我们就可以进行状压DP了。
f[i][j]表示第一个人选的情况为i第二个人选的情况j，(i,j∈[0,1<<8))时的方案数。
g[0][j][k]和g[1][j][k]分别表示第一/二个人选完后，第一个人情况为j第二个人情况为k时的方案数。
那么我们对于同一类之中，从大到小循环进行DP(保证不能从当前类中的状态转移而来类似01背包)，然后不同类之间是可以互相转移的，所以在切换类的时候要进行f数组的更新，即f[j][k]=g[0][j][k]+g[1][j][k]-f[j][k]，至于为什么要减去f[j][k],因为当前这一类数两人都不选的情况在g[0],g[1]中都出现过了，所以要减去一个。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,p;
int f[1<<8][1<<8],g[2][1<<8][1<<8];
pair<int,int> stack[505];
int prime[8]={2,3,5,7,11,13,17,19};
int main()
{
    scanf("%d%d",&n,&p);
    for (int i=1;i<=n;i++)
    {
        int tmp=i;
        for (int j=0;j<8;j++)
            if (tmp%prime[j]==0)
            {
                stack[i].second|=(1<<j);
                while (tmp%prime[j]==0) tmp/=prime[j];
            }
        stack[i].first=tmp;
    }
    sort(stack+2,stack+n+1);
    f[0][0]=1;
    for (int i=2;i<=n;i++)
    {
        if (i==2||stack[i].first==1||stack[i].first!=stack[i-1].first)
            memcpy(g[0],f,sizeof(f)),memcpy(g[1],f,sizeof(f));
        for (int j=(1<<8)-1;~j;j--)
            for (int k=(1<<8)-1;~k;k--)
            {
                if ((stack[i].second&k)==0)
                    (g[0][stack[i].second|j][k]+=g[0][j][k])%=p;
                if ((stack[i].second&j)==0)
                    (g[1][j][stack[i].second|k]+=g[1][j][k])%=p;
            }
        if (i==n||stack[i].first==1||stack[i].first!=stack[i+1].first)
            for (int j=0;j<1<<8;j++)
                for (int k=0;k<1<<8;k++)
                    f[j][k]=((g[0][j][k]+g[1][j][k]-f[j][k])%p+p)%p;
    }
    int ans=0;
    for (int j=0;j<1<<8;j++)
        for (int k=0;k<1<<8;k++)
            if ((j&k)==0) (ans+=f[j][k])%=p;
    cout << ans << endl;
    return 0;
}