Codeforces 117C DFS形成环

//题目连接http://www.codeforces.com/contest/117/problem/C
//题目大意是求一个有向图是否可以连接成三个元素的环,如果可以输出这三个数
//如何去深搜是重点..下面的代码中能够看到.DFS的参数含有now,prev,即当前节点和上一个节点.
//思路即是如果深搜过程中.记录的但prev节点和now节点以及now节点的子节点能够成环.则找到目标.
//判断是否成环就是w[son[now]][prev] = 1;即now的子节点可以指向now的前一个节点.
//还有一点就是如何去构造出深搜的一个特征..即搜过的点不再去DFS的理解..对于搜过的点..
//如果他不能够形成环,那么对于另外一个能够和这个点相连接的点必然也无法形成环.
//因为假设对于环1-->3-->2--->1来说..如果2能够形成环.那么1也能.3也能..但是如果2能够和
//1连接.但是1已经被搜索..但是不能形成环.那么如果2能够和1形成环的话..就与之前所说的
//1无法形成环矛盾了....
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<iostream>
#include<math.h>
#include<vector>
const int inf = 0x3f3f3f;
using namespace std;
char w[5010][5010];
bool flag;
int vis[5010];
int n;
void DFS(int now , int prev)
{
   vis[now] = 1;
   if(flag) return ;
   for(int i = 0 ; i < n ; i++)
   {
      if(w[now][i] == '1')
      {
          if(vis[i])
          {
              if(prev != -1 && !flag && w[i][prev] == '1')
              {
                   printf("%d %d %d\n",prev+1,now+1,i+1);
                   flag = 1 ;
                   return ;
              }
          }
          else
          {
              DFS(i,now);
          }
      }
   }
   return ;
}
int main()
{
    while(scanf("%d",&n) != EOF)
    {
        flag = 0;
        memset(vis,0,sizeof(vis));
        for(int i = 0 ; i < n ; i++)
          scanf("%s",w[i]);
        for(int i = 0; i < n && !flag; ++i)
            if(!vis[i])  DFS(0,-1);
        if(!flag)
        {
            printf("-1\n");
        }
    }
}