牛客国庆集训派对Day3 - H

题目描述

魔方国有n座城市，编号为。城市之间通过n-1条无向道路连接，形成一个树形结构。
澜澜打算在魔方国进行m次旅游，每次游览至少一座城市。为了方便，每次旅游游览的城市必须是连通的。此外，澜澜希望游览所有城市恰好一次。
澜澜想知道有多少种旅游方案满足条件，两个方案不同当且仅当存在某一次旅游游览了不同的城市。
澜澜不会数数，所以只好让你来帮他数方案。

输入描述:

第一行一个整数t表示数据组数 (1 ≤ t ≤ 100)。
每组数据第一行两个整数n,m ，接下来n-1行每行两个整数ai,bi表示一条道路 (1≤ ai,bi≤ n)。

输出描述:

每组数据输出一行一个整数表示方案数对109+7取模的结果。

示例1

输入

复制

2
3 1
1 2
1 3
3 2
1 2
1 3

输出

复制

1
4

思路：本来还以为是图论，其实是排列组合，答案是C(n-1,m-1)*m!

从n-1条边中选m-1条边断开（多模模……）

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cmath>
#include<map>
#include<set>
#include<string>
#include<cstring>
using namespace std;
#define ll long long
const int N=1000005,mod=1e9+7;
ll f[N];
void pre(){
    f[0]=1;f[1]=1;
    for(int i=2;i<N-2;i++){
        f[i]=(f[i-1]*i)%mod;
    }
}
ll mod_pow(ll a,ll b){
    ll res=1;
    while(b){
        if(b&1)res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}
int main(){
    int t,n,m,a,b;
    scanf("%d",&t);
    pre();
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<n;i++)scanf("%d%d",&a,&b);
        ll ans=(f[n-1]*mod_pow(f[n-m],mod-2)%mod*m)%mod;
        printf("%lld\n",ans);
    }
}