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;
}