[UOJ#129][BZOJ4197][Noi2015]寿司晚宴

[UOJ#129][BZOJ4197][Noi2015]寿司晚宴

试题描述

为了庆祝 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 取模）。注意一个人可以不吃任何寿司。

输入

输入文件的第 1 行包含 2 个正整数 n,p，中间用单个空格隔开，表示共有 n 种寿司，最终和谐的方案数要对 p 取模。

输出

输出一行包含 1 个整数，表示所求的方案模 p 的结果。

输入示例

输出示例

数据规模及约定

2≤n≤500

0<p≤1000000000

题解

果然还是第三题最厉害。

根据“互质”这个条件，我们发现每个数可以用它的质因数表示即可。然后发现一个性质：每个数最多有一个质因数大于 sqrt(500)；而小于 sqrt(500) 的质数又只有 8 个。

根据以上性质，我们就可以进行状压 dp 了。

首先对 2 到 n 的每个数处理出两个值：big 最大质因数（若这个质因数小于 sqrt(500) 则 big = 1），s 这个数包含较小质因数的集合（易知 s 是一个 8 位的二进制）；

现在我们再读一遍题，问题就转化成了给每个数进行三种决定（把它给小 G，给小 W，或都不给），满足小 G、小 W 拿到的数的质因子没有交集；

对于小于 sqrt(500) 部分的质因子显然可以状压解决，对于那些含有大于 sqrt(500) 的质因子的数我们把它们按 big 值分类，那么每一类中的数的主人必须是同一个，或者没有主人。

分析到这里，怎样 dp 应该很明了了吧。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
using namespace std;

const int BufferSize = 1 << 16;
char buffer[BufferSize], *Head, *Tail;
inline char Getchar() {
	if(Head == Tail) {
		int l = fread(buffer, 1, BufferSize, stdin);
		Tail = (Head = buffer) + l;
	}
	return *Head++;
}
int read() {
	int x = 0, f = 1; char c = Getchar();
	while(!isdigit(c)){ if(c == '-') f = -1; c = Getchar(); }
	while(isdigit(c)){ x = x * 10 + c - '0'; c = Getchar(); }
	return x * f;
}

#define maxn 510
#define maxs 256

int n, MOD, f[maxn][3][maxs][maxs], prime[] = {2, 3, 5, 7, 11, 13, 17, 19};

struct Num {
	int big, s;

	bool operator < (const Num& t) const { return big < t.big; }

	void getNum(int x) {
		s = 0;
		for(int i = 0; i < 8; i++) if(x % prime[i] == 0) {
			s |= (1 << i);
			while(x % prime[i] == 0) x /= prime[i];
		}
		big = x;
		return ;
	}
} ns[maxn];

int main() {
	n = read(); MOD = read();

	for(int i = 2; i <= n; i++) ns[i].getNum(i);
	sort(ns + 2, ns + n + 1);
	int all = (1 << 8) - 1;
	f[1][0][0][0] = 1;
	ns[1].big = 1;
	for(int i = 1; i < n; i++)
		for(int cho = 0; cho < 3; cho++)
			for(int s1 = 0; s1 <= all; s1++)
				for(int s2 = 0; s2 <= all; s2++) if(!(s1 & s2) && f[i][cho][s1][s2]) {
					int tmp = f[i][cho][s1][s2];
					if(ns[i+1].big == 1) {
						(f[i+1][0][s1|ns[i+1].s][s2] += tmp) %= MOD;
						(f[i+1][0][s1][s2|ns[i+1].s] += tmp) %= MOD;
						(f[i+1][0][s1][s2] += tmp) %= MOD;
					}
					else if(ns[i+1].big == ns[i].big) {
						if(cho == 0) {
							(f[i+1][0][s1][s2] += tmp) %= MOD;
							(f[i+1][1][s1|ns[i+1].s][s2] += tmp) %= MOD;
							(f[i+1][2][s1][s2|ns[i+1].s] += tmp) %= MOD;
						}
						if(cho == 1) (f[i+1][1][s1|ns[i+1].s][s2] += tmp) %= MOD, (f[i+1][1][s1][s2] += tmp) %= MOD;
						if(cho == 2) (f[i+1][2][s1][s2|ns[i+1].s] += tmp) %= MOD, (f[i+1][2][s1][s2] += tmp) %= MOD;
					}
					else {
						(f[i+1][0][s1][s2] += tmp) %= MOD;
						(f[i+1][1][s1|ns[i+1].s][s2] += tmp) %= MOD;
						(f[i+1][2][s1][s2|ns[i+1].s] += tmp) %= MOD;
					}
				}

	int ans = 0;
	for(int cho = 0; cho < 3; cho++)
		for(int s1 = 0; s1 <= all; s1++)
			for(int s2 = 0; s2 <= all; s2++)
				if(!(s1 & s2)) (ans += f[n][cho][s1][s2]) %= MOD;
	printf("%d\n", ans);

	return 0;
}