Vijos 1626 爱在心中

爱在心中

tarjan缩点

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000000
int n,m,x,y,z,num,low[maxn],tim,cnt,q[maxn],t,ta,a[maxn];
int dfn[maxn],point[maxn],color[maxn],Stack[maxn];
int top,tot,ans,sumcol,head2[maxn],head[maxn],in[maxn],out[maxn];
bool vis[maxn];

struct Edge{
    int to,next,from;
}edge[maxn],edge2[maxn];

void add(int u,int v)
{
    edge[++num].to=v;
    edge[num].from=u;
    edge[num].next=head[u];
    head[u]=num;
}

void ins(int u,int v)
{
    edge2[++tot].to=v;
    edge2[tot].from=u;
    edge2[tot].next=head[u];
    head2[u]=tot;
}

void tarjan(int x)
{
    dfn[x]=low[x]=++tim;
    Stack[++top]=x; vis[x]=true;
    for(int i=head[x];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(vis[v]) low[x]=min(low[x],dfn[v]);
        else if(!dfn[v]) tarjan(v),low[x]=min(low[x],low[v]);
    }
    if(low[x]==dfn[x])
    {
        sumcol++; color[x]=sumcol; point[sumcol]++;
        for(;Stack[top]!=x;top--)
            color[Stack[top]]=sumcol,point[sumcol]++,vis[Stack[top]]=false;
        vis[x]=false; top--;
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    for(int i=1;i<=n;i++) if(!dfn[i]) tarjan(i);
    for(int u=1;u<=n;u++)
        for(int i=head[u];i;i=edge[i].next)
            if(color[u]!=color[edge[i].to])
                ins(color[u],color[edge[i].to]);
    for(int i=1;i<=tot;i++)    out[edge2[i].from]++;
    for(int i=1;i<=sumcol;i++)
    {
        if(point[i]>1&&!out[i]) q[++t]=i;
        if(point[i]>1) ans++;
    }
    printf("%d
",ans);
    if(t==1)
    {
        for(int i=1;i<=n;i++)
            if(color[i]==q[t]) a[++ta]=i;
        sort(a+1,a+ta+1);
        for(int i=1;i<=ta;i++)
        printf("%d ",a[i]);    
    }
    else if(t!=1) printf("-1
");
    return 0;
}