Character Encoding（hdu 6397 容斥）

题目：

Character Encoding

 

题意：

x1+x2+x3+...+xm=k，其中0<=xi<=n-1，已知n,m,k，求所有可能的情况数。

 

思路：

这是一个方程整数解的经典问题，可用容斥定理求解。

 

若xi>=0，无上限，那么总的情况数 = C(m+k-1，m-1) 。

（此公式可用隔板法推得：总共有k个1，要把它分成m段，需要m-1个隔板，那就是在总共k+m-1个东西中随机选m-1个当做隔板，方案数为C(m+k-1,m-1) 。）

现在统计的所有情况中有2类数：

第一类：0<=xi<=n-1，即满足条件的数。

第二类：xi>=n，即违反条件的数，将其减去n之后，它就满足xi>=0，就可以使用上面那个式子进行计算了。

设有p个不合法的数：

p=1时，总的情况数就为C(m+k-1-n*1,m-1)*C(m,1)，其意义为找出有一个违法的数的情况，并把这个数随机放到1-m的位置。

p=2时，总的情况数就为C(m+k-1-n*2,m-1)*C(m,1)，表示有2个数同时违法，并把这2个数随机放到1-m的位置。这种情况其实就是符合p=1时的两种可能情况的交集，这就是容斥的意义。

同理，p=3是就是符合p=1时的3种可能情况的交集。

...

根据容斥的性质，奇数个交集乘以容斥系数-1，偶数个交集乘以容斥系数1 。

因此，总的公式就为 。

 

代码：

#include <iostream>
#include <string>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <set>
#include <vector>
#include <cstdlib>
#include <algorithm>
using namespace std;

typedef long long ll;

const int MAX = 1e6+10;
const ll mod=998244353;

ll F[MAX], Finv[MAX], inv[MAX];//F是阶乘，Finv是逆元的阶乘
void init(){
    inv[1] = 1;
    for(int i = 2; i < MAX; i ++){
        inv[i] = (mod - mod / i) * 1ll * inv[mod % i] % mod;
    }
    F[0] = Finv[0] = 1;
    for(int i = 1; i < MAX; i ++){
        F[i] = F[i-1] * 1ll * i % mod;
        Finv[i] = Finv[i-1] * 1ll * inv[i] % mod;
    }
}
ll comb(ll n, ll m){//comb(n, m)就是C(n, m)
    if(m < 0 || m > n) return 0;
    return F[n] * 1ll * Finv[n - m] % mod * Finv[m] % mod;
}

int main()
{
	int T;
	scanf("%d",&T);
	init();
	while(T--)
	{
        ll n,m,k;
        scanf("%lld%lld%lld",&n,&m,&k);
        ll ans=0;
        for(int i=0;i<=min(k/n,m);i++){
            if(i%2==0)
                ans=(ans+comb(m+k-1-i*n,m-1)*comb(m,i)%mod+mod)%mod;
            else
                ans=(ans-comb(m+k-1-i*n,m-1)*comb(m,i)%mod+mod)%mod;
        }
        printf("%lld\n",ans);
	}
	return 0;
}