算法作业_44(2017.6.27第十九周)(算法机考模拟题4)

1004. 无环图

Description

 

在图论中，如果一个有向图从任意顶点出发无法经过若干条边回到该点，则这个图是一个有向无环图（Directed Acyclic Graph，DAG）. 对于一个n个节点的有向图（节点编号从0到n-1），请判断其是否为有向无环图.

 

图的节点数和边数均不多于100000.

 

请为下面的Solution类实现解决上述问题的isDAG函数，函数参数中n为图的节点数，edges是边集，edges[i]表示第i条边从edges[i].first指向edge[i].second. 如果是有向无环图返回true，否则返回false.

 

class Solution {

public:

       bool isDAG(int n, vector<pair<int, int>>& edges) {

          

    }

};

 

例1：

n = 3，edges = {(0, 1), (0, 2)}，函数应返回true.

 

例2：

n = 2，edges = {(0, 1), (1, 0)}，函数应返回false.

 

注意：你只需要提交Solution类的代码，你在本地可以编写main函数测试程序，但不需要提交main函数的代码. 注意不要修改类和函数的名称.

#include<iostream>
#include<vector>
using namespace std;

class Solution {
public:
       bool isDAG(int n, vector<pair<int, int>>& edges) {
          vector<vector<int>> g(n,vector<int>(0));
          vector<int> visit(n,0);
          for(auto a :edges){
              g[a.first].push_back(a.second);
          }
          for(int i = 0 ;i<n; i++){
              if(!DFS(g,visit,i)) return false;
          }
          return true ;
    }
    
private:
       bool DFS(vector<vector<int>> &g ,vector<int> &visit , int i ){
           if(visit[i] == -1) return false ;
           if(visit[i] == 1) return true;
           
           visit[i]= -1;
           for(auto a :g[i]){
               if(!DFS(g,visit,a)) return false;
           }
           visit[i] = 1;
           return true ;
       }
};