【2018牛客多校Round 6 C】Generation I

最新推荐文章于 2022-08-11 09:32:35 发布

转载最新推荐文章于 2022-08-11 09:32:35 发布 · 94 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/greenty1208/p/9428072.html

本文介绍了一种利用组合数学解决特定类型问题的方法。给定若干不相交的集合及可选择的颜色，通过数学组合计算不同操作下最终集合的状态数量。文章详细分析了问题，并给出了具体的实现代码。

题意

给定n个不可重复的集合，序号1～n，m种元素
执行n次操作:第i次操作，m种颜色中选出一种颜色，插入到集合i～n中
问最后的集合有多少种不同的状态

分析

考虑到，如果一个颜色重复选，那么只有第一次插入对集合效果有影响
问题转化成ｍ种颜色里先后选ｋ种，再选出这ｋ种颜色所对应的集合位置
ｍ种颜色先后选ｋ种，方案数为Ａ(k,m)
第一种颜色一定放在第一个位置，后面还剩n-1个位置，要放上k-1个颜色，顺序已经确定，则为C(k-1,n-1);
枚举ｋ从１到min(n,m),根据乘法原理和加法原理，得到最后的答案

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll mod = 998244353;
ll ans,n,m;
ll A,fz,fm;
ll inv[1000005];
ll ksm(ll a,ll b)
{
    ll res = 1;
    while(b){
    if(b&1) {res*=a;res%=mod;}
    a = (a*a)%mod;
    b>>=1;
    }
    return res%mod;
}
int main()
{
    int _,ca=0;scanf("%d",&_);
    inv[1] = 1;
    for(int i = 2;i<=1000000;i++) inv[i]=inv[mod%i]*(mod-mod/i)%mod;
    while(_--)
    {
    scanf("%lld%lld",&n,&m);
    ans = 0,A=1,fz=1,fm=1;
    long long nn = n<m?n:m;
    for(int k=1;k<=nn;k++)
    {
        A*=(m-k+1)%mod;A%=mod;
        if(k-1){
        fz*=((n-1)-k+2 )%mod;fz%=mod;
        fm*=inv[k-1]%mod;fm%=mod;
        }
        ans+=((fz*fm%mod)*A%mod);
        ans%=mod;
        }
    printf("Case #%d: %lld\n",++ca,ans);
    }
}

转载于:https://www.cnblogs.com/greenty1208/p/9428072.html