【CF】438E. The Child and Binary Tree

系统 1538 0

http://codeforces.com/contest/438/problem/E

题意:询问每个点权值在 $c_1, c_2, ..., c_m$ 中,总权值和为 $s$ 的二叉树个数。请给出每个$s \in [1,S]$ 对应的答案。($S,m < 10^5$)

      #include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N=(1e5+10)*4, mo=998244353;

int two, G[30], nG[30], rev[N];

int ipow(int a, int b) { int x=1; for(; b; b>>=1, a=(ll)a*a%mo) if(b&1) x=(ll)x*a%mo; return x; }

void fft_init() {

	two=ipow(2, mo-2); G[23]=ipow(3, (mo-1)/(1<<23)); nG[23]=ipow(G[23], mo-2);

	for(int i=22; i; --i) G[i]=(ll)G[i+1]*G[i+1]%mo, nG[i]=(ll)nG[i+1]*nG[i+1]%mo;

}

int getlen(int n) {

	int len=1, bl=-1;

	for(; len<n; len<<=1, ++bl);

	for(int i=1; i<len; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<bl);

	return len;

}

void fft(int *a, int n, int f) {

	for(int i=0; i<n; ++i) if(i<rev[i]) swap(a[i], a[rev[i]]);

	for(int m=2, now=1; m<=n; m<<=1, ++now) {

		int mid=m>>1, w=1, wn=G[now], u, v; if(f) wn=nG[now];

		for(int i=0; i<n; i+=m, w=1) 

			for(int j=0; j<mid; ++j) {

				u=a[i+j], v=(ll)a[i+j+mid]*w%mo;

				a[i+j]=(u+v)%mo; a[i+j+mid]=(u-v+mo)%mo; w=(ll)w*wn%mo;

			}

	}

}

void getinv(int *a, int *b, int n) {

	if(n==1) { b[0]=ipow(a[0], mo-2); return; }

	getinv(a, b, (n+1)>>1);

	static int c[N], d[N];

	memcpy(c, a, sizeof(int)*(n)); memcpy(d, b, sizeof(int)*((n+1)>>1));

	int len=getlen(n+n-1), nlen=ipow(len, mo-2);

	fft(c, len, 0); fft(d, len, 0);

	for(int i=0; i<len; ++i) d[i]=(ll)d[i]*(2-(ll)d[i]*c[i]%mo+mo)%mo;

	fft(d, len, 1);

	for(int i=0; i<n; ++i) b[i]=(ll)d[i]*nlen%mo;

	memset(c, 0, sizeof(int)*(len)); memset(d, 0, sizeof(int)*(len));

}

void getroot(int *a, int *b, int n) {

	if(n==1) { b[0]=sqrt(a[0]); return; }

	getroot(a, b, (n+1)>>1);

	static int c[N], d[N];

	memcpy(c, a, sizeof(int)*(n));

	getinv(b, d, n);

	int len=getlen(n+n-1), nlen=ipow(len, mo-2);

	fft(c, len, 0); fft(d, len, 0);

	for(int i=0; i<len; ++i) d[i]=(ll)c[i]*d[i]%mo;

	fft(d, len, 1);

	for(int i=0; i<n; ++i) b[i]=(ll)two*((b[i]+(ll)d[i]*nlen%mo)%mo)%mo;

	memset(d, 0, sizeof(int)*(len)); memset(c, 0, sizeof(int)*(len));

}

int a[N], b[N];

int main() {

	fft_init();

	int m, n; scanf("%d%d", &n, &m);

	for(int i=0; i<n; ++i) { int x; scanf("%d", &x); if(x<=m) a[x]=mo-4; }

	a[0]=1;

	getroot(a, b, m+1);

	b[0]=(b[0]+1)%mo;

	getinv(b, a, m+1);

	for(int i=1; i<=m; ++i) printf("%d\n", (a[i]<<1)%mo);

	return 0;

}


    

多项式求根= =具体看picks博客..http://picks.logdown.com/posts/202388-square-root-of-polynomial

其实想到了母函数然后知道用倍增来求根本题就解决了= =...跪跪跪orz

一定要注意那些多项式的次数啊!!!一定要想明白啊!!!

【CF】438E. The Child and Binary Tree


更多文章、技术交流、商务合作、联系博主

微信扫码或搜索:z360901061

微信扫一扫加我为好友

QQ号联系: 360901061

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描下面二维码支持博主2元、5元、10元、20元等您想捐的金额吧,狠狠点击下面给点支持吧,站长非常感激您!手机微信长按不能支付解决办法:请将微信支付二维码保存到相册,切换到微信,然后点击微信右上角扫一扫功能,选择支付二维码完成支付。

【本文对您有帮助就好】

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描上面二维码支持博主2元、5元、10元、自定义金额等您想捐的金额吧,站长会非常 感谢您的哦!!!

发表我的评论
最新评论 总共0条评论