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

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

cpp 复制代码
#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;
}
相关推荐
浒畔居2 分钟前
C++中的状态模式实战
开发语言·c++·算法
naruto_lnq5 分钟前
C++中的状态模式
开发语言·c++·算法
近津薪荼1 小时前
优选算法——双指针4(单调性)
c++·学习·算法
IUGEI2 小时前
从原理到落地:DAG在大数据SLA中的应用
java·大数据·数据结构·后端·算法
云深麋鹿2 小时前
五.排序笔记
c语言·数据结构·算法·排序算法
spcier8 小时前
图论拓扑排序-Kahn 算法
算法·图论
知星小度S8 小时前
动态规划(一)——思想入门
算法·动态规划
ysa0510308 小时前
动态规划-逆向
c++·笔记·算法
燃于AC之乐8 小时前
我的算法修炼之路--7—— 手撕多重背包、贪心+差分,DFS,从数学建模到路径DP
c++·算法·数学建模·深度优先·动态规划(多重背包)·贪心 + 差分
闻缺陷则喜何志丹8 小时前
【BFS 动态规划】P12382 [蓝桥杯 2023 省 Python B] 树上选点|普及+
c++·蓝桥杯·动态规划·宽度优先·洛谷