求助!到底是哪里来的常数让它4s都跑不过去
  • 板块学术版
  • 楼主YingLi
  • 当前回复9
  • 已保存回复9
  • 发布时间2021/1/27 10:48
  • 上次更新2023/11/5 04:19:44
查看原帖
求助!到底是哪里来的常数让它4s都跑不过去
116824
YingLi楼主2021/1/27 10:48

NTT的一个题目,T的范围是1e3,a和b的范围是1e5,a和b的绝对值之差不超过5,保证tot全程>0。

程序复杂度按理说是O(T2logT)O(T^2logT)的,但4s都跑不过去就很离谱QAQ

#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
#include<cstdio>
#include<cmath>
#include<queue>
#define maxn 100005
using namespace std;
typedef long long ll;
const int mod = 998244353;
const int mx = 1e5;
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 fac[maxn], inv[maxn];
int pw(int a, int b) {int res = 1; while(b) {if(b & 1) res = 1ll * res * a % mod; a = 1ll *  a * a % mod, b >>= 1;} return res;}
int r[maxn], len, l;
void NTT(int *c, int flag) {
	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) {
		int gn = pw(3, (mod - 1) / (mid << 1));
		if(flag == -1) gn = pw(gn, mod - 2);
		for(int ls = 0, L = mid << 1; ls < len; ls += L) {
			int g = 1;
			for(int k = 0; k < mid; k++, g = 1ll * g * gn % mod) {
				ll x = c[ls + k], y = 1ll * g * c[ls + mid + k] % mod;
				c[ls + k] = (1ll * x + y) % mod, c[ls + mid + k] = (1ll * x - y + mod) % mod;
			}
		}
	}
	int rev = pw(len, mod - 2);
	if(flag == -1) for(int i = 0; i <= len; i++) c[i] = 1ll * c[i] * rev % mod;
}

int T, a, b;
int g[maxn], f[maxn];
signed main() {
	fac[0] = inv[0] = 1;
	for(int i = 1; i <= mx; i++) fac[i] = 1ll * fac[i - 1] * i % mod;
	inv[mx] = pw(fac[mx], mod - 2);
	for(int i = mx - 1; i; i--) inv[i] = 1ll * inv[i + 1] * (i + 1) % mod;
	
	register int tot = 0;
	T = read(); f[0] = 1;
	ll ans = 0;
	while(T--) {
		a = read(), b = read();
		len = 1, l = 0;
		while(len <= a - b + tot + tot + tot) len <<= 1, l++;
		for(int i = 0; i <= len; i++) r[i] = (r[i >> 1] >> 1) | ((i & 1) << l - 1), g[i] = 0;
		
		for(int i = 0; i <= a - b + tot + tot; i++) if(b + (i - tot) >= 0 && a - (i - tot) >= 0)
			g[i] = 1ll * inv[b + i - tot] * inv[a - i + tot] % mod;
		for(int i = tot + 1; i <= len; i++) f[i] = 0;
		NTT(f, 1), NTT(g, 1);
		for(int i = 0; i <= len; i++) f[i] = 1ll * f[i] * g[i] % mod;
		NTT(f, -1);
		for(int i = 0; i <= tot + a - b; i++) f[i] = 1ll * f[i + tot] * fac[a + b] % mod;
		tot += a - b;
	}
	for(int i = 0; i <= tot; i++) ans = (ans + f[i]) % mod;
	printf("%lld\n", ans);
	return 0;
}
2021/1/27 10:48
加载中...