Codeforces 1540B Tree Array(lca+概率dp)

题目链接：B. Tree Array

题意：给定一颗树，第一步会等概率的选一个点当作根，之后每步会在当前已选的点的可达点中等概率的选某个点作为下一个点，按选点顺序生成一个数组，问数组中逆序对的期望值

题解：我们可以先枚举根，在根确定的情况下，枚举每对点成为逆序对的概率，相加就是所求期望，考虑根确定情况下每对点成为逆序对的概率，若点i，j，在当前根确定情况下是父子关系，那么肯定是先选了父亲才能选儿子，所以能否构成逆序对是确定的（概率为1/0，根据父亲下标是否大于儿子觉得），若二者不是父子关系，因为在选到i，j之前所都必须先选lca（i，j），所以在选到lca（i，j）前的情况都可以忽略，只考虑从lca（i，j）到i，j过程中谁先被选到，通过求lca我们可以知道从lca（i，j）到选到i，j,分别要往两边走多少步，因为选取所有点都是等概率的，所以相对于i，j二点，被选的概率都是1/2，这时只需要预处理出个数组dp[i][j]表示左边走i步，右边走j步时先走到i的概率即可，递推方程dp[i][j]=(dp[i-1][j]+dp[i][j-1])*2^(mod-2)，表示第一步走的分别是左边和右边带来的贡献(倒着往回推) dp[0][j]赋为1，因为此时必定先走完i，并且下一步只能走j

#include<iostream>
#include<stack>
#include<list>
#include<set>
#include<vector>
#include<algorithm>
#include<math.h>
#include<numeric>
#include<map>
#include<cstring>
#include<queue>
#include<iomanip>
#include<cmath>
#include<queue>
#include <bitset>
#include<unordered_map>/*
	#ifndef local
	#define endl '\n'
#endif */
#define mkp make_pair
using namespace std;
using std::bitset;
typedef long long ll;
typedef long double ld;
const int inf=0x3f3f3f3f;
const ll MAXN=2e3+10;
const ll N=2e3+100;
const ll mod=1e9+7;
const ll hash_p1=1610612741;
const ll hash_p2=805306457;
const ll hash_p3=402653189;
//-----------------------------------------------------------------------------------------------------------------*/
// ll head[MAXN],net[MAXN],to[MAXN],edge[MAXN]/*流量*/,cost[MAXN]//费用;
/* 
void add(ll u,ll v,ll w,ll s){
	to[++cnt]=v;net[cnt]=head[u];edge[cnt]=w;cost[cnt]=s;head[u]=cnt;
	to[++cnt]=u;net[cnt]=head[v];edge[cnt]=0;cost[cnt]=-s;head[v]=cnt;
}
struct elemt{
	int p,v;
};
-----------------------------------
求[1,MAXN]组合式和逆元 
ll mi(ll a,ll b){
	ll res=1;
	while(b){
		if(b%2){
			res=res*a%mod;
		}	
		a=a*a%mod;
		b/=2;
	}
	return res;
}
---------------------------------
 unordered_map<int,int>mp;
//优先队列默认小顶堆 ， greater<int> --小顶堆  less<int> --大顶堆  
priority_queue<elemt,vector<elemt>,comp>q;
struct comp{
	public:
		bool operator()(elemt v1,elemt v2){
			return v1.v<v2.v;
		}
};
	set<int>::iterator it=st.begin();
*/
//emplace_back()  等于push_back(),但效率更高,传输pair时emplace_back(i,j)==push_back({i,j}) 
// vector<vector<int>>edge; 二维虚拟储存坐标 
//-----------------------------------------------------------------------------------------------------------------*/
  //map<int,bool>mp[N]; 
  //emplace_back()
int head[MAXN],net[MAXN],to[MAXN];
int cnt; 
void add(int u,int v){ //双向建边
	to[++cnt]=v;net[cnt]=head[u];head[u]=cnt;
	to[++cnt]=u;net[cnt]=head[v];head[v]=cnt;
}
ll mi(ll a,ll b){
	ll res=1;
	while(b){
		if(b%2){
			res=res*a%mod;
		}
		a=a*a%mod;
		b/=2;
	}
	return res;
}
//-----------------------------------------------倍增求lca 
ll f[N][22];
ll dep[N];//深度
ll dp[210][210];//左边走i步，右边走j步左边先走完的概率
void dfs(int x,int pre){
	dep[x]=dep[pre]+1;
	f[x][0]=pre;
	for(int i=1;i<20;i++){
		f[x][i]=f[f[x][i-1]][i-1];//倍增更新 
	}
	for(int i=head[x];i;i=net[i]){
		int v=to[i];
		if(v==pre){
			continue;
		}
		dfs(v,x);
	}
} 
int lca(int x,int y){
	if(dep[x]<dep[y]){
		swap(x,y);
	}
	for(int i=19;i>=0;i--){//使x,y,达到同一深度 
		if(dep[f[x][i]]>=dep[y]){
			x=f[x][i];
		}
	}
	if(x==y){//此时xy重合 
		return x;
	}
	for(int i=19;i>=0;i--){//因为x，y现在深度相同，所以可以倍增的查找lca 
		if(f[x][i]!=f[y][i]){
			x=f[x][i];y=f[y][i];
		}
	}
	return f[x][0];
}
int n;
ll solve(int k){//以k为根时的贡献
	memset(dep,0,sizeof(dep));
	memset(f,0,sizeof(f));
	dfs(k,0);//lca预处理
	ll res=0;
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++){
			int tmp=lca(i,j);
			if(tmp==j){
				res++;//如果j是i的父亲，那么j肯定排在i前面，并且j>i，形成一堆逆序对
			}
			else if(tmp!=i&&tmp!=j){
				ll lx=dep[i]-dep[tmp],ly=dep[j]-dep[tmp];//从lca往下走多少次才能分别到i，j
					res+=dp[ly][lx];
					res%=mod;
			}
		}
	}
	return res*mi(n,mod-2)%mod;//选k为根的概率为1/n
}
int main(){
/*cout<<setiosflags(ios::fixed)<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（不含整数部分）*/
/*cout<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（含整数部分）*/
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);//同步流
	cin>>n;
	for(int j=1;j<=n;j++){
		dp[0][j]=1;
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			dp[i][j]=((dp[i-1][j]+dp[i][j-1])%mod)*mi(2,mod-2)%mod;//预处理出dp
		}
	}
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		add(u,v);
	}
	ll ans=0;//存答案
	for(int i=1;i<=n;i++){
		ans+=solve(i);
		ans%=mod;
	}
	cout<<ans<<endl;
	return 0;
}