CodeForces-1109D. Sasha and Interesting Fact from Graph Theory（数论+组合数学）

题目：https://codeforces.com/contest/1109/problem/D

题意：给出n个点，由这些点连成一棵树，树中每一条边的权重介于[1,m]，给出两点a，b问有多少种树满足a，b之间的路径的权值之和为m

 分析：对于ab这条链来说，将m进行分割，利用插板法，将m分成m个1，进行插板，在m - 1 个空中插0,1,2，······ m - 1 个点，这也是权重分配方式，C(i - 1)/(m-1);这些点在除了ab两个端点外的点中选取所以还要乘C(i - 1)/(n - 2)；这些点还要全排列，乘上A(i - 1)/(i - 1)

剩下的点用cayley公式：

设F(n,m)表示n个点，形成m个有标号森林的方案数，F(n,m)=m*n^{n-1-m}

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
const int maxn = 1e6 + 5;
const int mod = 1e9 + 7;
LL fac[maxn];
LL inv[maxn];
LL quick_pow(LL a,LL b)
{
    a = a % mod;
    LL ans = 1;
    while(b)
    {
        if(b & 1) ans = ans * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ans;
}

void init()
{
    fac[0] = fac[1] = 1;
    for(int i = 2;i < maxn;i ++)
        fac[i] = fac[i - 1] * i % mod;
    inv[maxn - 1] = quick_pow(fac[maxn - 1],mod - 2);
    for(int i = maxn - 2;i >= 0;i --)
    {
        inv[i] = inv[i + 1] * (i + 1) % mod;
    }
}

inline LL C(int a,int b)
{
    return ((fac[a] * inv[a - b] % mod) * inv[b] % mod);
}

int main()
{
    int n,m;
    int a,b;
    cin >> n >> m >> a >> b;
    init();
    LL ans = 0;
    for(int i = 1;i <= m;i ++)
    {
        int vcnt = i + 1,ecnt = i;
        if(vcnt > n) break;
        if(vcnt == n) ans = (ans + C(m  - 1,ecnt - 1) * fac[vcnt - 2] % mod) % mod;
        else ans = (ans + C(n - 2,vcnt - 2) * C(m - 1,ecnt - 1) % mod * fac[vcnt - 2] % mod * quick_pow(m,n - 1 - ecnt) % mod * vcnt % mod * quick_pow(n,n - vcnt - 1)) % mod;
    }
    cout << ans << endl;
    return 0;
}