HDU 6036 Division Game

HDU 6036 Division Game

考虑每堆石头最多操作 $ \sum e $ 次，考虑设 $ f(x) $ 表示某一堆石头（最开始都是一样的）操作 $ x $ 次后变成了 $ 1 $ 的方案数量。

明显的，某一堆石头操作了 $ x $ 次后仍然没有变成 $ 1 $ 的方案数量是 $ f(x+1) $。因为最后一次操作必然是把石头从某个数字拿到1。（这个算个小trick吧？）

那么对于第 \(k\) 堆石头答案就是 $ f^{k-1}(x+1) f^{m-i+1}(x) $

因为前 $ k - 1 $ 堆石头进行了 $ x $ 次拿石头的操作还没变成 $ 1 $，而从 $ k $ 后面所有的石头都进行 $ x - 1 $ 次操作并且没变成 $ 1 $ ，而第 $ k $ 堆石头是变成了 $ 1 $ 的。

然后考虑怎么计算 $ f(x) $

相当于我们有 $ m $ 种球每种 $ e_i $ 个放进 $ x $ 个不同的盒子里面，并且最后不能有盒子是空的。

如果可以有盒子是空的，这个会比较好算，对每种球分开考虑，并且分别用插板法，最后乘法原理答案是 $ f'(x)=\displaystyle \prod_{i=1}^m\binom{e_i+x-1}{x-1} $

但是这个还不是答案，因为可以为空，不能为空可以考虑容斥，总方案数减去至少一个为空加上至少两个为空... 容斥系数是 $ (-1)^{t} $ 其中 $ t $ 代表至少有 $ t $ 个位置是空的。

$ f(x) = \displaystyle\sum_{i=0}^x (-1)^{x-i}f'(i)\binom{x}{i} $

$ f'(x) $ 化一下发现很容易求所以 $ f(x) $ 就可以NTT辣

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
#define P 985661441
#define MAXN (1 << 19) + 13
int m , k;
int a[MAXN];
int Pow(int x,int y) {
    int res=1;
    while(y) {
        if(y&1) res=res*(ll)x%P;
        x=x*(ll)x%P,y>>=1;
    }
    return res;
}
int wn[2][MAXN];
void getwn(int l) {
    for(int i=1;i<(1<<l);i<<=1) {
        int w0=Pow(3,(P-1)/(i<<1)),w1=Pow(3,P-1-(P-1)/(i<<1));
        wn[0][i]=wn[1][i]=1;
        for(int j=1;j<i;++j)
            wn[0][i+j]=wn[0][i+j-1]*(ll)w0%P,
                    wn[1][i+j]=wn[1][i+j-1]*(ll)w1%P;
    }
}
int rev[MAXN];
void getr(int l) { for(int i=1;i<(1<<l);++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<l-1); }
void NTT(int *A,int len,int f) {
    for(int i=0;i<len;++i) if(rev[i]<i) swap(A[i],A[rev[i]]);
    for(int l=1;l<len;l<<=1)
        for(int i=0;i<len;i+=(l<<1))
            for(int k=0;k<l;++k) {
                int t1=A[i+k],t2=A[i+l+k]*(ll)wn[f][l+k]%P;
                A[i+k]=(t1+t2)%P;
                A[i+l+k]=(t1-t2+P)%P;
            }
    if( f == 1 ) for(int inv=Pow(len,P-2),i=0;i<len;++i) A[i]=A[i]*(ll)inv%P;
}
int J[MAXN] , invJ[MAXN] , p[MAXN];
int kase = 0;
int pp[20] , E[20] , F[MAXN] , n;
signed main() {
    J[0] = invJ[0] = 1;
    for( int i = 1 ; i < MAXN ; ++ i )
		J[i] = 1ll * J[i - 1] * i % P , invJ[i] = Pow( J[i] , P - 2 );
	while( cin >> m >> k ) {
		memset( p , 0 , sizeof p ) , memset( F , 0 , sizeof F );
		n = 1;
		for( int i = 1 ; i <= m ; ++ i ) {
			scanf("%d%d",&pp[i],&E[i]);
			n += E[i];
		}
		p[0] = 1;
		for( int i = 1 ; i <= n ; ++ i ) {
			F[i] = invJ[i] , p[i] = ( ( ( i & 1 ) ? -1 : 1 ) * invJ[i] + P ) % P;
			for( int j = 1 ; j <= m ; ++ j )
				F[i] = 1ll * F[i] * J[E[j] + i - 1] % P * invJ[i - 1] % P * invJ[E[j]] % P;
		}
//		cout << F[2] << endl;
		int len = 1 , l = 0;
		while( len <= n * 2 ) len <<= 1 , ++ l;
		getr( l ) , getwn( l );
		NTT( F , len , 0 ) , NTT( p , len , 0 );
		for( int i = 0 ; i < len ; ++ i ) F[i] = 1ll * F[i] * p[i] % P;
		NTT( F , len , 1 );
		for( int i = 0 ; i < len ; ++ i ) F[i] = 1ll * F[i] * J[i] % P;
//		cout << F[2] << endl;
		printf("Case #%d: ",++kase);
		for( int i = 1 ; i <= k ; ++ i ) {
			int res = 0;
			for( int x = 0 ; x < n ; ++ x )
				res = ( res + 1ll * Pow( F[x + 1] , i - 1 ) * Pow( F[x] , k - i + 1 ) % P ) % P ;
			printf("%d",res);
			if( i != k ) putchar(' ');
		}
		puts("");
	}
}