BZOJ 4197: [Noi2015]寿司晚宴( dp )

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

上一篇:jQuery插件开发(转)


下一篇:发红包android