B-白兔的式子

链接： https://www.nowcoder.com/acm/contest/73/B
来源：牛客网

题目描述

已知 f[1][1]=1,f[i][j]=a*f[i-1][j]+b*f[i-1][j-1](i>=2,1<=j<=i)。

对于其他情况f[i][j]=0

有T组询问，每次给出a,b,n,m，求f[n][m] mod (998244353)

输入描述:

第一行为一个整数T，表示询问个数。
接下来一共T行，每行四个整数a,b,n,m。

输出描述:

一共T行，每行一个整数，表示f[n][m] mod (998244353)

根据公式可将f[n][m]看成是走n步到达m米，其中要么不走，要么一次走1米，由于f[1][1] = 1限定为了1种方式，所以实际上相当于只走n-1步，到达m-1米的方法数，另外不走的方法又有a种，走的方法又有b种，生成函数为(a+bx)^n-1，最终得到

f[n][m] = c(n-1,m-1) * a^(n - m) * b^(m - 1)

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define mod 998244353
long long d[100005];
long long quickpow(int a,int k){
    long long ans = 1,q = a % mod;
    while(k){
        if(k & 1){
            ans = ans * q % mod;
        }
        q = q * q % mod;
        k >>= 1;
    }
    return ans;
}
void init(){
    d[0] = d[1] = 1;
    for(int i = 2;i <= 100000;++i){
        d[i] = d[i - 1] * i % mod;
    }
}
int main()
{
    //cout << "Hello world!" << endl;
    int T,a,b,n,m;
    init();
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d%d",&a,&b,&n,&m);
        int p = m - 1;
        int q = n - 1;
        long long ans = quickpow(a,q-p) * quickpow(b,p) % mod;
        ans = ans * d[q] % mod;
        ans = ans * quickpow(d[q - p],mod - 2) % mod;
        ans = ans * quickpow(d[p],mod - 2) % mod;
        printf("%lld\n",ans);
 
    }
    return 0;
}