Description
问从 [0,n−1] [ 0 , n − 1 ] 中选取 m m 个数使得其和为的方案数
Input
第一行一整数 T T 表示用例组数,每组用例输入三个整数
(1≤T≤500,1≤n,m≤105,0≤k≤105) ( 1 ≤ T ≤ 500 , 1 ≤ n , m ≤ 10 5 , 0 ≤ k ≤ 10 5 )
Output
输出方案数,结果模 998244353 998244353
Sample Input
4
2 3 3
2 3 4
3 3 3
128 3 340
Sample Output
1
0
7
903
Solution
假设 m m 个数中有至少个数超过 n n 且和为的方案数为 F(i) F ( i ) ,由插板法知 F(i)=CimCm−1k−ni+m−1 F ( i ) = C m i C k − n i + m − 1 m − 1 ,由容斥原理知答案即为 ∑(−1)iF(i) ∑ ( − 1 ) i F ( i ) ,时间复杂度 O(m) O ( m )
Code
#include<cstdio>
using namespace std;
typedef long long ll;
#define mod 998244353
#define maxn 200005
int mul(int x,int y)
{
ll z=1ll*x*y;
return z-z/mod*mod;
}
int add(int x,int y)
{
x+=y;
if(x>=mod)x-=mod;
return x;
}
int fact[maxn],inv[maxn];
void init(int n=200000)
{
fact[0]=1;
for(int i=1;i<=n;i++)fact[i]=mul(i,fact[i-1]);
inv[1]=1;
for(int i=2;i<=n;i++)inv[i]=mul(mod-mod/i,inv[mod%i]);
inv[0]=1;
for(int i=1;i<=n;i++)inv[i]=mul(inv[i-1],inv[i]);
}
int C(int n,int m)
{
if(n<0||m<0||m>n)return 0;
return mul(fact[n],mul(inv[m],inv[n-m]));
}
int main()
{
init();
int T,n,m,k;
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&k);
if(k/m>n-1)
{
printf("0\n");
continue;
}
int ans=0;
for(int i=0;i<=k/n;i++)
{
if(i&1)ans=add(ans,mod-mul(C(m,i),C(k-n*i+m-1,m-1)));
else ans=add(ans,mul(C(m,i),C(k-n*i+m-1,m-1)));
}
printf("%d\n",ans);
}
return 0;
}