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

最新推荐文章于 2021-10-06 19:25:37 发布

原创最新推荐文章于 2021-10-06 19:25:37 发布 · 370 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

2018牛客多校同时被 3 个专栏收录

15 篇文章

订阅专栏

-------数论--------

8 篇文章

订阅专栏

-------图论--------

8 篇文章

订阅专栏

本文介绍了一种解决特定集合填充问题的方法。问题要求通过n次操作，在n个集合中填入1到m之间的数，每次操作从i到n的集合中填入一个数，且保证集合元素不重复。文章提供了完整的C++代码实现，并使用了动态规划的思想来求解。

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