hdu 4324(dfs)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4324

思路:dfs搜索即可，如果当前点u的下一个点v已经访问过了，那么就判断dist[u]==dist[[v]+2,成立返回true,否则更新dist[v]=dist[u]+1,继续深搜。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 2222
vector<int>vet[MAXN];
char str[MAXN];
int dist[MAXN];
bool mark[MAXN];
int n;

bool dfs(int u) {
    mark[u]=true;
    for(int i=0; i<vet[u].size(); i++) {
        int v=vet[u][i];
        if(mark[v]&&dist[u]==dist[v]+2) {
            return true;
        } else if(!mark[v]) {
            dist[v]=dist[u]+1;
            if(dfs(v))return true;
        }
    }
    return false;
}


int main() {
//    freopen("1.txt","r",stdin);
    int _case,t=1;
    scanf("%d",&_case);
    while(_case--) {
        scanf("%d",&n);
        for(int i=1; i<=n; i++)vet[i].clear();
        for(int i=1; i<=n; i++) {
            scanf("%s",str+1);
            for(int j=1; j<=n; j++) {
                if(str[j]=='1')vet[i].push_back(j);
            }
        }
        memset(dist,0,sizeof(dist));
        memset(mark,false,sizeof(mark));
        bool flag=false;
        for(int i=1; i<=n; i++) {
            if(vet[i].size()) {
                if(dfs(i)) {
                    flag=true;
                    break;
                }
            }
        }
        printf("Case #%d: ",t++);
        flag?puts("Yes"):puts("No");
    }
    return 0;
}