UVa 11504 - Dominos（缩点）

缩点之后dfs求连通块个数，水题

#include <stdio.h>
#include <string.h>
#include <vector>
#include <stack>
using namespace std;
#define maxn 100005
vector<int> G[maxn];
vector<int> H[maxn];
stack<int> S;
int fa[maxn];
int pre[maxn],lowlink[maxn],sccno[maxn],dfs_clock,scc_cnt;
int vis[100005];
void tarjan(int u)
{
    pre[u]=lowlink[u]=++dfs_clock;
    S.push(u);
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        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])
    {
        scc_cnt++;
        for(;;)
        {
            int x=S.top();
            S.pop();
            sccno[x]=scc_cnt;
            if(x==u)break;
        }
    }
}

void find_scc(int n)
{
    dfs_clock=scc_cnt=0;
    memset(sccno,0,sizeof(sccno));
    memset(pre,0,sizeof(pre));
    for(int i=0;i<n;i++)
    if(!pre[i])tarjan(i);
}

int dfs(int x)
{
    for(int i=0; i<H[x].size(); i++)
    {
        int u=H[x][i];
        if(!vis[u])
        {
            vis[u]=1;
            dfs(u);
        }
    }
}
int main()
{
    int T,n,m,x,y,ct;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0; i<n; i++)
           {
               G[i].clear();
               H[i].clear();
           }
        while(m--)
        {
            scanf("%d%d",&x,&y);
            x--;y--;
            G[x].push_back(y);
        }
        find_scc(n);
        memset(fa,0,sizeof(fa));
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<G[i].size();j++)
            {
                int u=G[i][j];
                if(sccno[i]!=sccno[u])
                {
                    H[sccno[i]].push_back(sccno[u]);
                    fa[sccno[u]]++;
                }
            }
        }
        memset(vis,0,sizeof(vis));
        ct=0;
        for(int i=0;i<scc_cnt;i++)
        {
            if(!fa[i])
            {
                vis[i]=1;
                dfs(i);
                ct++;
            }
        }
        printf("%d\n",ct);
    }
    return 0;
}