比赛 2024暑假C班集训B 评测结果 AWWAAAWWWWWWWWWWWWWW
题目名称 赛道修建 最终得分 20
用户昵称 wzh0425 运行时间 0.206 s
代码语言 C++ 内存使用 3.90 MiB
提交时间 2024-07-11 09:28:43
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
int n,m,hd[100005],nxt[100005],ver[100005],vis[100005],val[100005],maxs,idx,flag;
void add(int x,int y,int z){
    ver[++idx]=y;
    val[idx]=z;
    nxt[idx]=hd[x];
    hd[x]=idx;
}
void dfs(int x,int sum){
    if (sum>maxs){
        maxs=sum;
        flag=x;
    }
    for (int i=hd[x];i;i=nxt[i]){
        int y=ver[i];
        int v=val[i];
        if (vis[y]==0){
            vis[y]=1;
            dfs(y,sum+v);
            vis[y]=0;
        }
    }
}
int main(){
    freopen("2018track.in","r",stdin);
    freopen("2018track.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (int i=2;i<=n;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    if (m==1){
        vis[1]=1;
        dfs(1,0);
        vis[1]=0;
        vis[flag]=1;
        dfs(flag,0);
        vis[flag]=0;
        cout<<maxs;
    }
    
    return 0; 
}