题目描述
哲哲有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的不满意度

思想:
这是一道思维题,首先他给我们了n种颜色,l个球,让我们计算它的满意度(题意自己理解)。它是让我们选出一种方案之后,然后求出它的不满意度,然后求总的不满意度。C(k,m)实际上可看作是从k个不选择的颜色里选出m个来,而这相当于在n个选出m个不选就好了,之后剩下n-m个可选择的颜色,去填充L个格子
ans就是 C(n,m)*(n-m)^l

#include <iostream>
#include <string>
#include <vector>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<map>
using namespace std;
#define ll long long
const int maxn = 1000010;
const int mod = 998244353;
const int inf = 0x3f3f3f3f;
ll gcd(ll a, ll b)
{
    if (!b)
        return a;
    else
        return gcd(b, a % b);
}
inline int read() {
    int s = 0, w = 1;
    char ch = getchar();
    while (ch < '0' || ch>'9') { if (ch == '-')w = -1; ch = getchar(); }
    while (ch >= '0' && ch <= '9') s = s * 10 + ch - '0', ch = getchar();
    return s * w;
}
long long int pow_mod(long long int a, long long int k)
{
    long long int ans = 1;
    a %= mod;
    while (k)
    {
        if (k % 2) ans *= a;
        a = (a * a) % mod;
        k /= 2;
        ans %= mod;
    }
    return ans;
}
int main()
{
    std::ios::sync_with_stdio(false);
    ll n, m, l;
    cin >> n >> m >> l;
    ll ans = 1;
    for (ll i = m + 1; i <= n; i++)
    {
        ans *= i;
        ans %= mod;
    }
    ll sum = 1;
    for (ll i = 2; i <= n - m; i++)
    {
        sum *= i;
        sum %= mod;
    }
    if (n < m)
    {
        puts("0");
    }
    else
    {
        cout << ans * pow_mod(sum, mod - 2) * pow_mod(n - m, l) << endl;
    }
}