gdufe acm 1054 变形课

链接：
http://acm.gdufe.edu.cn/Problem/read/id/1054

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
big
keep
dim
0
Sample Output:

Yes.
No.

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <string>
#include <vector>
using namespace std;
vector<string> vec;
const int maxn = 10000;
int vis[maxn];
bool dfs(int x){
    vis[x] = 1;//标记vec[x]已访问
    for(int i = 0; i < vec.size(); i++){
        if(!vis[i] && vec[i][0] == vec[x][vec[x].size()-1]){/*
        如果vec[i]未访问，且其首字母等于vec[x]的末字母，判断vec[i]的末字母是否为m，是则返回true，否则递归dfs(i)
*/
            if(vec[i][vec[i].size()-1] == 'm'){
                return true;
            }
            if(dfs(i)) return true;
        }
//不需要把vis[i]标记回0
    }
    return false;
}
int main()
{
    string s;
    int flag = 1;
    while(cin >> s){
        if(s == "0"){
            for(int i = 0; i < vec.size(); i ++){
                if(vec[i][0] == 'b'){
                    if(vec[i][vec[i].size()-1] == 'm' || dfs(i))
                    {
                        cout << "Yes." << endl; flag = 0; break;
                    }
                }
            }
            if(flag) cout << "No." << endl;
            memset(vis, 0, sizeof(vis));
            vec.clear();
            flag = 1;
            continue;
        }
        vec.push_back(s);
    }
    return 0;
}