牛客练习赛76 C - CG的通关秘籍（组合数学）

传送门

---

题目大意

给出一个长度为$n$的空序列，有$m$种数即$[1,m]$填到这$n$个位置上，定义一个数$a_i$的下一个数$a_j$对整个序列的贡献为：

• 若$a_i=a_j$，贡献为$0$
• 若$a_i<a_j$，贡献为$1$
• 若$a_i>a_j$，贡献为$2$

求出所有方案的序列的贡献之和。

解题思路

#include <bits/stdc++.h>

using namespace std;
#define ENDL "\n"
typedef long long ll;
typedef pair<int, int> pii;
const int maxn = 1e5 + 10;
const int Mod = 1e9 + 7;

int t;

ll qkp(ll x, ll n, ll p) {
    ll ans = 1;
    x %= p;
    while (n) {
        if (n & 1) ans = ans * x % p;
        x = x * x % p;
        n >>= 1;
    }
    return ans;
}

ll inv(ll a, ll p) { return qkp(a, p - 2, p); }

int main() {
    // ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    ll n, m;
    scanf("%d", &t);
    while (t--) {
        scanf("%lld%lld", &n, &m);
        ll ans = qkp(m, n - 2, Mod);
        ans = ans * (n - 1) % Mod * m % Mod * (3 * m - 3) % Mod * inv(2, Mod) % Mod;
        printf("%lld\n", ans);
    }
    return 0;
}