变形课

变形课

呃…变形课上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.

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

解题思路：类似于成语接龙，并保证最后一个单词的最后一个字母是m。整型数组可以用来存字符 ，以前没碰到过 .其他就是简单的dfs。
Code:

#include<iostream>
#include<cstring>
using namespace std;
string s;
int vis[300];
int flag=0;
int map[300][300];
void dfs(char k){
	if(k=='m'){
		flag=1;
		return ;
	}
	else{
		for(char i='a';i<='z';i++){
			if(map[k][i]==1&&vis[i]==0){
				vis[i]=1;
				dfs(i);
			}	
		}
	}
}
int main(){
	while(cin>>s){
		if(s=="0"){
			vis['b']=1;// 整型数组可以用来存字符 ，以前没碰到过 
			dfs('b');
			if(flag) cout<<"Yes.
";
			else cout<<"No."<<endl;
			memset(map,0,sizeof(map));
			memset(vis,0,sizeof(vis));
			flag=0;
		}
		else map[s[0]][s[s.length()-1]]=1;
	}
	
	return 0;
}