HDU 1181 变形课（dfs）

变形课

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

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.

挺简单的一道深搜题，但是wa了好几次，没有考虑到多组数据，还有第一回输入就为0的情况，还是想的不全面，我的AC代码中主函数中代码的处理不是很完美。

#include<stdio.h>
#include<string>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
struct node{
	char st;
	char end;
}a[10005];
int ans,flag;
int b[10005];
int vis[10005];
void dfs(node t)
{
	if(flag) return;
	if(t.end == 'm')//词尾为m,成立 
	{
		flag=1;
		return;
	}
	for(int i=0;i<ans;i++)
	{
		if(!vis[i]&&t.end==a[i].st)
		{
			vis[i]=1;
			dfs(a[i]);
			vis[i]=0;
		}
	}
}
int main()
{
	string s;
	while(cin>>s)
	{
		flag=0;
		int num=1;//以b开头的单词数，相当于起点 
		ans=1;//单词数目 
		memset(vis,0,sizeof(vis));
		if(s.compare("0")==0)//第一次就为0时，没有单词 
		{
			printf("No.\n");
			continue;
		}
		else {//否则保存 
			int len=s.length();
			a[0].st = s[0];
			a[0].end =s[len-1];
			if(s[0]=='b')
			{
				b[0]=0;
			}
		}
		while(cin>>s)//接下来输入 
		{
			if(s.compare("0")==0) 
				break;
			int len=s.length();
			int res=ans;
			ans++;
			a[res].st = s[0];
			a[res].end =s[len-1];
			if(s[0]=='b')//把b开头单词保存
			{
				b[num++]=res;
			}
		}
		for(int i=0;i<num;i++)//以每个起点开始 
		{
			if(flag) 
				break;
			vis[b[i]]=1;
			dfs(a[b[i]]);
			vis[b[i]]=0;
		}
		if(flag)
			printf("Yes.\n");
		else
			printf("No.\n");
	}
	return 0;
}

这个处理的比我的完美多了！注释自己加的

int main()  
{  
    string s;  
    while(cin>>s)  
    {  
        if(s == "0")  
        {  
            memset(mark,0,sizeof(mark));  
            for(int j = 0;j < i;j++)  
            {  
                if(w[j].s == 'b')  //每次都已b开头的搜索,而我是把以b开头的单词下标保存到一个数组了
                {  
                    dfs(w[j]);  
                }  
            }  
            if(flag == 1)  
                cout<<"Yes."<<endl;  
            else  
                cout<<"No."<<endl;  
            flag = 0;  
            i = 0;  
        }  
        else  
        {  
            w[i].s = s[0];  
            w[i].e = s[s.length()-1];  
            i++;//总单词个数  
        }  
  
    }  
  
}