NOI2015 寿司晚宴

传送门

好题呀。
首先两者所拿的集合必然是不相交的。首先我们考虑30分的暴力做法,30以内质数很少,我们可以进行一波状压,用(dp[i][j])表示第一个人取集合i,第二个人取集合j的方案数。我们记录当前数字的质因子集合为k,之后就可以进行转移了。
但是数据范围到了500,就会导致无法状压那么多的质数,我们考虑500以内的数,最多只可能有1个大于22的质因子。
这就意味着我们完全可以把它单独提出来处理,之后剩下的8个质数我们继续进行状压。
具体的操作方法就是,我们首先把所有的数按照最大质因子进行排序,相同的排在一起,因为相同的两者是不能同时选的。对于每个最大质因子,我们用(f1[][])表示这个最大质因子让第一个人选取,(f2[][])表示这个最大质因子让第二个人选取。
之后在小范围内进行状压DP即可。

记录当前元素的质因子集合为k,转移方程是:
(f1[S_1 | k][S_2] += f1[S_1][S_2])((k & S_2 == 0))
(f2[S_1][S_2 | k] += f2[S_1][S_2])((k & S_1 == 0))
之后合并答案的时候,(dp[i][j] = f1[i][j] + f2[i][j] - dp[i][j]),其中减去一次是考虑到两个人都没有选择这个最大质因子,这种情况被计算了两次。

时间复杂度(O(n2^{16}))

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i = a;i <= n;i++)
#define per(i,n,a) for(int i = n;i >= a;i--)
#define enter putchar('
')
#define fr friend inline
#define y1 poj

using namespace std;
typedef long long ll;
const int M = 1005;
const int INF = 1000000009;
const double eps = 1e-7;

int read()
{
   int ans = 0,op = 1;char ch = getchar();
   while(ch < '0' || ch > '9') {if(ch == '-') op = -1;ch = getchar();}
   while(ch >= '0' && ch <= '9') ans = ans * 10 + ch - '0',ch = getchar();
   return ans * op;
}

int n,P,p[9] = {0,2,3,5,7,11,13,17,19};
int tot,dp[300][300],f1[300][300],f2[300][300],ans;
bool np[M];

struct node
{
   int val,mprime,S;
   bool operator < (const node &g){return mprime < g.mprime;}
}a[M];
int inc(int a,int b){return (a+b) % P;}

void get(node &x)
{
   int cur = 0,t = x.val;
   rep(i,1,8)
   {
      if(t % p[i]) continue;
      cur |= (1 << (i-1));
      while(!(t % p[i])) t /= p[i];
   }
   x.S = cur,x.mprime = (t > 1) ? t : -1;
}

int main()
{
   n = read(),P = read();
   rep(i,2,n) a[i-1].val = i,get(a[i-1]);
   sort(a+1,a+n),dp[0][0] = 1;
   rep(i,1,n-1)
   {
      if(i == 1 || a[i].mprime != a[i-1].mprime || a[i].mprime == -1) memcpy(f1,dp,sizeof(f1)),memcpy(f2,dp,sizeof(f2));
      per(j,255,0)
      per(k,255,0)
      {
	 if(j & k) continue;
	 if(!(a[i].S & j)) f2[j][k | a[i].S] = inc(f2[j][k | a[i].S],f2[j][k]);
	 if(!(a[i].S & k)) f1[j | a[i].S][k] = inc(f1[j | a[i].S][k],f1[j][k]);
      }
      if(i == n-1 || a[i].mprime != a[i+1].mprime || a[i].mprime == -1)
      {
	 rep(j,0,255)
	    rep(k,0,255) if(!(j & k)) dp[j][k] = inc(f1[j][k],inc(f2[j][k],P - dp[j][k]));
      }
   }
   rep(i,0,255)
      rep(j,0,255) if(!(i&j) && dp[i][j]) ans = inc(ans,dp[i][j]);
   printf("%d
",ans);
   return 0;
}

原文地址:https://www.cnblogs.com/captain1/p/10491634.html