有一棵由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;
}
题目说了是双向边,所存图的时候要注意反向再存一次。