2020牛客暑期多校训练营（第五场）C Easy —— 生成函数，有丶东西

This way

题意：

题解：

推荐题解：This way
大致方法就是将a和b的贡献列出来，然后会得到一个n*m的贡献矩阵，消掉之后会得到这样一个东西：
$(1-x)\ast S=xy\ast G(y)\ast G(xy)$
然后再乘上$G(x)$
$\to S=xy\ast G(x)\ast G(y)\ast G(xy)$
那么
$S^k=x^k{y}^k\ast G^k(x)\ast G^k(y)\ast G^k(xy)$
那么我们要求$x^n{y}^m$也就相当于
$x^k{y}^k\ast G^k(x)\ast G^k(y)\ast G^k(xy)=x^n{y}^m$
$\to G^k(x)\ast G^k(y)\ast G^k(xy)=x^{n-k}{y}^{m-k}$
我们知道$G^k(x)=\sum _{i=0}{C}_{k+i-1}^{i-1}{x}_i$
那么我们枚举$G(xy)$贡献$i$个$xy$，去求$G^k(x)\ast G^k(y)=x^{n-k-i}{y}^{m-k-i}$即可
于是最终的式子为$\sum _{i=0}^{min(n,m)}{C}_{n-k-i+k-1}^{k-1}\ast C_{m-k-i+k-1}^{k-1}\ast C_{i+k-1}^{k-1}$

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
#define ll long long 
const ll mod=998244353;
ll fac[N],inv[N];
ll c(int n,int m){
    if(m>n||m<0)return 0;
    return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
ll qpow(ll a,ll b){ll ans=1;for(;b;b>>=1,a=a*a%mod)if(b&1)ans=ans*a%mod;return ans;}
int main()
{
    fac[0]=inv[0]=1;
    for(int i=1;i<N;i++)fac[i]=fac[i-1]*i%mod;
    inv[N-1]=qpow(fac[N-1],mod-2);
    for(int i=N-2;i;i--)
        inv[i]=inv[i+1]*(i+1)%mod;
    int t;
    scanf("%d",&t);
    while(t--){
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        ll ans=0;
        for(int i=0;i<=min(n,m);i++)
            ans=(ans+c(n-i-1,k-1)*c(m-i-1,k-1)%mod*c(i+k-1,k-1))%mod;
        printf("%lld\n",ans);
    }
    return 0;
}