【POJ3585】Accumulation Degree 二次扫描与换根法

简单来说，这是一道树形结构上的最大流问题。

朴素的解法是可以以每个节点为源点，单独进行一次dp，时间复杂度是\(O(n^2)\)
但是在朴素求解的过程中，相当于每次都求解了一次整棵树的信息，会做了不少的重复工作。
对于一棵子树的孩子节点和根节点之间存在着最优解的某些关联，因此可以采用自顶向下的一次 dfs 遍历求得结果。

阶段：子树大小
状态：当前子树大小的情况下，最大的流量是多少
状态转移方程：见代码

代码如下

#include <cstdio>
#include <vector>
#include <memory.h>
using namespace std;
const int maxn=2e5+10;

int n,rt,d[maxn],f[maxn],deg[maxn];
struct node{
    int to,w;
    node(int x=0,int y=0):to(x),w(y){}
};
vector<node> G[maxn];

#define cls(a,b) memset(a,b,sizeof(a))
void init(){
    cls(d,0);cls(f,0);cls(deg,0);
    for(int i=1;i<=2e5;i++)G[i].clear();
}

inline void add_edge(int from,int to,int w){
    G[from].push_back(node(to,w)),++deg[from];
    G[to].push_back(node(from,w)),++deg[to];
}

void read_and_parse(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int from,to,w;
        scanf("%d%d%d",&from,&to,&w);
        add_edge(from,to,w);
    }
}

void dfs1(int u,int fa){
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i].to,w=G[u][i].w;
        if(v==fa)continue;
        dfs1(v,u);
        if(deg[v]==1)d[u]+=w;
        else d[u]+=min(w,d[v]);
    }
}

void dfs2(int u,int fa){
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i].to,w=G[u][i].w;
        if(v==fa)continue;
        if(deg[u]==1)f[v]=d[v]+w;
        else f[v]=d[v]+min(f[u]-min(w,d[v]),w);
        dfs2(v,u);
    }
}

void solve(){
    rt=1;
    dfs1(rt,0);
    f[rt]=d[rt];//自顶向下需要初始化
    dfs2(rt,0);
    
    int ans=0;
    for(int i=1;i<=n;i++)ans=max(ans,f[i]);
    printf("%d\n",ans);
}

int main(){
    int T;scanf("%d",&T);
    while(T--){
        init();
        read_and_parse();
        solve();
    }
    return 0;
}

转载于:https://www.cnblogs.com/wzj-xhjbk/p/9871141.html