第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛 H-哲哲的疑惑

第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛 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

说明

无可奉告

---

思路

答案显然是下式（枚举 i， 表示我用了 i 种颜色）：

（为了更好的阅读享受，以下内容使用LaTeX编写，由于第一次使用，折腾了好久）

回顾一下概率论与数理统计的知识：

$$\begin{eqnarray} 全排列公式&：&A_n^n = n! \\ 二项式系数&：&C_n^r = \binom{n}{r} = \cfrac{n!}{r! (n-r)!} \end{eqnarray}$$

$$\begin{eqnarray} ans &=& \sum_{i=0}^n \binom{n}{i} * \binom{l}{i} * i! * f(n-i) \\ &=& \sum_{i=0}^n \binom{n}{i} * f(n-i) * \sum_{j=0}^i \binom{i}{j} *(-1)^{i-j}*j^l \\ &=& \sum_{j=0}^n j^l * (-1)^j * \sum_{i=j}^n (-1)^i * \binom{n}{i} * \binom{i}{j} * f(n-i) \\ &=& \sum_{j=0}^n j^l * (-1)^j * \binom{n}{j} * \sum_{i=j}^n (-1)^i * \binom{n-j}{i-j} * f(n-i) \\ &=& \sum_{j=0}^n j^l * \binom{n}{j} * \sum_{i=0}^{n-j} (-1)^{i} * \binom{n-j}{i} *f(n-j-i) \\ &=& \sum_{j=0}^n j^l * \binom{n}{j} * \sum_{i=0}^{n-j}(-1)^{n-j-i} * \binom{n-j}{i} *f(i) \end{eqnarray}$$

$$\begin{eqnarray} &设：&f(i)=c_m * \binom{i}{m} + c_{m-1} * \binom{i}{m-1} +\ldots+c_0 * \binom{i}{0} \\ &\because& \sum_{i=0}^{n-j}(-1)^{n-j-i} * \binom{n-j}{i} *f(i) 是把f(i)差分k次后的f(0) \\ &\therefore& 原式=\sum_{j=0}^n j^l * \binom{n}{j} * c_{n-j} \\ &\because& 这里f(i)=\binom{i}{m} \\ &\therefore& 只有c_m=1 \\ &\therefore& 原式=（n-m）^l * \binom{n}{m} \end{eqnarray}$$

AC代码

注：使用原始快速幂可以AC。使用优化后的快速幂由于循环会报运行超时，case通过率为90.91%。

#include <iostream>
#include <array>
typedef long long ll;
using namespace std;
const int Mod = 998244353;
const int N = 30000050;
array<int, N> ni;

// 优化后的快速乘
ll QuickMulti_opt(ll a, ll b, ll c) {
    ll ret = 0;
    while(b) {
        if(b&1) {
            ret += a;
            if(ret >= c)
                ret -= c;
        }
        a <<= 1;
        if(a >= c)
            a -= c;
        b >>= 1;
    }
    return ret;
}
// 优化后的快速幂
ll QuickMod_opt(ll a, ll b, ll c) {
    ll ret = 1;
    while(b) {
        if(b&1) {
            ret = QuickMulti_opt(a,ret,c);
        }
        b >>= 1;
        a = QuickMulti_opt(a,a,c);
    }
    return ret;
}

// 原始的快速幂
ll QuickMod(ll a,ll b,ll c) {
    ll ans = 1;
    while(b) {
        if(b&1)
            ans = ans*a%c;
        b >>= 1;
        a = a*a%c;
    }
    return ans;
}

int main() {
    ll n,m,l;
    cin >> n >> m >> l;
    ll ans=1;
    for(int i=n-m+1; i<=n; i++) {
        ans=ans*i%Mod;
    }
    ni[1]=1;
    for(int i=2; i<=m; i++) {
        ni[i]=-1LL*(Mod/i)*ni[Mod%i]%Mod;
    }
    for (int i=1; i<=m; i++) {
        ans=ans*ni[i]%Mod;
    }
    ans=ans*QuickMod_opt(n-m,l,Mod)%Mod;
    cout << (ans+Mod)%Mod << endl;
    return 0;
}