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;
}