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;}