POJ1236图的连通性Tarjan

/*
Tarjan缩点后,
Task A:
统计没有入度的点,没有没有入度的点,答案为1(此时Task B的答案为0)
Task B:
n个点
a个有入边  n-a个没有入边
b个有出边 n-b个没有出边
要添入边数:n-a
要添出边数:n-b
入边和出边可互补,共需添边max(n-a,n-b)条.
*/
#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#define mem(a,x) memset(a,x,sizeof(a));
using namespace std;

const int NN=102;

int n;
vector<int> a[NN];

int bcnt,top,depth;
int dfn[NN],low[NN],belong[NN],stack[NN];
bool instack[NN];
void tarjan(int u)
{
    dfn[u]=low[u]=++depth;
    stack[++top]=u;
    instack[u]=true;
    int v;
    for (int i=0; i<a[u].size(); i++)
    {
        v=a[u][i];
        if (!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if (instack[v])
            low[u]=min(low[u],dfn[v]);
    }
    if (low[u]==dfn[u])
    {
        bcnt++;
        do
        {
            v=stack[top--];
            belong[v]=bcnt;
            instack[v]=false;
        }while (v!=u);
    }
}

void solve()
{
    int chu[NN],ru[NN];
    bcnt=top=depth=0;
    for (int i=1; i<=n; i++)
    {
        dfn[i]=0;
        instack[i]=false;
    }
    for (int i=1; i<=n; i++) if (!dfn[i]) tarjan(i);
    if (bcnt==1) { printf("1\n0\n"); return; }
    for (int i=1; i<=bcnt; i++) chu[i]=ru[i]=false;
    for (int u=1; u<=n; u++)
    {
        for (int i=0; i<a[u].size(); i++)
        {
            int v=a[u][i];
            if (belong[u]!=belong[v])
            {
                chu[belong[u]]=true;
                ru[belong[v]]=true;
            }
        }
    }
    int sum1=0,sum2=0;
    for (int i=1; i<=bcnt; i++)
    {
        if (!chu[i]) sum1++;
        if (!ru[i])  sum2++;
    }
    printf("%d\n",sum2);
    printf("%d\n",max(sum1,sum2));
}

int main()
{
    scanf("%d",&n);
    for (int i=1; i<=n; i++)
    {
        int j;
        while (scanf("%d",&j),j) a[i].push_back(j);
    }
    solve();
    return 0;
}

你可能感兴趣的:(POJ1236图的连通性Tarjan)