POJ P3585 Accumulation Degree

目录：

---

题目：

传送门

---

分析：

这道题可以用树形dp解决，以任意一个点为源点求出每个点的流量，再用二次扫描换根法找到最优的根
流量求法：

if (deg[e[i].y]==1) d[x]+=e[i].w; 
else d[x]+=min(d[e[i].y],e[i].w);//度为1就是边权，否则要综合子节点去考虑

动态规划：

if (deg[x]==1) f[e[i].y]=d[e[i].y]+e[i].w;//度只有1
else f[e[i].y]=d[e[i].y]+min(f[x]-min(d[e[i].y],e[i].w),e[i].w);//度有多个需要判断最小值，并且换根的差值也要算进去

---

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>  
#include<cstdlib>
#include<algorithm>
#include<set>
#include<vector>
#include<map>
#include<list>
#include<ctime>
#include<iomanip>
#include<string>
#include<bitset>
#define LL long long
#define h happy
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
int min(int x,int y)
{
    return x<y? x:y;
}
int max(int x,int y)
{
    return x>y? x:y;
}
struct edge{
    int next,to,w;
}e[400001];
int ls[400001];
int de[200001],cnt=0;
int d[200001],tf[200001],f[200001];
void dp(int k)
{
    tf[k]=1;
    d[k]=0;
    for(int i=ls[k];~i;i=e[i].next)
    {
        int y=e[i].to;
        if(tf[y]) continue;
        dp(y);
        if(de[y]==1) d[k]+=e[i].w;
        else d[k]+=min(d[y],e[i].w);
    }
    return;
}
void dfs(int k)
{
    tf[k]=1;
    for(int i=ls[k];~i;i=e[i].next)
    {
        int y=e[i].to;
        if(tf[y]) continue;
        if(de[k]==1) f[y]=d[y]+e[i].w;
        else f[y]=d[y]+min(f[k]-min(d[y],e[i].w),e[i].w);
        dfs(y);
    }
    return;
}
int main()
{
    int king=read();
    while(king--)
    {
        int n=read();
        int a,b,q;
        cnt=0;
        memset(de,0,sizeof(de));
        memset(d,0,sizeof(d));
        memset(ls,-1,sizeof(ls));
        for(int i=1;i<n;i++)
          {
            a=read();b=read();q=read();
            de[a]++;de[b]++;
            e[cnt]=(edge){ls[a],b,q};
            ls[a]=cnt++;
            e[cnt]=(edge){ls[b],a,q};
            ls[b]=cnt++;
          }
        if(n==0||n==1) {printf("0\n");continue;}
        memset(tf,0,sizeof(tf));
        dp(1);
        f[1]=d[1];
        memset(tf,0,sizeof(tf));
        dfs(1);
        int ans=0;
        for(int i=1;i<=n;i++) ans=max(ans,f[i]);
        printf("%d\n",ans);
    }
    return 0;
}