4197: [Noi2015]寿司晚宴

状压dp。

500分解质因数的话,除了最大的质因数只需要8个质数,用二进制x储存,最大的质因数用y来储存(若没有比那8个质数大的质因数就使y=1)

用f[i][j]表示第一个人方案为i,第二个人方案为j时的方案数。

递推时用p[0/1][i][j]表示第1/2个人选当前数,i和j分别为两人方案时的方案数。

有f[i][j]=(p[0][i][j]+p[1][i][j]-f[i][j])%mod。 因为p数组都选了当前数,导致之前的方案计算两遍。

转移方程很好懂。

//quick
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 300 + 10;
const int d[]={2,3,5,7,11,13,17,19};
int f[maxn][maxn];
int p[3][maxn][maxn];
int n,mod,res;
struct data {
    int x,y;
}a[600];

bool cmp(data a,data b) {
    return a.y<b.y || (a.y==b.y && a.x<b.x);    
}

inline void build() {
    for(int i=2,t;i<=n;i++) {
        t=i;
        for(int j=0;j<8;j++) if(t%d[j]==0) {
            a[i].x=a[i].x|(1<<j);
            while(t%d[j]==0) t/=d[j];
            if(t==1) break;
        }
        a[i].y=t;
    }
}

int main() {
    scanf("%d%d",&n,&mod);
    build();    
    sort(a+2,a+n+1,cmp);
    f[0][0]=1;
    for(int i=2;i<=n;i++) {
        if(a[i].y==1 || a[i].y!=a[i-1].y) {
            memcpy(p[1],f,sizeof(f));
            memcpy(p[2],f,sizeof(f));
        }
        for(int j=255;j>=0;j--)
        for(int k=255;k>=0;k--) if((j&k)==0) {
            if((k&a[i].x)==0) p[1][j|a[i].x][k]=(p[1][j|a[i].x][k]+p[1][j][k])%mod;
            if((j&a[i].x)==0) p[2][j][k|a[i].x]=(p[2][j][k|a[i].x]+p[2][j][k])%mod;     
        }
        if(a[i].y==1||i==n||a[i].y!=a[i+1].y) {
            for(int j=0;j<=255;j++)
            for(int k=0;k<=255;k++) if((j&k)==0) 
                f[j][k]=(((p[1][j][k]+p[2][j][k]-f[j][k])%mod)+mod)%mod;
        }
    }
    for(int i=0;i<=255;i++)
    for(int j=0;j<=255;j++) 
    res=(res+f[i][j])%mod;
    
    printf("%d\n",res);
    return 0;
}
posted @ 2016-07-07 13:52  invoid  阅读(163)  评论(0编辑  收藏  举报