码迷,mamicode.com
首页 > 其他好文 > 详细

【POJ】1935 Journey(树形dp)

时间:2018-06-29 22:29:07      阅读:221      评论:0      收藏:0      [点我收藏+]

标签:continue   cto   algorithm   nbsp   include   col   using   ++   int   

题目

传送门:QWQ

 

 

分析

凉凉。

答案是所有要经过的点到根所经过的边权和减去最大的边权。

 

 

代码

vector好慢啊

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn=105000;
int d[maxn], fa[maxn];
struct Edge{int u,v,dis;};
vector<int> G[maxn];vector<Edge> edges;
int n, ans;
int val[maxn], vis[maxn];
void Addedge(int u,int v,int dis){
    edges.push_back((Edge){u,v,dis});edges.push_back((Edge){v,u,dis});
    int m=edges.size()-1;G[u].push_back(m-1);G[v].push_back(m);
}
int dfs(int x,int f,int depth){
    fa[x]=f; d[x]=depth;
    for(int i=0;i<G[x].size();i++){
        Edge v=edges[G[x][i]];
        if(v.v==f) continue;
        dfs(v.v,x,depth+v.dis);
        if(vis[v.v]) vis[x]=1,ans+=edges[G[x][i]].dis*2;
    }
}
int main(){
    int n,k,a,b,c;
    scanf("%d%d",&n,&k);
    for(int i=0;i<n-1;i++){
        scanf("%d%d%d",&a,&b,&c);
        Addedge(a,b,c);
    }
    int m; scanf("%d",&m);
    for(int i=0;i<m;i++){
        scanf("%d",&val[i]); vis[val[i]]=1;
    }
    dfs(k,0,0);
    int maxnum=0;
    for(int i=1;i<=n;i++){
        if(vis[i]) maxnum=max(maxnum,d[i]);
    }
    printf("%d",ans- maxnum);
    return 0;
}
/*
4 2
1 2 1
4 2 2
2 3 3
1
1
*/

 

 

【POJ】1935 Journey(树形dp)

标签:continue   cto   algorithm   nbsp   include   col   using   ++   int   

原文地址:https://www.cnblogs.com/noblex/p/9245476.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!