HDU 1181 变形课

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): 15071    Accepted Submission(s): 5473

Problem Description

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

 

Input

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

 

Output

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

 

Sample Input

sosoonrivergoesthemgotmoonbeginbig0

 

Sample Output

Yes.
Hint
Hint
 Harry 可以念这个咒语:"big-got-them".

用结构体将单词的头尾储存起来，然后dfs。

我的代码：

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;char a[1005];int len,total,flag;int vis[1005];struct tw{char head,end;}word[1005];void dfs(int x){vis[x]=1;                                                            //标记已访问if(word[x].end=='m') {flag=1;return;}for(int i=0;i<=total-1;i++){if(!vis[i]){if(word[x].end==word[i].head) dfs(i);}}}int main(){while(scanf("%s",a)!=EOF){if(strcmp(a,"0")==0) {printf("No.\n");continue;}len=strlen(a);word[0].head=a[0];word[0].end=a[len-1];total=1;while(1){scanf("%s",a);if(strcmp(a,"0")==0) break;len=strlen(a);word[total].head=a[0];word[total].end=a[len-1];total++;                                                      //total为输入的单词个数}//for(int i=0;i<=total-1;i++)//printf("%c%c\n",word[i].head,word[i].end);flag=0;for(int i=0;i<=total-1;i++){if(word[i].head=='b') {memset(vis,0,sizeof(vis));dfs(i);} //搜索前把标记清0}if(flag) printf("Yes.\n");else printf("No.\n");}return 0;}