题目: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;
}