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

  

   so
soon
river
goes
them
got
moon
begin
big
0
  

 

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