CSU - 1960 单源最大权路径

CSU - 1960 单源最大权路径
有一棵由N个结点构成的树,每一条边上都有其对应的权值。现在给定起点,求从该点出发的一条路径(至少有一条边)使得这条路径上的权值之和最大,并输出这个最大值。

Input
第一行一个正整数T,代表数据组数。每组数据第一行两个正整数n(2<=n<=10^5),s(1<=s<=n),分别表示树结点数目以及给定的起点,点的编号从1至N。接下来M行,每行三个整数x,y,z,(1<=x,y<=n,|z|<=1000),代表编号为x和y的点之间有一条权值为z的双向边。

Output
每组数据输出一行,即所找到路径的最大权值(格式参见样例)。

Sample Input
2
3 1
1 2 10
1 3 5
5 5
1 5 70
4 3 100
5 3 -10
2 5 60
Sample Output
Case #1: 10
Case #2: 90

题目大意:其实我有点小懵,因为我也不知道为什么M就等于N-1了…
给出n个点,n-1条边,这是双向边,走一条边要花费一定的权值。给出一个起点,边与其对应的权值信息,走出一条路径,求最长路径。这是一棵树,不会有环的
思路:DFS走啊走,把走到每一个点需要的权值都存起来,然后找到最大值即可。
开一个vector ve数组,就相当于二维数组了。第一维先找到出发点,第二维存该点下一步可以往哪走。

#include <iostream>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <cctype>
#include <cstdlib>
#include <vector>

using namespace std;
const int INT = 0x3f3f3f3f;
int T,n,s,x,y,z,ans;
int a[100010],vis[100010];

struct node{
    int v;
    int cost;
    node(int V = 0,int Cost = 0):v(V),cost(Cost){}
}; 
vector<node> ve[100010];
void dfs(int begin){
    vis[begin] = 1;
    for(int i = 0 ; i < ve[begin].size(); ++i){
        int v = ve[begin][i].v;
        int cost = ve[begin][i].cost;
        if(vis[v]) continue;
        a[v] = cost + a[begin];
        dfs(v);
    }
}
int main(){
    scanf("%d",&T);
    int c = 1;
    while(T--){
        memset(a,-INT,sizeof(a));
        memset(vis,0,sizeof(vis));
        scanf("%d%d",&n,&s);
        for(int i = 1 ; i <= n-1 ; ++i){
            scanf("%d%d%d",&x,&y,&z);
            ve[x].push_back(node(y,z));
            ve[y].push_back(node(x,z));
        }
        a[s] = 0;
        dfs(s);
        ans = -INT;
        for(int i = 1 ; i <= n ; ++i){
            ans = max(ans,a[i]);
        }
        printf("Case #%d: %d\n",c++,ans);

        for(int i = 0 ; i <= n ; ++i){
            ve[i].clear();
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Love_Yourself_LQM/article/details/81836544