51nod1348 乘积之和 分治+NTT+中国剩余定理

Description

---

给出由N个正整数组成的数组A，有Q次查询，每个查询包含一个整数K，从数组A中任选K个（K <= N）把他们乘在一起得到一个乘积。求所有不同的方案得到的乘积之和，由于结果巨大，输出Mod 100003的结果即可。例如：1 2 3，从中任选1个共3种方法，{1} {2} {3}，和为6。从中任选2个共3种方法，{1 2} {1 3} {2 3}，和为2 + 3 + 6 = 11。

Solution

---

考虑设f[x]表示选x个数字的答案，那么合并两段序列就是一个卷积的事情
我们分治这个序列然后卷积合并两半，由于模数比较特殊可以拆系数FFT也可以2模数然后天朝剩余定理合并，由于我比较菜就写了第二种
合并模数会很大因此要上取模黑科技

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define copy(x,t) memcpy(x,t,sizeof(x))

typedef long long LL;
const int N=131075;

int rev[N],v[N],n;

LL s[21][N],b[21][N],c[21][N];

LL ksm(LL x,LL dep,int MOD) {
	LL res=1;
	for (;dep;dep>>=1) {
		(dep&1)?(res=res*x%MOD):0;
		x=x*x%MOD;
	}
	return res;
}

void NTT(LL *a,int n,int MOD,int f) {
	for (int i=0;i<n;++i) if (i<rev[i]) std:: swap(a[i],a[rev[i]]);
	for (int i=1;i<n;i<<=1) {
		int tmp=(MOD-1)/i/2;
		LL wn=ksm(3,(f==1)?(tmp):(MOD-1-tmp),MOD);
		for (int j=0;j<n;j+=(i<<1)) {
			LL w=1;
			for (int k=0;k<i;++k) {
				LL u=a[j+k],v=a[j+k+i]*w%MOD;
				a[j+k]=u+v; a[j+k+i]=u-v;
				(a[j+k]>=MOD)?(a[j+k]-=MOD):0;
				(a[j+k+i]<0)?(a[j+k+i]+=MOD):0;
				w=w*wn%MOD;
			}
		}
	}
	if (f==-1) {
		LL ny=ksm(n,MOD-2,MOD);
		for (int i=0;i<n;++i) a[i]=ny*a[i]%MOD;
	}
}

LL exgcd(LL a,LL b,LL &x,LL &y) {
	if (!b) {x=1; y=0; return a;}
	exgcd(b,a%b,x,y);
	LL tmp=x; x=y; y=tmp-(a/b)*x;
}

LL wjp(LL a,LL b,LL MOD) {
	return (a*b-(LL)((long double)a/MOD*b)*MOD+MOD)%MOD;
}

LL merge(LL a,LL b) {
	const LL MOD1=1004535809;
	const LL MOD2=998244353;
	const LL MOD=MOD1*MOD2;
	LL res=0,y;
	LL x1; exgcd(MOD2,MOD1,x1,y);
	res=(res+wjp(a*MOD2%MOD,x1,MOD))%MOD;
	LL x2; exgcd(MOD1,MOD2,x2,y);
	res=(res+wjp(b*MOD1%MOD,x2,MOD))%MOD;
	return res%100003;
}

void mul(LL *a,LL *b,LL *c,int n) {
	static LL a1[N],a2[N],b1[N],b2[N];
	int len,lg;
	for (len=1,lg=0;len<=n*2;len<<=1,lg++);
	for (int i=0;i<len;++i) {
		a1[i]=a2[i]=a[i];
		b1[i]=b2[i]=b[i];
	}
	for (int i=0;i<len;++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));
	const int MOD1=1004535809;
	NTT(a1,len,MOD1,1);
	NTT(b1,len,MOD1,1);
	for (int i=0;i<len;++i) a1[i]=a1[i]*b1[i]%MOD1;
	NTT(a1,len,MOD1,-1);
	const int MOD2=998244353;
	NTT(a2,len,MOD2,1);
	NTT(b2,len,MOD2,1);
	for (int i=0;i<len;++i) a2[i]=a2[i]*b2[i]%MOD2;
	NTT(a2,len,MOD2,-1);
	for (int i=0;i<len;++i) c[i]=merge(a1[i],a2[i]);
}

void solve(int dep,int l,int r) {
	if (l==r) {
		s[dep][0]=1; s[dep][1]=v[l]%100003;
		return ;
	}
	int mid=(l+r)>>1;
	solve(dep+1,l,mid);
	for (int i=0;i<mid-l+2;++i) b[dep][i]=s[dep+1][i];
	for (int i=mid-l+2;i<=r-l+1;++i) b[dep][i]=0;
	solve(dep+1,mid+1,r);
	for (int i=0;i<r-mid+1;++i) c[dep][i]=s[dep+1][i];
	for (int i=r-mid+1;i<=r-l+1;++i) c[dep][i]=0;
	mul(b[dep],c[dep],s[dep],r-l+1);
}

int main(void) {
	freopen("data.in","r",stdin);
	freopen("myp.out","w",stdout);
	int q; scanf("%d%d",&n,&q);
	for (int i=1;i<=n;++i) scanf("%d",&v[i]);
	solve(1,1,n);
	for (;q--;) {
		int x; scanf("%d",&x);
		printf("%lld\n", s[1][x]);
	}
	return 0;
}