BestCoder冠军赛 - 1009 Exploration 【Tarjan+并查集缩点】

题意

给一个图,这个图中既有有向边,又有无向边,每条边只能走一次,问图中是否存在环。

最多10^6个点,10^6个无向边,10^6个有向边

 

【题解】

因为既有有向边又有无向边,所以不能单纯的用tarjan判环,阻碍咱使用tarjan的主要还是无向边,显然不可以把无向边拆成两个有向边来搞,因为这条边只能走一次,拆成两条边可能来回分别走一次,就不满足题意了。

咱可以先处理无向边,无向边可能会把图连成一个个的连通块,用并查集,把这些连通块缩成一个点,注意,如果当发现某个无向边的两个端点在同一个并查集中,说明已经找到环了。

接下来就是处理有向边,直接用tarjan判环即可,还是要注意处理加入有向边的时候如果两个端点在同一个并查集中,说明已经找到环了,但是要注意你并不能立即输出并退出,因为这个是多CASE,所以需要读完整个数据才能退出。

 

============

写挂了两发,最后发现tarjan函数开始部分非常傻逼的把栈清空了。。。

 

 

#pragma comment(linker, "/STACK:102400000,102400000")

#include<cstdio>

#include<cstring>

#include<cmath>

#include<iostream>

#include<algorithm>

#include<set>

#include<map>

#include<stack>

#include<vector>

#include<queue>

#include<string>

#include<sstream>

#define eps 1e-9

#define ALL(x) x.begin(),x.end()

#define INS(x) inserter(x,x.begin())

#define FOR(i,j,k) for(int i=j;i<=k;i++)

#define MAXN 2000005

#define MAXM 4000005

#define INF 0x3fffffff

#define PB push_back

#define MP make_pair

#define X first

#define Y second

#define lc (k<<1)

#define rc ((k<<1)1)

using namespace std;

typedef long long LL;

int i,j,k,n,m,x,y,T,ans,big,cas,num,len;

bool flag;

int fa[MAXN]; 

int find(int x)

{

    if (x==fa[x]) return fa[x];

    return fa[x]=find(fa[x]);

}



int edge,head[MAXN];

struct edgenode

{

    int to,next;

} G[MAXM];



void add_edge(int x,int y)

{

    G[edge].to=y;

    G[edge].next=head[x];

    head[x]=edge++;

}





int K,indexs,nn,mm;

int dfn[MAXN],low[MAXN],instack[MAXN],belong[MAXN];

stack <int> tar;

 

bool tarjan(int k)

{

    int p,j;

    tar.push(k);

    instack[k]=1;

    dfn[k]=low[k]=++indexs;

    for(int j=head[k];j!=-1;j=G[j].next)

    {

        p=G[j].to;

        if (instack[p])

        {

            low[k]=min(low[k],dfn[p]);

        }

        else

        if(dfn[p]==-1)

        {

            if (tarjan(p)) return true;

            low[k]=min(low[k],low[p]);

        }

    }



    if(low[k]==dfn[k])

    {

        nn++;

        int sz=0;

        do

        {

            j=tar.top();

            tar.pop();

            instack[j]=0;

            belong[j]=nn;

            sz++;

        }while(j!=k);

        

        if (sz>1) return true;

    }

    return false;

}



bool build_map()

{

    memset(dfn,-1,sizeof(dfn));

    memset(low,-1,sizeof(low));

    memset(instack,0,sizeof(instack));

    indexs=0;nn=0;

    while (!tar.empty()) tar.pop();

    for (i=0;i<=n;i++) belong[i]=i;

    for (i=1;i<=n;i++)

    {

        if (fa[i]==i && dfn[i]==-1)

            if (tarjan(i)) return true;

    }

    

    return false;

}







bool RUN()

{

    memset(head,-1,sizeof(head));

    edge=0;

    int m1,m2;

    scanf("%d%d%d",&n,&m1,&m2);

    for (i=0;i<=n;i++) fa[i]=i;

    flag=false;

    for (i=1;i<=m1;i++)

    {

        scanf("%d%d",&x,&y);

        int fx=find(x);

        int fy=find(y);

        if (fx==fy) flag=true;

        else fa[fx]=fy;

    }

    

    for (i=1;i<=m2;i++)

    {

        scanf("%d%d",&x,&y);

        x=find(x);

        y=find(y);

        if (x==y) flag=true;

        add_edge(x,y);

    }

    if (flag) return true;

    return build_map();

}



int main()

{

    scanf("%d",&T);

    while (T--)

    {

        if (RUN()) printf("YES\n"); else printf("NO\n");

    }

    return 0;

}

 

你可能感兴趣的:(code)