BZOJ 4197([Noi2015]寿司晚宴-状压dp)

Description

为了庆祝 NOI 的成功开幕，主办方为大家准备了一场寿司晚宴。小 G 和小 W 作为参加 NOI 的选手，也被邀请参加了寿司晚宴。

在晚宴上,主办方为大家提供了 n−1 种不同的寿司，编号 1,2,3,…,n−1，其中第 i 种寿司的美味度为 i+1 （即寿司的美味度为从 2 到 n）。
现在小 G 和小 W 希望每人选一些寿司种类来品尝，他们规定一种品尝方案为不和谐的当且仅当：小 G 品尝的寿司种类中存在一种美味度为 x 的寿司，小 W 品尝的寿司中存在一种美味度为 y 的寿司，而 x 与 y 不互质。
现在小 G 和小 W 希望统计一共有多少种和谐的品尝寿司的方案（对给定的正整数 p 取模）。注意一个人可以不吃任何寿司。
Input

输入文件的第 1 行包含 2 个正整数 n,p，中间用单个空格隔开，表示共有 n 种寿司，最终和谐的方案数要对 p 取模。

Output

输出一行包含 1 个整数，表示所求的方案模 p 的结果。

Sample Input

3 10000
Sample Output

9
HINT
$2≤n≤500$
$0<p≤1000000000$
Source

我们把n以内的数，提出来，显然每个数$>sqrt(n)$的素数只有一个，我们根据它把数分组，如果没有这个素数单独算一组。
<sqrt(n) <script type="math/tex" id="MathJax-Element-287"> 然后我们对每组01背包，注意分成p1,p2，保证每组数只分给1位同学。
然后合并，注意2者都不吃寿司的情况要减去。

#include<bits/stdc++.h> 
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define Forpiter(x) for(int &p=iter[x];p;p=next[p])  
#define Lson (o<<1)
#define Rson ((o<<1)+1)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,0x3f,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define MEMx(a,b) memset(a,b,sizeof(a));
#define INF (0x3f3f3f3f)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define vi vector<int> 
#define pi pair<int,int>
#define SI(a) ((a).size())
#define Pr(kcase,ans) printf("Case #%d: %lld\n",kcase,ans);
#define PRi(a,n) For(i,n-1) cout<<a[i]<<' '; cout<<a[n]<<endl;
#define PRi2D(a,n,m) For(i,n) { \
                        For(j,m-1) cout<<a[i][j]<<' ';\
                        cout<<a[i][m]<<endl; \
                        } 
#pragma comment(linker, "/STACK:102400000,102400000")
#define ALL(x) (x).begin(),(x).end()
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
ll F;
ll mul(ll a,ll b){return (a*b)%F;}
ll add(ll a,ll b){return (a+b)%F;}
ll sub(ll a,ll b){return ((a-b)%F+F)%F;}
#define upd(a,b) a=(a+b)%F;
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while(!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}
    while(isdigit(ch)) { x=x*10+ch-'0'; ch=getchar();}
    return x*f;
} 
int f[256][256],p[3][256][256];
pair<int,int> s[511];
int pri[9]={2,3,5,7,11,13,17,19},n,m=8;
int main()
{
//  freopen("bzoj4197.in","r",stdin);
//  freopen(".out","w",stdout);
    n=read();cin>>F;
    For(i,n) {
        int t=i;
        Rep(j,m) {
            while (t%pri[j]==0) {
                t/=pri[j]; s[i].se|=1<<j;
            }
        }
        s[i].fi=t;
    }
    sort(s+1,s+1+n);
    MEM(f) f[0][0]=1;
    Fork(i,2,n) {
        if (i==2|| s[i].fi==1 || s[i].fi!=s[i-1].fi) { //
            memcpy(p[1],f,sizeof(f));
            memcpy(p[2],f,sizeof(f));
        }
        int S=1<<m;
        RepD(j,S-1) RepD(k,S-1) {
            if (!(k&s[i].se)) upd(p[1][j|s[i].se][k],p[1][j][k]);
            if (!(j&s[i].se)) upd(p[2][j][k|s[i].se],p[2][j][k]);
        }
        if (i==n || s[i].fi==1 || s[i].fi!=s[i+1].fi) {
            RepD(j,S-1) RepD(k,S-1) {
                f[j][k]=((ll)p[1][j][k]+p[2][j][k]-f[j][k]+F)%F;
            }
        }
    }
    int S=1<<m;
    int ans=0;
    RepD(j,S-1) RepD(k,S-1) {
        upd(ans,f[j][k]); 
    }
    cout<<ans;
    return 0;
}