hdu 1181 变形课(dfs)

题目：
变形课

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 131072/65536 K (Java/Others)
Total Submission(s): 31204 Accepted Submission(s): 11247

Problem Description
呃……变形课上Harry碰到了一点小麻烦,因为他并不像Hermione那样能够记住所有的咒语而随意的将一个棒球变成刺猬什么的,但是他发现了变形咒语的一个统一规律:如果咒语是以a开头b结尾的一个单词,那么它的作用就恰好是使A物体变成B物体.
Harry已经将他所会的所有咒语都列成了一个表,他想让你帮忙计算一下他是否能完成老师的作业,将一个B(ball)变成一个M(Mouse),你知道,如果他自己不能完成的话,他就只好向Hermione请教,并且被迫听一大堆好好学习的道理.

Input
测试数据有多组。每组有多行，每行一个单词,仅包括小写字母,是Harry所会的所有咒语.数字0表示一组输入结束.

Output
如果Harry可以完成他的作业,就输出”Yes.”,否则就输出”No.”(不要忽略了句号)

Sample Input
so
soon
river
goes
them
got
moon
begin
big
0

Sample Output
Yes.

Hint
Hint

Harry 可以念这个咒语:”big-got-them”.

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

代码：

#include<fstream>
#include<iostream>
#include<string>
#include<cstring>
using namespace std;
const int maxn = 1010;
string ss[maxn];
int n = 0;
bool vis[maxn];
bool flag = false;

void dfs(char c){
    if(c == 'm'){
        flag = true;
        return ;
    }
    for(int i = 0; i < n; ++i){
        if(!vis[i] && *(ss[i].begin()) == c){
            vis[i] = true;
            dfs(*(ss[i].end() - 1));
        }
    }
}

int main(){
//    freopen("a.txt", "r", stdin);
    string s;
    while(cin >> s){
        n = 0;
        flag = false;
        while(s != "0"){
            ss[n++] = s;
            cin >> s;
        }
        for(int i = 0; i < n; ++i){
            if(*(ss[i].begin()) == 'b'){
                memset(vis, 0, sizeof(vis));
                vis[i] = true;
                dfs(*(ss[i].end() - 1));
                if(flag) break;
            }
        }
        if(flag) printf("Yes.\n");
        else printf("No.\n");

    }
    return 0;
}