kosaraju算法应用(一)

题目链接:POJ 2186

解题思路:
kosaraju算法,本以为要缩点,但是题目只要求找到拓扑排序的一个唯一的头,可以水过——通过计算强连通分量的出度。

代码:

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

int n,m,v[10005],num[10005],cnt,scc,out[10005],ans[10005];
vector<int> vec[10005];
vector<int> nvec[10005];

void dfs(int index)
{
    v[index]=1; 
    int len=vec[index].size();
    for(int i=0;i<len;i++)
        if(!v[vec[index][i]])
            dfs(vec[index][i]);         
    num[++cnt]=index;
}

void ndfs(int index)
{
    v[index]=scc,ans[scc]++;    
    int len=nvec[index].size();
    for(int i=0;i<len;i++)
        if(!v[nvec[index][i]])
            ndfs(nvec[index][i]);
}

int main()
{
    while(~scanf("%d %d",&n,&m))
    {
        memset(ans,0,sizeof(ans));
        memset(vec,0,sizeof(vec));
        memset(nvec,0,sizeof(nvec));

        int x,y;
        for(int i=0;i<m;i++)
        {
            scanf("%d %d",&x,&y);
            vec[x].push_back(y);
            nvec[y].push_back(x);
        }

        memset(v,0,sizeof(v));
        for (int i=1;i<=n;i++)
            if(!v[i])
                dfs(i);
        memset(v,0,sizeof(v));
        for(int i=cnt;i>=1;i--)
        {
            if(!v[num[i]])
            {
                scc++;
                ndfs(num[i]);
            }
        }

        memset(out,0,sizeof(out));
        int res=0,count=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<vec[i].size();j++)
            {
                if(v[i]!=v[vec[i][j]])
                    out[v[i]]++;
            }
        }
        for(int i=1;i<=scc;i++)
        {
            if(!out[i]) count++,res=ans[i];
            if(count>1) res=0;              
        }

        printf("%d\n",res);
    }

    return 0;
}

下次讲缩点~

你可能感兴趣的:(kosaraju算法应用(一))