hdu 1181 变形课 （dfs）简单搜索

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

变形课

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

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".

开始以为是个水题，直接检索有没有同时存在b开头及m结尾的单词；

wa了一次之后才仔细看的题目。

a开头b结尾， A可以变成B，题目要求有没有b开头，m结尾的单词，而单词可以通过首尾相连“变长”；  "big-got-them".

【源代码】

#include<iostream>
#include <string>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1010;
string str[maxn];
bool vis[maxn];
int k;//将dfs中需要的变量设为全局
bool dfs(int po){
	vis[po]=1;
	int len=str[po].length();
	if(str[po][len-1]=='m')  //由于每一个进入dfs的单词都是 b 开头的 ，所以 直接检索 是否以 m 结尾
		return true;
	for(int i=0;i<k;i++){
		if(vis[i]) continue;
		if(str[i][0]==str[po][len-1]){
			if(dfs(i)) return true;
			vis[i]=0;
		}
	}
	return false;
}
int main(){
	string a;
	k=0;
	int sign=0;
	memset(vis,0,sizeof(vis));
	while(cin>>a){
		if(a=="0"){
			for(int i=0;i<k;i++){
				
				if(str[i][0]=='b'){
				//	cout<<"bingo"<<endl;
					if(dfs(i)){
						cout<<"Yes."<<endl; sign=1;break;
					}
				}
					
			}
			if(!sign)
				cout<<"No."<<endl;
			k=0;sign=0;  //初始化
				memset(vis,0,sizeof(vis));
		}
		else{
			str[k++]=a;
		
		}
	}
	return 0;
}