p2150 [NOI2015]寿司晚宴

传送门

分析

我们发现对于大于$\sqrt(n)$的数每个数最多只会包含一个

所以我们把每个数按照大质数的大小从小到大排序

我们知道对于一种大质数只能被同一个人取

所以f1表示被A取，f2表示被B取

最终答案就是这两个的答案减去啥都不去的答案

因为啥都不去会被重复记录两次

对于小质数则直接状压转移即可

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define int long long
const int p1[] = {,,,,,,,,};
const int cnt1 = ;
int dp[][],n,mod,f1[][],f2[][];
struct node {
    int msk,ano;
};
node d[];
inline void init(int x){
    int res=x+;
    for(int i=;i<=cnt1;i++){
      if(res==)break;
      if(res%p1[i]==)d[x].msk|=(<<(i-));
      while(res%p1[i]==)res/=p1[i];
    }
    d[x].ano=res;
}
inline bool cmp(const node x,const node y){return x.ano<y.ano;}
signed main(){
    int i,j,k,p,q;
    scanf("%lld%lld",&n,&mod);
    n--;
    for(i=;i<=n;i++)init(i);
    sort(d+,d+n+,cmp);
    dp[][]=;
    for(i=;i<=n;i++){
      if(d[i].ano==||d[i].ano!=d[i-].ano){
          memcpy(f1,dp,sizeof(f1));
          memcpy(f2,dp,sizeof(f2));
      }
      for(j=(<<)-;j>=;j--)
        for(k=(<<)-;k>=;k--)if(!(j&k)){
          int msk1=j|d[i].msk,msk2=k|d[i].msk;
          if(!(msk1&k))f1[msk1][k]=(f1[msk1][k]+f1[j][k])%mod;
          if(!(msk2&j))f2[j][msk2]=(f2[j][msk2]+f2[j][k])%mod;
        }
      if(d[i].ano==||d[i].ano!=d[i+].ano||i==n){
        for(j=(<<)-;j>=;j--)
          for(k=(<<)-;k>=;k--)if(!(j&k)){
            dp[j][k]=(-dp[j][k]+mod)%mod;
            dp[j][k]=(dp[j][k]+f1[j][k])%mod;
            dp[j][k]=(dp[j][k]+f2[j][k])%mod;
          }
      }
    }
    int Ans=;
    for(j=;j<(<<);j++)
      for(k=;k<(<<);k++)
        if(!(j&k))Ans=(Ans+dp[j][k])%mod;
    cout<<Ans;
    return ;
}