BZOJ4835 [Lydsy2017年4月月賽]遺忘之樹

f[x]表示分治樹以x爲根的子樹的情況數

f[x]就等於對於每個分治兒子y，f[y]*siz[y]的乘積

但是注意到如果有兩個重心他要求最小那個

所以如果有一個分治子樹大小恰好等於本身子樹大小/2，那麼對於這個子樹乘的就是f[y]*siz(y,x)，siz(y,x)代表y的分治子樹內標號比x大的點的個數

因爲分支結構的樹高是log的，所以sigma 子樹大小也是log的（因爲sigma 子樹大小=sigma 點深度），所以暴力求siz(y,x)即可

#include<iostream>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<cstdlib>
#include<cstdio>
#include<map>
#include<bitset>
#include<set>
#include<stack>
#include<vector>
#include<queue>
using namespace std;
#define MAXN 100010
#define MAXM 1010
#define ll long long
#define eps 1e-8
#define MOD 1000000007
#define INF 1000000000
struct vec{
	int to;
	int fro;
};
vec mp[MAXN];
int tai[MAXN],cnt;
bool cr[MAXN];
int n;
ll f[MAXN],siz[MAXN];
int mn[MAXN];
inline void be(int x,int y){
	mp[++cnt].to=y;
	mp[cnt].fro=tai[x];
	tai[x]=cnt;
}
int count(int x,int lim){
	int i,y;
	int re=0;
	if(x>lim){
		re++;
	}
	for(i=tai[x];i;i=mp[i].fro){
		y=mp[i].to;
		re+=count(y,lim);
	}
	return re;
}
void dfs(int x){
	int i,y;
	f[x]=siz[x]=1;
	mn[x]=x;
	for(i=tai[x];i;i=mp[i].fro){
		y=mp[i].to;
		dfs(y);
		siz[x]+=siz[y];
		mn[x]=min(mn[x],mn[y]);
	}
	for(i=tai[x];i;i=mp[i].fro){
		y=mp[i].to;
		if((siz[x]&1^1)&&(siz[y]==siz[x]>>1)){
			(f[x]*=f[y]*count(y,x)%MOD)%=MOD;
		}else{
			(f[x]*=f[y]*siz[y]%MOD)%=MOD;
		}
	}
}
int main(){
	int i,x,y;
	int tmp;
	scanf("%d",&tmp);
	while(tmp--){
		scanf("%d%*d",&n);
		cnt=0;
		for(i=1;i<=n;i++){
			tai[i]=0;
			cr[i]=1;
		}
		for(i=1;i<n;i++){
			scanf("%d%d",&x,&y);
			cr[y]=0;
			be(x,y);
		}
		for(i=1;i<=n;i++){
			if(cr[i]){
				dfs(i);
				printf("%lld\n",f[i]);
				break;
			}
		}
	}
	return 0;
}

/*
1
6 5
3 4
4 5
3 6
6 1
1 2
*/