HDU 4118 树形DP Holiday's Accommodation

最新推荐文章于 2019-11-05 00:42:39 发布

weixin_34408717

最新推荐文章于 2019-11-05 00:42:39 发布

阅读量75

点赞数

本文详细介绍了HDU4118 'Holiday's Accommodation' 的解题思路，通过分析题目特点，提出利用图论中的深度优先搜索（DFS）算法来解决实际问题的方法。代码实现简洁高效，展示了算法在实际问题解决中的应用。

题目链接: 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;
}