Proving Equivalences （LA 4287）

#include<cstdio>
#include<cstring>
#include<stack>
#include<algorithm>
using namespace std;

struct my{
   int v;
   int next;
};

my bian[1000000+10];
stack<int>s;
int sccno[200000+10];
int adj[200000+10];
int lowlink[200000+10];
int pre[200000+10];
int out[200000+10];
int in[200000+10];
int fa;
int dfsnum;
int sccnum;
int m,n;

void myinsert(int u,int v){
     bian[++fa].v=v;
     bian[fa].next=adj[u];
     adj[u]=fa;
}

void init(){
     memset(bian,-1,sizeof(bian));
     memset(adj,-1,sizeof(adj));
     memset(lowlink,0,sizeof(lowlink));
     memset(sccno,0,sizeof(sccno));
     memset(pre,0,sizeof(pre));
     fa=0;
     dfsnum=0;
     sccnum=0;
     while(!s.empty()) s.pop();
}

void tarjan(int u){
     pre[u]=lowlink[u]=++dfsnum;
     s.push(u);
     for (int i=adj[u];i!=-1;i=bian[i].next){
        int v=bian[i].v;
        if(!pre[v]){
            tarjan(v);
            lowlink[u]=min(lowlink[u],lowlink[v]);
        }
        else if(!sccno[v]){
            lowlink[u]=min(lowlink[u],pre[v]);
        }
     }
     if(lowlink[u]==pre[u]){
        sccnum++;
        for(;;){
            int e=s.top();
            s.pop();
            sccno[e]=sccnum;
            if(e==u) break;
        }
     }
}
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
    init();
    int u,v;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        myinsert(u,v);
      }
     for (int i=1;i<=n;i++)
        if(!pre[i]) tarjan(i);
     for (int i=1;i<=sccnum;i++) in[i]=out[i]=1;
     for (int u=1;u<=n;u++){
        for (int j=adj[u];j!=-1;j=bian[j].next){
            int v=bian[j].v;
        if(sccno[v]!=sccno[u]) in[sccno[v]]=out[sccno[u]]=0;
        }
     }
     int ans,a=0,b=0;
     for (int i=1;i<=sccnum;i++){
        if(in[i]) a++;
        if(out[i]) b++;
     }
     ans=max(a,b);
     if(sccnum==1) ans=0;
     printf("%d\n",ans);
    }
return 0;
}