【校内模拟】小B的班级（贪心）（组合数学）

简要题意：

给一棵 $n$ 个点的树，边有边权。现在有 $m$ 个A类点， $m$ 个B类点，两两不同，这 $2m$ 个点在树上选定位置 （可以有多个点在同一个位置），然后进行配对，最大化配对点的距离之和，问所有方案的距离之和是多少。

$n,m\le 2.5e3$

---

题解：

经典结论 （虽然我已经忘了上次见是在什么时候） ，直接考虑每条边被经过多少次，假设其中一边有 A 类点 $a$ 个，同侧有 B 类点 $b$ 个，则在最佳方案中这条边被经过次数为 $\min (a,m-b)+\min (b,m-a)=\min (a+b,2m-a-b)$。

所以次数只与某一侧点的总数有关，和点的种类无关。

枚举边，枚举某侧点数，分配方式直接组合数+预处理幂算一算即可。

复杂度 $O(nm)$

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

namespace IO{
	inline char gc(){
		static cs int Rlen=1<<22|1;static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++; 
	}template<typename T>T get_integer(){
		char c;bool f=false;while(!isdigit(c=gc()))f=c=='-';T x=c^48;
		while(isdigit(c=gc()))x=((x+(x<<2))<<1)+(c^48);return f?-x:x;
	}inline int gi(){return get_integer<int>();}
}using namespace IO;

using std::cerr;
using std::cout;

cs int mod=1e9+7;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
inline int dec(int a,int b){return a-b<0?a-b+mod:a-b;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}
inline int po(int a,int b){int r=1;for(;b;b>>=1,Mul(a,a))if(b&1)Mul(r,a);return r;}

cs int N=2.5e3+7,M=N<<1|1;

int n,m;

int el[N],nx[M],to[M],w[M],ec=1;
inline void adde(int u,int v,int vl){
	nx[++ec]=el[u],el[u]=ec,to[ec]=v,w[ec]=vl;
	nx[++ec]=el[v],el[v]=ec,to[ec]=u,w[ec]=vl;
}

int sz[N];
void dfs(int u,int p){
	for(int re e=el[u];e;e=nx[e])if(to[e]!=p)
		{dfs(to[e],u);sz[u]+=sz[to[e]];}++sz[u];
}

int C[M],inv[M];
void init_math(){
	inv[1]=1;for(int re i=2;i<=2*m;++i)inv[i]=mul(mod-mod/i,inv[mod%i]);
	C[0]=1;for(int re i=1;i<=2*m;++i)C[i]=mul(C[i-1],mul(inv[i],2*m-i+1));
}

int p1[M],p2[M],ans;
void Main(){
	n=gi(),m=gi();init_math();
	for(int re i=1;i<n;++i){
		int u=gi(),v=gi(),w=gi();
		adde(u,v,w);
	}dfs(1,0);p1[0]=p2[0]=1;
	for(int re e=3;e<=ec;e+=2)if(w[e]){
		int u=to[e],v=to[e^1],w=::w[e];
		if(sz[u]>sz[v])std::swap(u,v);
		for(int re i=1;i<=2*m;++i){
			p1[i]=mul(p1[i-1],sz[u]);
			p2[i]=mul(p2[i-1],n-sz[u]);
		}
		for(int i=1;i<=m;++i)
			Inc(ans,mul(C[i],mul(mul(p1[i],p2[2*m-i]),mul(i,w))));
		for(int re i=m+1;i<2*m;++i)
			Inc(ans,mul(C[i],mul(mul(p1[i],p2[2*m-i]),mul(2*m-i,w))));
	}cout<<ans<<"\n";
}

inline void file(){
#ifdef zxyoi
	freopen("class.in","r",stdin);
#else
#ifndef ONLINE_JUDGE
	freopen("class.in","r",stdin);
	freopen("class.out","w",stdout);
#endif
#endif
}
signed main(){file();Main();return 0;}