POJ 2553 The Bottom of a Graph

#include <stdio.h>
#include <string.h>
#define MAXN 5001

int m;
int cnt;
struct node{
    int to,next;
}edge[MAXN*MAXN];

int temp[MAXN];
int head[MAXN];
int dfn[MAXN];
int low[MAXN];
int sta[MAXN];
int flag[MAXN];

void addedge(int u, int v){
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}

int min(int a, int b){
    if(a<b)return a;
    else return b;
}

int tarbfs(int k ,int lay, int& scc_num){
    temp[k]=1;
    low[k]=lay;
    dfn[k]=lay;
    sta[++m]=k;
    for(int i=head[k]; i!=-1; i=edge[i].next){
        if(temp[edge[i].to]==0){
            tarbfs(edge[i].to,++lay,scc_num);
        }
        if(temp[edge[i].to]==1)low[k]=min(low[k],low[edge[i].to]);
    }
    if(dfn[k]==low[k]){
        ++scc_num;
        do{
            low[sta[m]]=scc_num;
            temp[sta[m]]=2;
        }while(sta[m--]!=k);
    }
    return 0;
}

int tarjan(int n){
    int scc_num=0;
    int lay=1;
    m=0;
    memset(temp,0,sizeof(temp));
    memset(low,0,sizeof(low));
    for(int i=1; i<=n; i++){
        if(temp[i]==0)tarbfs(i,lay,scc_num);
    }
    return scc_num;
}

int main(int argc, char *argv[])
{
    int n,t,v,u;
    cnt=0;
    while(scanf("%d",&n),n){
        scanf("%d",&t);
        memset(head,-1,sizeof(head));
        memset(flag,0,sizeof(flag));
        for(int i=0; i<t; i++){
            scanf("%d %d",&v,&u);
            addedge(v,u);
        }
        tarjan(n);
        for(int i=1; i<=n; i++){
            for(int j=head[i]; j!=-1; j=edge[j].next){
                if(low[edge[j].to]!=low[i]){
                    flag[low[i]]=1;
                    break;
                }
            }
        }
        for(int i=1; i<=n; i++){
            if(!flag[low[i]]){
                if(i!=n){
                    printf("%d ",i);
                }else{
                    printf("%d",i);
                }
            }
        }
        printf("\n");
    }
    return 0;
}