DFS 单源最大权路径<vetor>和普通存图法

有一棵由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

这道题目可谓是经典的DFS了,题目给出了点点之间路径权值,要求我们求出最大的权路径。

我们首先按照题目要求把图存起来,在这里我用的是领接表法。从给出点出发,遍历整个图。

由于题目已经说了,这个图是树,所以也就不会有成环的情况,我们就可以用数组来存储权值。

我用了两种方法来进行邻接表存图,一种是vector,还有一种是手写的邻接表。

vector

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
int const size=1e+5;
const int INF=0x3f3f3f3f;
struct node
{
    int to;
    int v;
    node(int _to=0,int _v=0):to(_to),v(_v){}
};
int vis[size],cost[size];
vector<node> G[size];

int dfs(int id)
{
    vis[id]=1;
    for(int i=0;i<G[id].size();i++){
        int v=G[id][i].to;
        if(vis[v]) continue;
        cost[v]=cost[id]+G[id][i].v; 
        dfs(v);
       // vis[id]=0;
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    int cot=1;
    while(t--){
       int n,s;
       memset(vis,0,sizeof(vis));
       scanf("%d%d",&n,&s);
       for(int i=1;i<=n;i++) cost[i]=0;
       int i;
       node e,k;
       for(int i=1;i<=n;i++) G[i].clear();
       for(i=0;i<n-1;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        G[x].push_back(node(y,z));
        G[y].push_back(node(x,z));

       }
       cost[0]=0;
       dfs(s);
       int ans=-INF;

       for(i=1;i<=n;i++){

        ans=max(cost[i],ans);
       }
       printf("Case #%d: %d\n",cot++,ans);

    }
    return 0;
}

普通的

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>

using namespace std;
const int size=1e5+5;
const int INF=0x3f3f3f3f;
int first[size],nxt[size],flag[size];
struct node
{
    int b,e,v;
}edge[size];
int tot;
void Link_build(int id,int e,int val)
{
    edge[tot].b=id;
    edge[tot].e=e;
    edge[tot].v=val;
    nxt[tot]=first[id];
    first[id]=tot++;
}
int ans;
void dfs(int n,int vl)
{
    int id;
    if(first[n]==-1) return ;
    int val;
      for(int i=first[n];i!=-1;i=nxt[i])
      {
          if(!flag[edge[i].e])
          {
              id=edge[i].e;
              val=vl+edge[i].v;
              ans=max(ans,val);
              flag[edge[i].e]=1;
              dfs(id,val);
              flag[edge[i].e]=0;
          }
      }
}
int main()
{
    int T;
    scanf("%d",&T);
    int cot=1;
    while(T--)
    {
        tot=0;
        int start,n,s;
        memset(nxt,0,sizeof(nxt));
        memset(flag,0,sizeof(flag));
        memset(first,-1,sizeof(first));
        ans=-INF;
        scanf("%d%d",&n,&s);
        flag[s]=1;
        for(int i=0;i<n-1;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            Link_build(x,y,z);
            Link_build(y,x,z);
        }
        dfs(s,0);
        printf("Case #%d: ",cot++);
        cout<<ans<<endl;

    }
    return 0;
}

题目说了是双向边,所存图的时候要注意反向再存一次。

猜你喜欢

转载自blog.csdn.net/qq_40620465/article/details/81810958