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 MB
Submit: 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