这是一道深搜的题,WA了好多次,最后还是参考大牛的代码自己重编一次才AC的,惭愧惭愧。
#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
//用结构体保存每条道路的信息
struct road
{
int roadnum;
int end;
int dis;
};
vector<road> a[10005];
int visit[10005]; //标记某条路是否已经走过了
int maxdis=0;
void dfs(int k,int dis)
{
for(int i=0;i<a[k].size();i++)
{
if(visit[a[k][i].roadnum]==0)
{
visit[a[k][i].roadnum]=1;
dis+=a[k][i].dis;
if(dis>maxdis)
maxdis=dis;
dfs(a[k][i].end,dis); //这里递归访问从end开始的路
dis-=a[k][i].dis; //深搜到尽头的时候返回上一条边,返回到上一节点访问另外一条边时减去原来这条边的长度
visit[a[k][i].roadnum]=0;//同时把标记也抹掉
}
}
}
int main()
{
int n,k;
while(cin>>n>>k)
{
memset(a,0,sizeof(a));
memset(visit,0,sizeof(visit));
int begin,end,distance;
road r;
for(int i=0;i<n-1;i++)
{
cin>>begin>>end>>distance; //因为是双向边故两边的信息都要存储
r.roadnum=i;
r.dis=distance;
r.end=end;
a[begin].push_back(r);
r.end=begin; //修改起点和终点
a[end].push_back(r);
}
maxdis=0;
dfs(k,0);
cout<<maxdis<<endl;
}
return 0;
}