Popular Cows POJ – 2186 – tarjan缩点

题目链接

https://cn.vjudge.net/problem/POJ-2186#author=hepan

解题思路

先将强连通分量都缩为一个点,就是用color数组对所有点进行染色,一个强连通分量中的点染色的编号相同。这样一来我们计算一下有几个强连通分量的出度为0,若出度为零的强连通分量数量正好为1,则这个强连通分量中的牛都是符合题目要求的,记一下数即可,否则不存在被所有牛都欢迎的牛。

代码

#include <iostream>
#include <vector>
#include <cstring>
#include <stack>
using namespace std;
const int maxn=10010;
vector<int>mp[maxn];

int dfn[maxn],low[maxn],du[maxn],color[maxn],ans[maxn],vis[maxn];

int cnt,tot;

stack<int>q;

void init(int n)
{
    cnt=0;
    tot=0;
    while(!q.empty())
    {
        q.pop();
    }
    for(int i=0;i<=n;i++)
    {
        mp[i].clear();
        dfn[i]=0;
        low[i]=0;
        du[i]=0;
        color[i]=0;
        ans[i]=0;
        vis[i]=0;
    }
}

void tarjan(int u)
{
    q.push(u);
    vis[u]=1;
    low[u]=dfn[u]=++cnt;
    int len=mp[u].size();
    for(int i=0;i<len;i++)
    {
        int v=mp[u][i];
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(vis[v])
        {
            low[u]=min(low[u],dfn[v]);
        }
    }
    if(low[u]==dfn[u])
    {
        color[u]=++tot;
        vis[u]=0;
        while(q.top()!=u)
        {
            color[q.top()]=tot;
            vis[u]=0;
            q.pop();
        }
        q.pop();
    }
}

int main()
{
    ios::sync_with_stdio(false);
    int n,m;
    while(cin>>n>>m)
    {
        init(n);
        int u,v;
        for(int i=0; i<m; i++)
        {
            cin>>u>>v;
            mp[u].push_back(v);
        }
        for(int i=1;i<=n;i++)
        {
            if(!dfn[i])
                tarjan(i);
        }
        for(int i=1;i<=n;i++)
        {
            int len=mp[i].size();
            for(int j=0;j<len;j++)
            {
                int v=mp[i][j];
                if(color[v]!=color[i])
                {
                    du[color[i]]++;
                }
            }
            ans[color[i]]++;
        }
        int num=0,aans;
        for(int i=1;i<=tot;i++)
        {
            if(du[i]==0)
            {
                num++;
                aans=ans[i];
            }
        }
        if(num>1)
        {
            cout<<"0"<<endl;
        }
        else cout<<aans<<endl;
    }
    return 0;
}

发表评论

电子邮件地址不会被公开。 必填项已用*标注