来自FallDream的博客,未经允许,请勿转载,谢谢。
为了庆祝NOI的成功开幕,主办方为大家准备了一场寿司晚宴。小G和小W作为参加NOI的选手,也被邀请参加了寿司晚宴。
在晚宴上,主办方为大家提供了n−1种不同的寿司,编号1,2,3,⋯,n-1,其中第种寿司的美味度为i+1(即寿司的美味度为从2到n)。
现在小G和小W希望每人选一些寿司种类来品尝,他们规定一种品尝方案为不和谐的当且仅当:小G品尝的寿司种类中存在一种美味度为x的寿司,小W品尝的寿司中存在一种美味度为y的寿司,而x与y不互质。
现在小G和小W希望统计一共有多少种和谐的品尝寿司的方案(对给定的正整数p取模)。注意一个人可以不吃任何寿司。
n<=500
考虑状压dp,发现500以内的数字最多只会有一个大于19的因数,而19以内的质数只有8个
用f[i][j]表示一个人选的寿司含有质因数的状态是i,另一个人是j的方案数,我们按照大于19的因数排序,同一个因数一起转移即可
复杂度n*2^16
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #define pa pair<int,int> #define mp(x,y) make_pair(x,y) #define MN 256 using namespace std; inline int read() {int x = 0; char ch = getchar();while(ch < '0' || ch > '9') ch = getchar();while(ch >= '0' && ch <= '9'){x = x * 10 + ch - '0';ch = getchar();}return x; }int n,P,f[1<<8][1<<8],G[1<<8][1<<8],H[1<<8][1<<8]; const int S[8]={2,3,5,7,11,13,17,19}; pa num[505];void Solve(int x) {int T=num[x].second;for(int i=MN-1;~i;--i)for(int j=MN-1;~j;--j){int g=G[i][j],h=H[i][j];if(!(i&T)){G[i][j|T]=((G[i][j|T]+g)%P+f[i][j])%P;if(num[x].first==1) (G[i][j|T]+=h)%=P;}if(!(j&T)){H[i|T][j]=((H[i|T][j]+h)%P+f[i][j])%P;if(num[x].first==1) (H[i|T][j]+=g)%=P;}} }int main() {n=read();P=read();for(int i=2;i<=n;++i){int t=i;for(int j=0;j<8;++j)for(;!(t%S[j]);num[i].second|=(1<<j))t/=S[j];num[i].first=t;}sort(num+2,num+n+1);f[0][0]=1;for(int i=2,j;i<=n;i=j){memset(G,0,sizeof(G));memset(H,0,sizeof(H));for(j=i;j<=n&&num[j].first==num[i].first;++j)Solve(j);for(int k=0;k<MN;++k)for(int l=0;l<MN;++l)f[k][l]=((f[k][l]+G[k][l])%P+H[k][l])%P;}int ans=0;for(int i=0;i<MN;++i)for(int j=0;j<MN;++j)(ans+=f[i][j])%=P;printf("%d\n",ans);return 0; }