2018 牛客多校第六场 C Generation I （组合数）

https://www.nowcoder.com/acm/contest/144/C

 

题意：给定n个集合， 要求用n次操作， 第i次操作用1~m中一个数填入 i ~ n个集合中， 集合无序而且元素不重复。

 

思路：

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e6+5;
const int mod=998244353;
ll inv[maxn];
void init()
{
    inv[0]=inv[1]=1;
    for(int i=2; i<maxn; ++i) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
}
int main()
{
    int T;
    scanf("%d",&T);
    init();
    int cas=0;
    while(T--)
    {
        ll n,m;
        scanf("%lld%lld",&n,&m);
        ll ans=0,A=m%mod,C=1;
        for(int i=1,j=0;i<=m&&j<=n-1;i++,j++)
        {
            ans=(ans+A*C%mod)%mod;
            A=(m-i)%mod*A%mod;
            C=(n-j-1)%mod*C%mod*inv[j+1]%mod;
        }
        printf("Case #%d: %lld\n", ++cas, ans);
    }
    return 0;
}