记录编号	417679	评测结果	AAAAAAAAAA
题目名称	2710.多项式多点求值	最终得分	100
用户昵称	FoolMike	是否通过	通过
代码语言	C++	运行时间	24.714 s
提交时间	2017-06-26 19:34:37	内存使用	46.08 MiB

#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const int N=1e6+10,p=119<<23|1;
int inc(int x,int y){x+=y;return x>=p?x-p:x;}
int dec(int x,int y){x-=y;return x<0?x+p:x;}
int mul(int x,int y){return (ll)x*y%p;}
int power(int x,int y){
	int ans=1;
	for (;y;y>>=1,x=mul(x,x))
		if (y&1) ans=mul(ans,x);
	return ans;
}
int w[N],iw[N];
void init(int n){
	w[0]=1;w[1]=power(3,(p-1)/n);
	for (int i=2;i<=n;i++) w[i]=mul(w[i-1],w[1]);
	for (int i=0;i<=n;i++) iw[i]=w[n-i];
}
void fft(int n,int *a,int *w){
	for (int i=0,j=0;i<n;i++){
		if (i<j) swap(a[i],a[j]);
		for (int k=n>>1;(j^=k)<k;k>>=1);
	}
	for (int i=2;i<=n;i<<=1){
		int m=i>>1,step=n/i;
		for (int j=0;j<n;j+=i)
		for (int k=0,pos=0;k<m;k++,pos+=step){
			int t=mul(a[j+k+m],w[pos]);
			a[j+k+m]=dec(a[j+k],t);
			a[j+k]=inc(a[j+k],t);
		}
	}
	if (w==iw){
		int del=power(n,p-2);
		for (int i=0;i<n;i++) a[i]=mul(a[i],del);
	}
}
void getinv(int n,int *a,int *b){
	if (n==1) return void(b[0]=power(a[0],p-2));
	getinv((n+1)>>1,a,b);
	int size=1;
	while (size<n+n) size<<=1;
	init(size);
	static int tmp[N];
	for (int i=0;i<n;i++) tmp[i]=a[i];
	for (int i=n;i<size;i++) tmp[i]=0;
	fft(size,tmp,w);fft(size,b,w);
	for (int i=0;i<size;i++) b[i]=mul(b[i],dec(2,mul(b[i],tmp[i])));
	fft(size,b,iw);
	for (int i=n;i<size;i++) b[i]=0;
}
void getmod(int n,int *a,int m,int *b,int *c){
	static int tmp[N],inv[N];
	int size=1;
	while (size<n*2) size<<=1;
	reverse(b,b+m);
	for (int i=0;i<size;i++) inv[i]=0;
	getinv(n-m+1,b,inv);
	reverse(b,b+m);
	for (int i=0;i<n;i++) tmp[i]=a[i];
	for (int i=n;i<size;i++) tmp[i]=0;
	reverse(tmp,tmp+n);
	init(size);
	fft(size,inv,w);fft(size,tmp,w);
	for (int i=0;i<size;i++) c[i]=mul(inv[i],tmp[i]);
	fft(size,c,iw);
	for (int i=n-m+1;i<size;i++) c[i]=0;
	reverse(c,c+n-m+1);
	for (int i=0;i<m;i++) tmp[i]=b[i];
	for (int i=m;i<size;i++) tmp[i]=0;
	fft(size,c,w);fft(size,tmp,w);
	for (int i=0;i<size;i++) c[i]=mul(c[i],tmp[i]);
	fft(size,c,iw);
	for (int i=0;i<size;i++) c[i]=dec(a[i],c[i]);
}
vector<int> k[N];
int n,m,a[N],X[N];
#define lc x<<1
#define rc x<<1|1
void build(int x,int l,int r){
	if (l==r){
		k[x].push_back(dec(0,X[l]));
		k[x].push_back(1);
		return;
	}
	int mid=(l+r)>>1;
	build(lc,l,mid);
	build(rc,mid+1,r);
	int size=1;
	while (size<=r-l+1) size<<=1;
	init(size);
	static int L[N],R[N];
	for (int i=0;i<k[lc].size();i++) L[i]=k[lc][i];
	for (int i=k[lc].size();i<size;i++) L[i]=0;
	for (int i=0;i<k[rc].size();i++) R[i]=k[rc][i];
	for (int i=k[rc].size();i<size;i++) R[i]=0;
	fft(size,L,w);fft(size,R,w);
	for (int i=0;i<size;i++) L[i]=mul(L[i],R[i]);
	fft(size,L,iw);
	for (int i=0;i<=r-l+1;i++) k[x].push_back(L[i]);
}
void calc_value(int x,int n,int *a,int *X){//求a(X)的值并保留到x中 
	if (n==1) return void(X[0]=a[0]);
	int m=n>>1;
	int *tmp=new int[n*2],*mod=new int[n*2];
	for (int i=0;i<k[lc].size();i++) tmp[i]=k[lc][i];
	for (int i=k[lc].size();i<n;i++) tmp[i]=0;
	for (int i=0;i<n;i++) mod[i]=0;
	getmod(n,a,k[lc].size(),tmp,mod);
	calc_value(lc,m,mod,X);
	for (int i=0;i<k[rc].size();i++) tmp[i]=k[rc][i];
	for (int i=k[rc].size();i<n;i++) tmp[i]=0;
	for (int i=0;i<n;i++) mod[i]=0;
	getmod(n,a,k[rc].size(),tmp,mod);
	calc_value(rc,m,mod,X+m);
}
int main()
{
	freopen("polynomial_calc_value.in","r",stdin);
	freopen("polynomial_calc_value.out","w",stdout);
	scanf("%d",&n);
	for (int i=0;i<n;i++) scanf("%d",&a[i]);
	scanf("%d",&m);
	for (int i=0;i<m;i++) scanf("%d",&X[i]);
	int size=1;
	while (size<n||size<m) size<<=1;
	build(1,0,size-1);
	calc_value(1,size,a,X);
	for (int i=0;i<m;i++) printf("%d ",X[i]);
	puts("");
	return 0;
}