1021 Deepest Root

题目大意：

给出n个节点与N-1条边，问：它们能否形成一棵N个节点的树？如果能，则从中选出节点作为树根，使得整棵树的高度最大。输出所有满足要求的可以作为树根的节点，

解题思路：

DFS深搜即可，走过的点都标记上，如果一轮搜索下来还有点没有访问过，证明它不是一棵树。
代码如下：

#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set<int>::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int n,book[10010],mi=-1;
vector<int> v[10100];//邻接表
vector<int> result;
void dfs(int root,int len)
{
  if(v[root].size()==1&&book[v[root][0]])//到达叶节点
  {
    book[v[root][0]]=1;
    if(len>mi)mi=len;//更新最长路径长度
    return;
  }
  for(int i=0;i<v[root].size();i++)
  {
    if(!book[v[root][i]])
    {
      book[v[root][i]]=1;
      len++;
      dfs(v[root][i],len);
      len--;
    }
  }
  return;
}
int main()
{
//  std::ios::sync_with_stdio(false);
//  cin.tie(0);
//  freopen("test.txt","r",stdin);
//  freopen("output.txt","w",stdout);
 scanf("%d",&n);
 if(n==1)//特例，只有根节点的树
 {
   cout<<1<<endl;
   return 0;
 }
 for(int i=1;i<n;i++)
 {
   int a,b;
   scanf("%d %d",&a,&b);
   v[a].push_back(b);v[b].push_back(a);
 }
 int k=0;//统计树的数量
 for(int i=1;i<=n;i++)
 {
   if(!book[i])
   {
     book[i]=1;
     dfs(i,1);//搜索树
     k++;
   }
 }
 if(k>1)cout<<"Error: "<<k<<" components"<<endl;
 else
 {
   int len=0;
   for(int i=1;i<=n;i++)
   {
     mem0(book);
     mi=-1;
     book[i]=1;
     dfs(i,1);
     if(mi>len)
     {
       len=mi;
       result.clear();
       result.push_back(i);
     }
     else if(mi==len)result.push_back(i);
   }
   sort(result.begin(),result.end());//升序
   for(int i=0;i<result.size();i++)cout<<result[i]<<endl;
 }
 return 0;
}