CF1251F·Red-White Fence

最新推荐文章于 2022-02-15 20:50:16 发布

樱狸❀

最新推荐文章于 2022-02-15 20:50:16 发布

阅读量249

点赞数

分类专栏：数论文章标签：多项式NTT 组合计数

本文链接：https://blog.csdn.net/qq_43326267/article/details/104136596

版权

数论专栏收录该内容

47 篇文章 2 订阅

订阅专栏

初见安~这里是传送门：Codeforces Educational Codeforces Round 75 F Red-White Fence

题解

首先要抓住两个很显然的要点：第一个，所谓周长就是2*（红木板高度+木板数量），所以方便处理我们可以直接把询问的周长除以2。第二个，关于放置的过程我们选出来了比红板子小的板子后肯定是从小到大排序，然后每一个都看是放在左边还是放在右边，这样堆过来就是一个单峰序列了。但问题是，数会给出重复的，并且序列要求严格单调。也就是说，同一个数左边放一个右边放一个，顶多放两个。所以就可以分成两个集合：【假设要在集合里选k个板子出来】
1、只出现一次的有n1个，两边都可以放，方案数为 $2^{n_1}C_{n_1}^k$ ；
2、出现一次以上的有n2/2种，我们都视为每种有两个，选了前一个就是放左边，后一个就是放右边，方案数为 $C_{n_2}^k$ 。

至此，我们再回到题目，我们可以枚举红板子，再处理出与该红板子组合的所有方案对各种周长的贡献，最后直接O1查询即可。也就是说我们要枚举周长的一一半，然后用两个集合内的板子来凑数量。式子就是：【假设一共要选k个白板】

$\sum_{i=0}^kC_{n_1}^iC_{n_2}^{k-i}$

可以发现这个可以用NTT卷起来。最后累加到ans数组里即可。注意，统计的时候实际周长的一半是红板高度+白板数量+1（红板一个）。

上代码——

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#define maxn 1200005
using namespace std;
typedef long long ll;
const int mod = 998244353, mx = 300001;
int read() {
	int x = 0, f = 1, ch = getchar();
	while(!isdigit(ch)) {if(ch == '-') f = -1; ch = getchar();}
	while(isdigit(ch)) x = (x << 1) + (x << 3) + ch - '0', ch = getchar();
	return x * f;
}

int n, k, cnt[maxn], a[maxn], Q;
ll fac[maxn], inv[maxn], ans[maxn], F[maxn], G[maxn];
ll pw(ll a, ll b) {ll res = 1; while(b) {if(b & 1) res = res * a % mod; a = a * a % mod, b >>= 1;} return res;}

int len, l, r[maxn];
void NTT(ll *c, int op) {
	for(int i = 1; i <= len; i++) if(i < r[i]) swap(c[i], c[r[i]]);
	for(int mid = 1; mid < len; mid <<= 1) {
		ll gn = pw(3, (mod - 1) / (mid << 1));
		if(op == -1) gn = pw(gn, mod - 2);
		for(int ls = 0, L = mid << 1; ls < len; ls += L) {
			ll g = 1;
			for(int k = 0; k < mid; k++, g = g * gn % mod) {
				ll x = c[ls + k], y = c[ls + mid + k] * g % mod;
				c[ls + k] = (x + y) % mod, c[ls + k + mid] = (x - y + mod) % mod;
			}
		}
	}
	if(op == -1) {
		ll rev = pw(len, mod - 2);
		for(int i = 0; i <= len; i++) c[i] = c[i] * rev % mod;
	}
}

signed main() {
	n = read(), k = read();
	for(int i = 1; i <= n; i++) a[i] = read();
	for(int i = 1; i <= n; i++) cnt[a[i]] = min(2, cnt[a[i]] + 1);
	
	fac[0] = inv[0] = 1;
	for(int i = 1; i <= mx; i++) fac[i] = fac[i - 1] * i % mod;
	inv[mx] = pw(fac[mx], mod - 2);
	for(int i = mx - 1; i > 0; i--) inv[i] = inv[i + 1] * (i + 1) % mod;
	
	for(int i = 1, h; i <= k; i++) {
		h = read(); register int n1 = 0, n2 = 0;
		for(int j = 1; j < h; j++) if(cnt[j] == 1) n1++; else if(cnt[j] == 2) n2++;
		n2 *= 2;
		
		memset(F, 0, sizeof F), memset(G, 0, sizeof G);//F和G卷积
		for(int i = 0; i <= n1; i++) F[i] = pw(2, i) * fac[n1] % mod * inv[i] % mod * inv[n1 - i] % mod;
		for(int i = 0; i <= n2; i++) G[i] = fac[n2] * inv[i] % mod * inv[n2 - i] % mod;
		
		len = 1, l = 0;
		while(len <= n1 + n2) len <<= 1, l++;
		for(int i = 1; i <= len; i++) r[i] = (r[i >> 1] >> 1) | ((i & 1) << l - 1);
		NTT(F, 1), NTT(G, 1);
		for(int i = 0; i <= len; i++) F[i] = F[i] * G[i] % mod;
		NTT(F, -1);//↓注意ans的下标
		for(int i = 0; i <= n1 + n2; i++) ans[i + h + 1] = (ans[i + h + 1] + F[i]) % mod;
	}
	
	Q = read();
	while(Q--) {
		register int x = read();
		printf("%lld\n", ans[x / 2]);//这里要除以二，因为是统计的周长的一半
	}
	return 0;
}

迎评：）
——End——