bzoj5306 [HAOI2018]染色

http://www.elijahqi.win/archives/3800
题目背景
HAOI2018 Round2 第二题

题目描述
为了报答小 C 的苹果, 小 G 打算送给热爱美术的小 C 一块画布, 这块画布可 以抽象为一个长度为

N

N 的序列, 每个位置都可以被染成

M

M 种颜色中的某一种.

然而小 C 只关心序列的

N

N 个位置中出现次数恰好为

S

S 的颜色种数, 如果恰 好出现了

S

S 次的颜色有

K

K 种, 则小 C 会产生

W_k

Wk​ 的愉悦度.

小 C 希望知道对于所有可能的染色方案, 他能获得的愉悦度的和对

1004535809

1004535809 取模的结果是多少.

输入输出格式
输入格式：

从标准输入读入数据. 第一行三个整数

N, M, S

N,M,S .

接下来一行

M + 1

M+1 个整数, 第

$i$个数表示

$W_{i-1}$

输出格式：

输出到标准输出中. 输出一个整数表示答案．

输入输出样例
输入样例#1： 复制

8 8 3
3999 8477 9694 8454 3308 8961 3018 2255 4910
输出样例#1： 复制

524070430
输入样例#2： 复制

见 https://www.luogu.org/paste/rxrv9utg
输出样例#2： 复制

231524284
说明
特殊性质:

$\forall 1 \le i \le m, W_i = 0$

对于 $100\%$ 的数据, 满足

$0 \le W_i < 1004535809$

设lim=min(m,n/s)即 最多放多少个S长度的颜色
假设现在有k个s长度的颜色尝试求方案数
$C_{m}^{k}\times (C_{m}^{s}\times C_{m-s}^{s}...)\times (m-k)^{n-ks}$
可以发现这样会有重复的计算
那么不妨分成两段来计算 设g[m][n]表示不包含k种颜色中的颜色的随意组合且不会再出现s长度这种情况的方案数
那么我们可以把后面这段来容斥计算
带入
$\sum\limits_{i=0}^{lim}\frac{m!}{(m-i)!\times i!} \frac{n!}{(s!)^i(n-s)!}\sum\limits_{j=0}^{m-i}(-1)^{j}\frac{(m-i)!}{(m-i-j)!j!}\times \frac{(n-is)!}{(s!)^j(n-js-is)!}\times (m-i-j)^{n-js-is}$
考虑i和lim互换是并不影响什么的所以我们将i看成lim-i此外 将j替换为j-i
$\sum\limits_{i=0}^{lim}\frac{m!}{(m-i)!i!}\frac{n!}{(s!)^i(n-s)!}\sum\limits_{j=i}^{lim}(-1)^{j-i}\frac{(m-i)!}{(m-j)!(j-i)!}\times\frac{(n-is)!}{(s!)^{j-i}(n-js-is)!}(m-j)^{n-is}$
考虑构造卷积形式
$m!n!\sum\limits_{j=1}^{lim}\frac{(m-j)^{n-js}}{(m-j)!(s!)^j(n-js)!}\sum\limits_{i=0}^{j}\frac{w_i}{i!}\frac{(-1)^{j-i}}{(j-i)!}$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if (ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f;
}
const int g=3;
const int N=1e5+10;
const int mod=1004535809;
inline int ksm(ll b,int t){static ll tmp;
    for (tmp=1;t;b=b*b%mod,t>>=1) if (t&1) tmp=tmp*b%mod;return tmp;
}
inline int inc(int x,int v){return x+v>=mod?x+v-mod:x+v;}
inline int dec(int x,int v){return x-v<0?x-v+mod:x-v;}
int n,m,s,S,jc[N*100],inv[N*100],nn,a[N<<2],b[N<<2],R[N<<2],ans,w[N];
inline void ntt(int *x,int f){
    for (int i=0;i<n;++i) if (i<R[i]) swap(x[i],x[R[i]]);
    for (int i=1;i<n;i<<=1){
        ll wn=ksm(g,f==1?(mod-1)/(i<<1):mod-1-(mod-1)/(i<<1));
        for (int j=0;j<n;j+=i<<1){
            ll w=1,t1,t2;
            for (int k=0;k<i;++k,w=w*wn%mod){
                t1=x[j+k],t2=x[j+k+i]*w%mod;
                x[j+k]=inc(t1,t2);x[j+k+i]=dec(t1,t2);
            }
        }
    }
    if (f==-1){
        int invn=ksm(n,mod-2);
        for (int i=0;i<n;++i) x[i]=(ll)x[i]*invn%mod;
    }
}
int main(){
    freopen("bzoj5306.in","r",stdin);
    n=read();m=read();s=read();S=1;jc[0]=1;nn=n;
    for (int i=0;i<=m;++i) w[i]=read();int lim=min(m,n/s);
    for (int i=1;i<=s;++i) S=(ll)S*i%mod;S=ksm(S,mod-2);int nm=max(n,m);
    for (int i=1;i<=nm;++i) jc[i]=(ll)jc[i-1]*i%mod;inv[nm]=ksm(jc[nm],mod-2);
    for (int i=nm-1;~i;--i) inv[i]=(ll)inv[i+1]*(i+1)%mod;int l=0;
    for (n=1;n<=lim<<1;n<<=1,++l);for (int i=0;i<n;++i) R[i]=(R[i>>1]>>1)|((i&1)<<l-1);
    for (int i=0;i<=lim;++i) a[i]=(ll)inv[i]*w[i]%mod,b[i]=inv[i]*((i&1)?-1:1),b[i]=inc(b[i],mod);
    //n=4;a[0]=1;a[1]=2;b[0]=1;b[1]=2;b[2]=1;
    ntt(a,1);ntt(b,1);
    for (int i=0;i<n;++i) a[i]=(ll)a[i]*b[i]%mod;ntt(a,-1);
    for (int i=0;i<=lim;++i) 
    ans=inc(ans,(ll)ksm(m-i,nn-s*i)*inv[m-i]%mod*ksm(S,i)%mod*inv[nn-i*s]%mod*a[i]%mod);
    ans=(ll)ans*jc[nn]%mod*jc[m]%mod;printf("%d\n",ans);
    return 0;
}