codevs 2066 三角恋（dfs）

题目描述 Description
人们的恋爱关系与如下特点：两个人a，b，如果a爱着b，那么b一定不爱a；如果a不爱b，那么b一定爱着a（yh的研究成果果然与众不同……）。
现在，yh想知道，在n个人当中，是否存在三角恋现象（即a爱着b，b爱着c，c爱着a）。

输入描述 Input Description
本题有多组测试数据
第一行一个整数t，表示有t组测试数据。
对于每组数据，第一行为一个整数n，表示共有n人。
接下来为n*n的0,1矩阵a，如果a[i，j]=1,则i爱着j，否则表示i不爱j;数据保证a[i,j]<>a[j,i]。

输出描述 Output Description
每组数据输出一行，如果存在三角恋，输出’Yes’，否则输出’No’。

样例输入 Sample Input
2
5
00100
10000
01001
11101
11000
5
01111
00000
01000
01100
01110

样例输出 Sample Output
Yes
No

数据范围及提示 Data Size & Hint
对于40%的数据，n≤10
对于100%的数据，n≤2000, t≤5

题解：根据题目描述可以看出，这道题是问你一个有向图中有没有环，而且要求环的长度为3。我们可以用dfs来做，dfs的时候记录一下起点，时间戳，当搜到一个用过的点且记录的起点相同即为找到了一个环，环的长度即为他们时间戳的差。

代码如下：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int MAXN=2001000;
int first[MAXN],nxt[MAXN<<1];
int n,tot,han[MAXN],tim[MAXN];
bool used[MAXN],flag;
struct edge
{
    int from,to;
}es[MAXN<<1];
void init()//多组数据，每次都要初始化 
{
    memset(first,-1,sizeof(first));
    memset(used,0,sizeof(used));
    memset(tim,0,sizeof(tim));
    memset(han,0,sizeof(han));
    tot=0;
}
void build(int f,int t)
{
    es[++tot]=(edge){f,t};
    nxt[tot]=first[f];
    first[f]=tot;
}
void dfs(int s)
{
    if(flag) return ;
    for(int i=first[s];i!=-1;i=nxt[i])
    {
        int v=es[i].to;
        if(!used[v])
        {
            tim[v]=tim[s]+1;//记录时间戳 
            han[v]=han[s];//记录起点编号 
            used[v]=1;
            dfs(v);
        }
        else if(han[v]==han[s])//找到一个环 
            if(tim[s]-tim[v]==2)//环的长度为3 
            {
                flag=1;
                return ;
            }
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        init();
        for(int i=1;i<=n;i++)
        {
            string s;
            cin>>s;
            for(int j=0;j<n;j++)
                if(s[j]=='1') build(i,j+1);
        }
        for(int i=1;i<=n;i++)
            if(!used[i])
            {
                flag=0;//初始化 
                used[i]=1;
                han[i]=i;
                dfs(i);
                if(flag)
                {
                    printf("Yes\n");
                    break;
                }
            }
        if(!flag) printf("No\n");
    }
    return 0;
}