单源最大权路径 CSU - 1960(DFS)

版权声明:仅供研究,转载请注明出处。 https://blog.csdn.net/CSUstudent007/article/details/82556025

 思路:由于n的数据上限为1e5,不能直接用二维数组存边的权值(x,y分别做第一维和第二维)。

所以把子节点写入node里,用vector存。

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#define INF 0x3f3f3f3f
#define maxn 100050
using namespace std;
struct node{
	int v,w;
	node(int v1,int w1):v(v1),w(w1){}
};
vector<node> G[maxn];
int visit[maxn];
int d[maxn];
void dfs(int v){
	visit[v]=1;
	for(int i=0;i<G[v].size();i++){
	int c=G[v][i].w;
	if(visit[G[v][i].v]==0){
		d[G[v][i].v]=d[v]+c;
		dfs(G[v][i].v);
	}
}
}
int main(){
	int T,n,s,x,y,z;
	cin>>T;
	for(int k=1;k<=T;k++){
		memset(visit,0,sizeof(visit));
		cin>>n>>s;
		for(int i=0;i<n-1;i++){
			cin>>x>>y>>z;
			G[x].push_back(node(y,z));
			G[y].push_back(node(x,z));
		}
		for(int i=0;i<maxn;i++){
			d[i]=-INF; 
		}
		d[s]=0;
		dfs(s);
		int ans=-INF;
		for(int i=1;i<=n;i++){
			ans=max(ans,d[i]);
		}
		for(int i=1;i<=n;i++){
			G[i].clear();
		}
		cout<<"Case #"<<k<<":"<<" "<<ans<<endl;
	}
} 

猜你喜欢

转载自blog.csdn.net/CSUstudent007/article/details/82556025