动态规划（算法竞赛、蓝桥杯）--树形DP积蓄程度

最新推荐文章于 2024-09-11 00:16:09 发布

仁公智能

最新推荐文章于 2024-09-11 00:16:09 发布

阅读量459

点赞数 11

分类专栏：动态规划Dynamic Programming 文章标签：算法动态规划蓝桥杯

本文链接：https://blog.csdn.net/lmessi10_/article/details/136482029

版权

动态规划Dynamic Programming 专栏收录该内容

38 篇文章 0 订阅

订阅专栏

1、B站视频链接：E35 树形DP 积蓄程度_哔哩哔哩_bilibili

#include <bits/stdc++.h> 
using namespace std;
const int N=200010,INF=0x3f3f3f3f;
int h[N],to[N*2],w[N*2],ne[N*2],tot;  //邻接表
int d[N];   //d[i]记录从i点向下流出的最大流量
int f[N];   //f[i]记录从i点向外流出的最大流量
int deg[N]; //deg[i]记录点i的度数

void add(int a, int b, int c){
  to[++tot]=b,w[tot]=c,ne[tot]=h[a],h[a]=tot;
}
int dfs_d(int u, int fa){ //从叶点向上递推，获取u点的下流量 
  for(int i=h[u];i;i=ne[i]){
    int v=to[i];
    if(v==fa) continue;   //避免向上查找 
    int s=dfs_d(v,u);     //返回v点的下流量 
    d[u]+=min(w[i], s);   //累加u点的下流量 
  } 
  if(deg[u]==1) return INF; //特判叶的情况 
  return d[u];              //返回u点的下流量 
}
void dfs_f(int u, int fa){ //从根点向下递推，获取v点的全流量
  for(int i=h[u];i;i=ne[i]){
    int v=to[i];
    if(v==fa) continue;       
    if(deg[u]==1) f[v]=d[v]+w[i]; //特判根的情况 
    else f[v]=d[v]+min(w[i],f[u]-min(w[i],d[v])); 
    dfs_f(v, u);
  }
}
int main(){
  int t, n;
  scanf("%d", &t);
  while(t--){
    scanf("%d", &n);
    tot = 0;
    memset(h, 0, sizeof h);
    memset(d, 0, sizeof d);  
    memset(f, 0, sizeof f);     
    memset(deg, 0, sizeof deg);
    for(int i=1; i<n; i++){
      int a, b, c;
      scanf("%d%d%d", &a, &b, &c);
      add(a, b, c), add(b, a, c);
      deg[a]++, deg[b]++;
    }
    
    dfs_d(1,-1);  //向上递推下流量 
    f[1]=d[1];    //根只有向下的流量 
    dfs_f(1,-1);  //向下递推全流量 
    int ans=0;
    for(int i=1;i<=n;i++) ans=max(ans,f[i]);
    printf("%d\n",ans);
  }
  return 0;
}