HDU 3342 Legal or Not 拓扑排序

http://acm.hdu.edu.cn/showproblem.php?pid=3342

题意：

　　输入N,M代表有0~N-1的点，M组数据 ，输入的每组数据代表 a指向b ，然后用这M组数据构成一个有向图，判断这个

图有没有环，有就NO，没有就YES。

 

坑爹：

　　用了vector开的数组，如果中途发现  k == -1 的时候，有很多节点没有删除，所以要在每一次拓扑完之后将map数组

清空。

 

解法：

　　利用拓扑排序，每次找入度为0的点，然后把与他相连的边清除掉，并且将与他相连的点的入度减1。

 

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

const int maxn = 500 + 10;
int IN[maxn];
int used[maxn];
vector <int> map[maxn];
int N;

int find_zero()
{
    int i;
    for(i=0; i<N; i++)
    {
        if(!used[i] && IN[i]==0)
        {
            return i;
        }
    }
    return -1;
}

void cut(int k)
{
    int i;
    for(i=0; i<map[k].size(); i++)
    {
        IN[map[k][i]]--;
    }
    map[k].clear();
}

void topology()
{
    int i;
    int k=0;
    int flag=0;
    for(i=0; i<N; i++)
    {
        k=find_zero();
        if(k==-1)
        {
            flag=1;
            cout<<"NO"<<endl;
            break;
        }
        cut(k);
        used[k]=1;
    }
    if(!flag)
    {
        cout<<"YES"<<endl;
    }
}

int main()
{
    int M;
    while(cin>>N>>M,N)
    {
        memset(used,0,sizeof(used));
        memset(IN,0,sizeof(IN));
        int i;
        
        for(i=0; i<M; i++)
        {
            int a;
            int b;
            cin>>a>>b; 
            map[a].push_back(b);
            IN[b]++;
        }
        topology();
        for(i=0; i<N; i++)
        {
            map[i].clear();
        }
    }
    return 0;
}