BZOJ 4197: [Noi2015]寿司晚宴( dp )
N^0.5以内的质数只有8个, dp(i, j, k)表示用了前i个大质数(>N^0.5), 2人选的质数(<=N^0.5)集合分别为j, k时的方案数. 转移时考虑当前的大质数p是给哪个人即可. 时间复杂度O(N*2^16)
-----------------------------------------------------------------
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
#define b(x) (1 << (x))
const int maxn = 509;
const int maxp = 8;
const int p[8] = {2, 3, 5, 7, 11, 13, 17, 19};
int N, P;
int b[maxn], bp[maxn], r[maxn];
int dp[b(maxp)][b(maxp)], A[b(maxp)][b(maxp)], B[b(maxp)][b(maxp)];
inline void upd(int &x, int t) {
if((x += t) >= P) x -= P;
}
inline bool Cmp(const int &l, const int &r) {
return bp[l] < bp[r];
}
int main() {
bool F = false;
scanf("%d%d", &N, &P);
for(int i = 2; i <= N; i++) {
b[i] = bp[i] = 0;
int v = r[i] = i;
for(int j = 0; j < 8; j++) if(v % p[j] == 0) {
b[i] |= b(j);
while(v % p[j] == 0) v /= p[j];
if(v == 1) break;
}
if(v != 1) bp[i] = v, F = true;
}
sort(r + 2, r + N + 1, Cmp);
memset(dp, 0, sizeof dp);
dp[0][0] = 1;
for(int i = 2; i <= N; i++) if(!bp[r[i]]) {
for(int x = b(maxp); x--; )
for(int y = b(maxp); y--; ) {
upd(dp[x | b[r[i]]][y], dp[x][y]);
upd(dp[x][y | b[r[i]]], dp[x][y]);
}
}
if(!F) {
int ans = 0;
for(int i = b(maxp); i--; )
for(int j = b(maxp); j--; )
if(!(i & j)) upd(ans, dp[i][j]);
printf("%d\n", ans);
return 0;
}
for(int i = 2, v = 0; i <= N; i++) if(bp[r[i]]) {
if(bp[r[i]] != v) {
if(v) {
for(int x = b(maxp); x--; )
for(int y = b(maxp); y--; )
dp[x][y] = ((A[x][y] + B[x][y] - dp[x][y]) % P + P) % P;
}
memcpy(A, dp, sizeof A);
memcpy(B, dp, sizeof B);
v = bp[r[i]];
}
for(int x = b(maxp); x--; )
for(int y = b(maxp); y--; ) {
upd(A[x | b[r[i]]][y], A[x][y]);
upd(B[x][y | b[r[i]]], B[x][y]);
}
}
int ans = 0;
for(int i = b(maxp); i--; )
for(int j = b(maxp); j--; )
if(!(i & j)) upd(ans, ((A[i][j] + B[i][j] - dp[i][j]) % P + P) % P);
printf("%d\n", ans);
return 0;
}
-----------------------------------------------------------------
4197: [Noi2015]寿司晚宴
Time Limit: 10 Sec Memory Limit: 512 MBSubmit: 164 Solved: 119
[Submit][Status][Discuss]
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
