剛學點分治,簡要的記錄一下

每次選擇一個點把這棵樹分成幾個不相交的子樹

挑一個使得最大子樹size最小的點

計算過這個點的答案

遞迴處理這幾個子樹

Code:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=40005;
struct edge{int u,v,w;};
vector<edge>G[maxn];
int n,m,k,root,All,ans=0;
int siz[maxn],f[maxn],d[maxn],data[maxn];
bool cant[maxn];
void makert(int u,int fa){
	siz[u]=1;f[u]=0;
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];
		if(fa==e.v||cant[e.v])continue;
		makert(e.v,u);
		siz[u]+=siz[e.v];
		f[u]=max(f[u],siz[e.v]);
	}f[u]=max(f[u],All-siz[u]);
	if(f[root]>f[u])root=u;
}
void dfs(int u,int fa){
	data[++data[0]]=d[u];
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];
		if(e.v!=fa&&!cant[e.v]){
			d[e.v]=d[u]+e.w;
			dfs(e.v,u);			
		}
	}
}
int calc(int u){
	data[0]=0;dfs(u,0);
	int ans=0,l=1,r=data[0];
	sort(data+1,data+1+data[0]);
	while(l<r){
		if(data[l]+data[r]<=k)ans+=(r-l),l++;
		else r--;
	}return ans;
}
void solve(int u){
	d[u]=0;ans+=calc(u);cant[u]=1;
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];if(cant[e.v])continue;
		d[e.v]=e.w;ans-=calc(e.v);
		f[root=0]=n+1;
		All=siz[e.v];makert(e.v,0);
		solve(root);	
	}
}
int main(){
	scanf("%d",&n);All=n;
	for(int i=1;i<n;i++){
		int u,v,w;scanf("%d%d%d",&u,&v,&w);
		G[u].push_back((edge){u,v,w});
		G[v].push_back((edge){v,u,w});
	}scanf("%d",&k);
	f[root]=n+1;
	makert(1,0);
	solve(root);
	cout<<ans<<endl;
	return 0;
}