Sicily1024. Magic Island

这是一道深搜的题,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;
}

 

你可能感兴趣的:(Sicily1024. Magic Island)