hdu 4324 Triangle LOVE

题意：给出一个有向图，问有没有一个长度为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;
}