蓝书——287. 积蓄程度换根DP

首先：

树形DP常规套路——  f[i]：以i为根的子树，以i为源点，的最大流量

转移就是枚举i连接的点v的最大流量f[v]，肯定是这些流量与连接边的最小值，之和是f[i]的最大流量。

这样我们就可以求出根节点的f[1]，同时也是ans[1]，即整棵树以1为源点的最大流量。

如果暴力树形dp所有的就是n^2的复杂度，显然不行

于是我们可以用换根dp，其实就是自上而下递推，上面的dfs是自下而上推出所有的f

现在我们知道所有的f，和根节点的ans，要一边dfs求出所有点的ans

这个时候我们可以画张图，找出父亲节点和儿子节点u,v   f[u],f[v],ans[u],ans[v]的关系即可。

关系在代码中，自己画图推下就会了。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
//typedef __int128 LL;
//typedef unsigned long long ull;
//#define F first
//#define S second
typedef long double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef pair<ld,ld> pdd;
const ld PI=acos(-1);
const ld eps=1e-9;
//unordered_map<int,int>mp;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
//#define a(i,j) a[(i)*(m+2)+(j)]  //m是矩阵的列数
//pop_back()
const int seed=131;
const int M = 2e5+7;

int head[M],cnt;
void init()
{
		cnt=0,memset(head,0,sizeof(head));
		memset(f,0,sizeof(f));
  		memset(ans,0,sizeof(ans));
  		memset(du,0,sizeof(du));
}
struct EDGE{int to,nxt;ll val;}ee[M*2];
void add(int x,int y,ll z){ee[++cnt].nxt=head[x],ee[cnt].to=y,ee[cnt].val=z,head[x]=cnt;}
ll f[M];//以i为根的子树，以i为源点最大流量 
ll ans[M];//以i为源点的最大流量
int du[M];
void dfs1(int u,int fa)
{
//	cout<<u<<"  "<<fa<<endl;
	for(int i=head[u];i;i=ee[i].nxt)
	{
		int v=ee[i].to;
		ll w=ee[i].val;
	//	cout<<u<<"  "<<v<<endl;
		if(v==fa)continue;
		dfs1(v,u);
		if(du[v]==1)f[v]=w;
		f[u]+=min(f[v],w);
	}
}
void dfs2(int u,int fa)
{
	for(int i=head[u];i;i=ee[i].nxt)
	{
		int v=ee[i].to;
		ll w=ee[i].val;
		if(v==fa)continue;
		
		ans[v]=f[v]+min(ans[u]-min(f[v],w),w);
		dfs2(v,u);
	}
}
int main()
{
//	ios::sync_with_stdio(false);
//  	cin.tie(0);
  	int t;
  	scanf("%d",&t);
  	while(t--)
  	{
  		init();
  		
  		int u,v;
		ll val;
  		int n;
  		scanf("%d",&n);
  		for(int i=1;i<n;i++)
  		scanf("%d%d%lld",&u,&v,&val),add(u,v,val),add(v,u,val),du[u]++,du[v]++;
  	//	for(int i=2;i<=n;i++)if(du[i]==1)f[i]=1e10+7;
  		dfs1(1,0);ans[1]=f[1];
  		dfs2(1,0);
  		ll ma=0;
  		for(int i=1;i<=n;i++)ma=max(ans[i],ma);
		  //cout<<f[i]<<" "<<ans[i]<<" "<<i<<endl;
  		cout<<ma<<endl;	
	}
	return 0;
}