hdu 4324 Triangle LOVE

来源:互联网 发布:淘宝cpu散片能买吗 编辑:程序博客网 时间:2024/05/22 06:21

题意:给出一个有向图,问有没有一个长度为3的环(三角恋)。

思路:这题挺有意思的,刚开始YY了下,想着把tarjan改一改就能过了吧,写完以后交上去果断wrong了,后来想着暴力一点吧,直接每个点搜3歩试试,果断TLE了……后来想了想,可以再dfs的时候记录其父节点,然后当前节点找到它的子节点时,看看这个子节点和记录的父节点时候连边(由于已经给出了矩阵,所以很方便),这样遍历一遍就行了。这样搞虽然并不是长度为3的环上的每个点都满足这样的结论,但由于dfs的时候会访问每一个点,如果有长度为3的环,那么一定有一个点是满足这个要求的。写完以后找了找题解,发现好多都是拓扑排序做的Orz,我还是太年轻了。

 

代码:

 

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<algorithm>#include<map>#include<queue>#include<stack>#include<cmath>#include<vector>#define inf 0x3f3f3f3f#define Inf 0x3FFFFFFFFFFFFFFFLL#define eps 1e-9#define pi acos(-1.0)using namespace std;typedef long long ll;const int maxn=2000+10;char a[maxn][maxn];struct Edge{    int v,next;};Edge edges[maxn*maxn];int head[maxn],nEdge,n;bool vis[maxn],finds;void AddEdge(int u,int v){    nEdge++;    edges[nEdge].v=v;    edges[nEdge].next=head[u];    head[u]=nEdge;}bool dfs(int u,int fa){    vis[u]=true;    for(int k=head[u];k!=-1;k=edges[k].next)    {        int v=edges[k].v;        if(vis[v]) continue;        if(fa!=-1&&a[v-1][fa-1]=='1')          return true;        if(dfs(v,u)) return true;    }    return false;}int main(){    //freopen("in.txt","r",stdin);    //freopen("out.txt","w",stdout);    int t,tcase=0;    scanf("%d",&t);    while(t--)    {        tcase++;        memset(head,0xff,sizeof(head));        nEdge=-1;        scanf("%d",&n);        for(int i=0;i<n;++i)        {            scanf("%s",a[i]);            for(int j=0;j<n;++j)                if(a[i][j]=='1')                 AddEdge(i+1,j+1);        }        finds=false;        memset(vis,0,sizeof(vis));        for(int i=1;i<=n;++i)          if(!vis[i])          {              if(dfs(i,-1))              {                  finds=true;                  break;              }          }        printf("Case #%d: ",tcase);        if(finds) puts("Yes");        else puts("No");    }    return 0;}


 

原创粉丝点击