[机房练习赛4.3]graph

Problem 3. graph

Input file: graph.in
Output file: graph.out
Time limit: 1 second
jyb 给大家讲过强连通分量，强连通分量中的任意两点之间都可以互相到达。这个条件感觉很苛刻，大部分图
都不能满足。现在jyb 告诉你一个新的概念：单向连通图；如果有向图中，对于任意节点v1 和v2，至少存
在从v1 到v2 和从v2 到v1 的路径中的一条，则为单向连通图。现在给出若干个有向图，jyb 想问你它们是
不是单向连通图。
Input
第1 行，1 个整数T, 表示数据组数，对于每组数据：
第1 行，2 个整数n;m，表示点数和边数
接下来m 行，每行2 个整数u,v, 表示u 到v 有一条单向边。题目保证u! = v
Output
对于每组数据，如果是则输出”Yes”, 不是则输出”No”（均不含引号）
Sample
graph.in
2
3 2
1 3
2 3
3 2
1 2
2 3
graph.out
No
Yes
Note
• 对于30% 的数据,1 n 100，1 m n2
• 对于100% 的数据,1 T 5,1 n 30000，1 m 2 105。

tarjan求SCC+拓扑排序
经过tarjan缩点后，得到了一个DAG，我们很容易想到如果一个图“分叉”，那么分叉上的点肯定是互不可达的，所以原图必须是“一条链”。为什么对链打引号呢？是因为这条链不用很严格，比如1->2 2->3 3->4 1->4 2->4这样的图也是满足的，且可能还有重边，所以只用入度出度去判断的方法容易出错，dfs去走一条链的办法也比较麻烦（我没想出有什么简单的办法）。我们利用拓扑排序，若一直都是一个入度为0的点（队中至多有一个点），那么就是满足题意的。（by JYB）

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 100010;
const int maxm = 400010;
struct edge{
    int v,next;
}e[maxm];
int h[maxn],num,hh[maxn];
int indexx,top,stack[maxn],belong[maxn],cnt;
int p[maxn],dfn[maxn],low[maxn];
int q[maxn],head,tail;
bool vis[maxn];
int u,v,n,m;
int T;
void dfs( int u ){
    indexx++; dfn[u] = indexx; low[u] = indexx; vis[u] = true; stack[++top] = u;int v;
    for( int i = h[u]; i; i = e[i].next ){
        v = e[i].v;
        if( !dfn[v] ){
            dfs(v);
            if( low[v] < low[u] ) low[u] = low[v];
        } else if( vis[v] && dfn[v] < low[u] ) low[u] = dfn[v];
    }if( dfn[u] == low[u] ){
        cnt++;
        while(1){
            int x = stack[top];
            vis[x] = false;
            belong[x] = cnt;top--;
            if( x == u ) break;
        }
    }
}
void  adde(int u,int v) {num++;e[num].v = v;e[num].next = h[u];h[u] = num;}
void  adde2(int u,int v){num++;e[num].v = v;e[num].next = hh[u];hh[u] = num;}
int main() {
    freopen("graph.in","r",stdin);
    freopen("graph.out","w",stdout);
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &m);
        num = cnt = indexx = 0;
        memset(h,0,sizeof(h));memset(hh,0,sizeof(hh));
        while(m--){
            scanf("%d%d", &u, &v);
            adde( u, v );
        }
        memset(dfn,0,sizeof(dfn));memset(low,0,sizeof(low));memset(p,0,sizeof(p));
        for( int i = 1; i <= n; i++ ) if( !dfn[i] ) dfs(i);
        for( int i = 1; i <= n; i++ )
            for( int j = h[i]; j; j = e[j].next)
                if( belong[i] != belong[e[j].v] ){
                    p[belong[e[j].v]]++;
                    adde2(belong[i],belong[e[j].v]);
                }
        bool ans = true; head = tail = 0;
        for( int i = 1; i <= cnt; i++ ) if(!p[i]) q[tail++] = i;
        while( head < tail ){
            if( tail - head > 1 ){ans = false;break;} 
            u = q[head];
            for( int i = hh[u]; i; i = e[i].next){
                v = e[i].v; p[v]--;
                if( p[v] == 0 ) q[tail++] = v;
            }
            head++;
        }
        if( ans )printf("Yes\n");else printf("No\n"); 
    }
    return 0;
}