题目描述 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;
}