版权声明:仅供研究,转载请注明出处。 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;
}
}