[HAOI2018]染色

题意：给你一个长度为n的序列，你有m种颜色，每个位置可以染这m种颜色中的一种，如果序列中出现次数恰好为ｓ的颜色种类数为ｋ,那么会有的贡献，问所有染色方案的贡献和

解析：

设f[i]表示至少有i种颜色他们在这序列中出现的次数恰为s次的染色方案数

设g[i]表示恰好有i种颜色他们在这序列中出现的次数恰为s次的染色方案数

那么答案就是

我们可以发现放松限制之后f[i]挺好求

所以我们考虑一下f[i]还等于什么？

所以那么就直接二项式反演

还是把组合数拆开

所以上述东西直接NTT就行了

// luogu-judger-enable-o2
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=1004535809;
const int N=1e7;
const int M=4e5+10;
ll fac[N+10],inv[N+10];
ll w[M];
int n,m,s,len,sum;
ll f[M],g[M],rev[M];
ll ksm(ll x,ll y){
	ll ans=1;
	for (;y;y>>=1,x=(x*x)%mod) if (y&1) ans=(ans*x)%mod;
	return ans;
}
void init()//这里不是max(n,m)就Ｔ掉了，想想不知道为什么，可能数据不够强吧
{
	fac[0]=1;
	for (int i=1;i<=max(n,m);i++) fac[i]=(fac[i-1]*(ll)i)%mod;
	inv[0]=inv[1]=1;
    for (int i=2;i<=max(n,m);i++) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    for (int i=2;i<=max(n,m);i++) inv[i]=(inv[i-1]*inv[i])%mod;	
}
ll C(int x,int y){
	if (x<0||y<0||x<y) return 0;
	return fac[x]*inv[y]%mod*inv[x-y]%mod;
}
void NTT(ll *a,int len,int t) 
{
    for (int i=0;i<len;i++) if (i<rev[i]) swap(a[i],a[rev[i]]);
    for (int i=1;i<len;i<<=1) 
    {
        int s=(i<<1);
        ll wn=ksm(3,(mod-1)/s);
        if (t==-1) wn=ksm(wn,mod-2);
        for (int j=0;j<len;j+=s) 
        {
            ll w=1;
            for (int k=j;k<j+i;k++)
             {
                 ll x=a[k]; ll y=(a[k+i]*w)%mod;
                 a[k]=(x+y)%mod; a[k+i]=(x-y+mod)%mod;
                 w=(w*wn)%mod;
             }
        }
    }
    if (t==-1) {
        ll w=(ksm(len,mod-2));
        for (int i=0;i<len;i++) a[i]=(a[i]*w)%mod;
    }
}
void solve()
{
	for (len=1;len<=m*2;len<<=1) sum++;
	for (int i=0;i<len;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(sum-1));
	NTT(f,len,1); NTT(g,len,1);
	for (int i=0;i<len;i++) f[i]=(f[i]*g[i])%mod;
	NTT(f,len,-1);
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);
    init();
    for (int i=0;i<=m;i++) scanf("%lld",&w[i]);	
    for (int i=0;i<=m;i++) if (i*s<=n) f[i]=C(m,i)*C(n,i*s)%mod*fac[i*s]%mod*ksm(inv[s],i)%mod*fac[i]%mod*ksm(m-i,n-i*s)%mod; else break;
    for (int i=0;i<=m;i++) {g[m-i]=inv[i];if (i&1) g[m-i]=(mod-g[m-i])%mod;}
    // cout << f[0] << ' ' << f[1] << ' ' << f[2] << endl;
    solve();
    long long ans=0;
    for (int i=0;i<=m;i++) ans=(ans+f[i+m]*w[i]%mod*inv[i]%mod)%mod;
    cout << ans << endl;
}