牛客网国庆集训派对Day3 H-Travel(计数问题)

牛客网国庆集训派对Day3 H-Travel(计数问题)_第1张图片

牛客网国庆集训派对Day3 H-Travel(计数问题)_第2张图片

思路来源

https://blog.csdn.net/Enterprise_/article/details/81462003

题解

因为这m次旅游互不相干,所以在这n-1条边里选m-1条边将其割断,将原树划分为m块即可。

总方案数考虑顺序,取这m块的全排列即可。

由于原图是一棵树,显然与怎么连无关,所以输入部分多余。

心得

学了卢卡斯定理之后,总想用卢卡斯,实际上用不到。

这里C(n,m)用原公式处理一下(n!)*((m!)*((n-m)!))^{-1}即可,

n,m1e5;初始化处理阶乘,费马小定理解决逆元。

注意中间结果及时模(1e9+7),最后答案乘以\left ( m! \right )

代码

#include 
#include  
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
const double INF=0x3f3f3f3f;
const int maxn=1e5+10; 
const int mod=1e9+7;
const int MOD=998244353;
const double eps=1e-7;
typedef long long ll;
#define vi vector 
#define si set
#define pii pair 
#define pi acos(-1.0)
#define pb push_back
#define mp make_pair
#define lowbit(x) (x&(-x))
#define sci(x) scanf("%d",&(x))
#define scll(x) scanf("%lld",&(x))
#define sclf(x) scanf("%lf",&(x))
#define pri(x) printf("%d",(x))
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define mem(a,b) memset(a,b,sizeof(a)) 
using namespace std;
ll jc[maxn];
void init()//预处理阶乘 
{
	jc[0]=1,jc[1]=1;
	rep(i,2,maxn)
	{
		jc[i]=jc[i-1]*i;
		if(jc[i]>=mod)jc[i]%=mod;
	} 
}
ll modpow(ll x,ll n)
{
	ll res=1;
	while(n)
	{
		if(n&1)res*=x;
		if(res>=mod)res%=mod;
		x*=x;
		if(x>=mod)x%=mod;
		n>>=1;
	}
	return res;
}
ll c(ll n,ll m)
{
	return jc[n]*modpow(jc[m]*jc[n-m]%mod,mod-2)%mod; 
}
int main()
{ 
    ll t,n,m;
    init();
    scll(t);
    while(t--)
    {
    	scll(n),scll(m);
    	int a,b;
    	rep(i,0,n-2)sci(a),sci(b);
    	printf("%lld\n",c(n-1,m-1)*jc[m]%mod);
    }
	return 0;
}

 

你可能感兴趣的:(牛客网国庆集训派对Day3 H-Travel(计数问题))