H 哲哲的疑惑

链接:https://www.nowcoder.com/acm/contest/90/H
来源:牛客网

题目描述
哲哲有l个球,球是不同的,现在她要用n种颜色给它们染色,如果一种染色方案有k种颜色没有用到,那么哲哲会产生C(k,m)的不满意度。
现在哲哲想求所有方案的不满意度之和,然而她只知道1+1=9,而她的神犇朋友maple去FW战队当中单了不在她身边,所以并不会求,你能帮帮她吗?
输入描述:
三个数n,m,l
1<=n,m<=10^7,l<=10^18
输出描述:
一个数(对998244353取模),表示所有方案的不满意度之和
示例1
输入
3 2 2
输出
3
说明
有以下方案:
两个球同色,有2种颜色没有用到,哲哲产生C(2,2)=1的不满意度,然后这里有三种方案,共产生3的不满意度
两个球不同色,有1种颜色没有用到,哲哲很开心(^▽^)
所以总共产生3的不满意度
示例2
输入
1634 1542 130
输出
93812204
说明
无可奉告
大佬们都在考虑m的意义,弱鸡却只会想到枚举k的取值,哎,C(k,m)实际上可看作是从k个不选择的颜色里选出m个来,而这相当于在n个选出m个不选就好了,之后剩下n-m个可选择的颜色,去填充l个格子即可。

#include <iostream>
#include <string>
#include <set>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <queue>
#include <cstring>
#include <stack>
#include <vector>
#include <map>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
#define mod 998244353ll
ll n, m, l;
ll quickmod(ll x, ll y) {
    ll ans = 1, p = x;
    while (y) {
        if (y & 1) {
            ans = ans * p;
            if (ans >= mod) {
                ans %= mod;
            }
        }
        p = p * p;
        if (p >= mod) {
            p %= mod;
        }
        y >>= 1;
    }
    return ans;
}
ll C(ll m, ll n) {
    ll s = 1, d = 1;
    for (int i = m - n + 1; i <= m; ++i) {
        s = (s * i) % mod;
    }
    for (int i = 1; i <= n; ++i) {
        d = (d * i) % mod;
    }
    return s * quickmod(d, mod - 2) % mod;
}
int main() {
    while (cin >> n >> m >> l) {
        printf("%lld\n", C(n, m) * quickmod(n - m, l) % mod);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/jack_zhuiyi/article/details/79721044