【树形DP】 HDU 4118 Holiday's Accommodation

来源：互联网发布：安卓手机数据恢复系统编辑：程序博客网时间：2024/06/01 08:04

题目链接: HDU 4118 Holiday's Accommodation

分析: 可以知道每条边要走的次数刚好的是这条边两端的点数的最小值的两倍。

代码:

#include<iostream>#include<cstdio>#include<cstring>#include<stack>using namespace std;const int maxn=100000+10;struct node{    int to, dix, next;}tree[maxn<<1];int head[maxn],g[maxn],ptr;bool vis[maxn];void Init(){    ptr=1;     memset(vis,false,sizeof(vis));    memset(head,-1,sizeof(head));}void AddEdge(int a,int b,int c){    tree[ptr].to=b;    tree[ptr].dix=c;    tree[ptr].next=head[a];    head[a]=ptr++;}void DFS(){    vis[1]=true;    stack<int>M;    M.push(1);    int rt=head[1];    while(true){        if(rt==-1){            int a=M.top(); M.pop();            if(M.empty()) break;            g[M.top()]+=g[a];        }        rt=head[M.top()];        while(rt!=-1){            if(!vis[tree[rt].to]){                vis[tree[rt].to]=true;                M.push(tree[rt].to);                break;            }            rt=tree[rt].next;        }    }}int main(){    int T,cas=1;    scanf("%d",&T);    while(T--){        Init();        int n; scanf("%d",&n);        for(int i=1;i<n;++i){            int a,b,c;            scanf("%d%d%d",&a,&b,&c);            AddEdge(a,b,c);            AddEdge(b,a,c);            g[i]=1;        }        g[n]=1;        DFS();        __int64 ans=0;        for(int i=1;i<ptr;i+=2){            int m=min(g[tree[i].to],g[tree[i+1].to]);            ans+=2*min(n-m,m)*(__int64)tree[i].dix;        }        printf("Case #%d: %I64d\n",cas++,ans);    }    return 0;}