Codeforces Beta Round #88 C. Cycle（dfs+找环）

题目链接

不知道为什么这个题用ans记录会mle，下面是mle代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5e3+5;
int n,m,num[maxn],pre[maxn],flag=0;
vector<int>g[maxn],ans;
char s[maxn];
void dfs(int u,int fa,int deep)
{
	num[u]=deep;
	pre[u]=fa;
	for(int to:g[u])
	{
		if(flag) return ;
		if(num[to]>0){
			if(num[u]-num[to]+1==3)
			{
				flag=1;
			for(int i=u;i!=to;i=pre[i]) ans.push_back(i);
			ans.push_back(to);
			return ;
			}
			else continue;
		}
		dfs(to,u,deep+1);
	}
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
    	scanf("%s",s+1);
    	for(int j=1;j<=n;++j)
    	if(s[j]=='1') g[i].push_back(j); 
	}
	dfs(1,0,1);
	if(!flag) puts("-1");
	else {
		for(int i=ans.size()-1;i>=0;--i) printf("%d ",ans[i]);
	}
 } 

无奈只能改下面的写法。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5e3+5;
int n,m,num[maxn],flag=0;
char str[maxn],s[maxn][maxn];
void dfs(int u,int fa)
{
	num[u]=1;
	for(int i=1;i<=n;++i)
	{
		if(s[u][i]=='1')
		{
			if(!flag&&s[i][fa]=='1') {
				printf("%d %d %d",fa,u,i);
				flag=1;
				return ;
			}
			if(!num[i]) dfs(i,u);
		}
	}
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
    	scanf("%s",str+1);
    	for(int j=1;j<=n;++j)
    	s[i][j]=str[j];
	}
	for(int i=1;i<=n;++i)
	if(!num[i]) dfs(i,0);
	if(!flag) puts("-1");
 }