【bzoj4197】[Noi2015]寿司晚宴

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

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#define inf 1000000
#define ll long long
#define M 255
int prime[8]={2,3,5,7,11,13,17,19};
using namespace std;
struct node{int x,y;}a[505];
int n,mo,ans;
int dp[2][300][300],f[300][300];
ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
bool cmp(node a,node b)
{
    if (a.y!=b.y) return a.y<b.y;
    else return a.x<b.x;
}
void pre()
{
    int x;
    for (int i=2;i<=n;i++)
    {
        x=i;
        for (int j=0;j<8;j++)
        {
            if (i%prime[j]==0)
            {
                a[i].x|=(1<<j);
                while (x%prime[j]==0)
                {
                    x/=prime[j];
                }
            }
        }
        a[i].y=x;
    }
    sort(a+2,a+n+1,cmp);
}
int main()
{
    n=read();mo=read();
    pre();f[0][0]=1;
    for (int i=2;i<=n;i++)
    {
        if (i==2||a[i].y!=a[i-1].y||a[i].y==1)
        {
            for (int j=0;j<=M;j++)
                for (int k=0;k<=M;k++)
                    dp[0][j][k]=dp[1][j][k]=f[j][k];
        }
        int p=a[i].x;
        for (int j=M;j>=0;j--)
            for (int k=M;k>=0;k--)
                if ((j&k)==0)
                {   
                    if ((p&k)==0) dp[0][j|p][k]+=dp[0][j][k],dp[0][j|p][k]%=mo;
                    if ((p&j)==0) dp[1][j][k|p]+=dp[1][j][k],dp[1][j][k|p]%=mo;
                }
        if (a[i].y!=a[i+1].y||a[i].y==1||i==n)
            for (int j=0;j<=M;j++)
                for (int k=0;k<=M;k++)
                    if ((j&k)==0)
                    {
                        f[j][k]=(dp[0][j][k]+dp[1][j][k]-f[j][k])%mo;
                        f[j][k]=(f[j][k]+mo)%mo;
                    }
    }
    for (int j=0;j<=M;j++)
        for (int k=0;k<=M;k++)
            if ((j&k)==0) ans+=f[j][k],ans%=mo;
    cout<<ans;
    return 0;
}