LintCode:图是否是树

给出 n 个节点，标号分别从 0 到 n - 1 并且给出一个 无向 边的列表 (给出每条边的两个顶点), 写一个函数去判断这张｀无向｀图是否是一棵树

 注意事项

你可以假设我们不会给出重复的边在边的列表当中. 无向边　[0, 1]和 [1, 0]　是同一条边， 因此他们不会同时出现在我们给你的边的列表当中。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出n = 5 并且 edges = [[0, 1], [0, 2], [0, 3], [1, 4]], 返回 true.

给出n = 5 并且 edges = [[0, 1], [1, 2], [2, 3], [1, 3], [1, 4]], 返回 false.

标签 

深度优先搜索 脸谱网 Zenefits Union Find 宽度优先搜索谷歌

主要思想是采用堆栈的方式实现深度优先搜索

class Solution {
public:
    /**
     * @param n an integer
     * @param edges a list of undirected edges
     * @return true if it's a valid tree, or false
     */
    //没有环，没有孤立点的图就是树
    bool validTree(int n, vector<vector<int>>& edges) {
        // Write your code here
        if( n-1 > edges.size())//如果成立，说明有孤立点
           return false;
           
        if(edges.size()==0)    //只有一个头节点
           return true;
          
        stack<int> s;
        vector<bool> v(n, false);//用来标记点是否被遍历过
        
        int now;
        now = edges[0][0];
        v[now] = true;
             
        while(1)
        {
          int flag = 0;//如果for执行完毕flag还为0，说明已经到树末端
          for(auto it = edges.begin(); it != edges.end();)
          {
             if(now == (*it)[0])//找到了下一个节点
             {
                s.push(now);    //当前节点存入堆栈
                now = (*it)[1]; //下一个节点为当前节点
                if(v[now] == true)//如果这个节点已经被遍历过了
                   return false;  //说明有环，返回
                v[now] = true;    //如果没有遍历，则标记
                it = edges.erase(it);//删除上一条边，防止以后被遍历到
                flag = 1;         
                break;
             }
              if(now == (*it)[1])
             {
                s.push(now);
                now = (*it)[0];
                if(v[now] == true)
                   return false;
                v[now] = true;
                it = edges.erase(it);
                flag = 1;
                break;
             }
             it++;
          }
          
          if(edges.empty())//所有边都被遍历完毕且没有重复
              return true;
           
          if(s.empty())//边没有遍历完毕，堆栈空了，说明有孤立点
              return false;
              
          if(flag == 0)//遍历到末端，则从上一个点开始遍历
          {
              now = s.top();
              s.pop();
          }
        }
    }
};

这种方式复杂度还是高了，更好的办法是构造邻接图，然后用迭代的方式遍历

参照http://blog.csdn.net/waltonhuang/article/details/52103401