【DP】省选四校联考 画画图

---

分析：

这样建树的深度期望是$\sqrt{N}$的。

然后可以暴力dp了。

边权从小到大依次考虑贡献，比其小的算作-1，比其大的算作1，无非就是求这条边两侧边权和为0的连通子树方案数。

深度很小，所以暴力DP即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define SF scanf
#define PF printf
#define MAXN 32010
#define MAXP 610
#define ZERO 300
using namespace std;
typedef long long ll;
void Read(int &x){
	char c;
	while(c=getchar(),c!=EOF&&(c<'0'||c>'9'));
	x=c-'0';
	while(c=getchar(),c!=EOF&&c>='0'&&c<='9')
		x=x*10+c-'0';	
}
struct node{
	int u,v,val;
	bool operator <(const node &a) const{	
		if(val!=a.val)
			return val<a.val;
		return u<a.u;
	}
}edge[MAXN];
vector<int> a[MAXN],w[MAXN];
int fa[MAXN],dep[MAXN];
int dp[MAXN][MAXP];
ll totr;
int fai[MAXN];
void dfs(int x,int f){
	fa[x]=f;
	dp[x][ZERO]=1;
	for(int	i=0;i<int(a[x].size());i++){
		int u=a[x][i];
		if(u==f)
			continue;
		dfs(u,x);
		fai[u]=i;
		dep[x]=max(dep[x],dep[u]+1);
		for(int j=ZERO-dep[u];j<=ZERO+dep[u];j++)
			dp[x][j+w[x][i]]+=dp[u][j];
	}
}
void pushup(int x){
	int add=0;
	for(int y=x;fa[y];y=fa[y]){
		add+=w[fa[y]][fai[y]];
		for(int j=ZERO-dep[x];j<=ZERO+dep[x];j++)
			dp[fa[y]][j+add]-=dp[x][j];
	}
	w[fa[x]][fai[x]]=1;
	add=0;
	for(int y=x;fa[y];y=fa[y]){
		add+=w[fa[y]][fai[y]];
		for(int j=ZERO-dep[x];j<=ZERO+dep[x];j++)
			dp[fa[y]][j+add]+=dp[x][j];
	}
}
int upd[MAXP],fav[MAXN],L,R;
void add(int x,int d,int addv){
	for(int j=max(ZERO-dep[x],L-addv);j<=min(ZERO+dep[x],R-addv);j++)
		upd[j+addv]+=(dp[x][j]-dp[d][j-fav[d]]);
	if(fa[x])
		add(fa[x],x,addv+fav[x]);
}
int main(){
	int n,u,v,val;
	SF("%d",&n);
	for(int i=1;i<n;i++){
		Read(u),Read(v),Read(val);
		edge[i].u=u;
		edge[i].v=v;
		edge[i].val=val;
		a[u].push_back(v);
		a[v].push_back(u);
		w[u].push_back(-1);
		w[v].push_back(-1);
	}
	memset(fav,-1,sizeof fav);
	sort(edge+1,edge+n);
	dfs(1,0);
	ll ans=0;
	for(int i=1;i<n;i++){
		int u=edge[i].u,v=edge[i].v;
		if(fa[u]==v)
			swap(u,v);
		fav[v]=1;
		pushup(v);
//		for(int i=1;i<=n;i++){
//			PF("[%d]\n",i);
//			for(int j=ZERO-dep[i];j<=ZERO+dep[i];j++)
//				PF("{%d %d}\n",j-ZERO,dp[i][j]);
//		}
//		PF("-----------\n");
		memset(upd,0,sizeof upd);
		L=ZERO-dep[v];
		R=ZERO+dep[v];
		add(u,v,0);
		totr=0;
//		for(int j=ZERO-5;j<=ZERO+5;j++)
//			PF("--[%d %d]--\n",j-ZERO,upd[j]);
//		for(int j=ZERO-dep[v];j<=ZERO+dep[v];j++)
//			PF("++{%d %d}++\n",j-ZERO,dp[v][j]);
		for(int j=ZERO-dep[v];j<=ZERO+dep[v];j++)
			totr+=1ll*dp[v][j]*upd[2*ZERO-j];
		ans+=totr*edge[i].val;
//		PF("{{%d %d}%lld}\n",u,v,ans);
	}
	PF("%lld",ans);
}