杭电 1181 变形课 DFS

变形课

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 131072/65536 K (Java/Others)

Total Submission(s): 18640    Accepted Submission(s): 6720

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.

#include<cstdio>  
#include<cstring>  
#include<algorithm>  
using namespace std;  
char str[10000];  
int flag,cot,vis[10000];  
struct node   
{  
    char s,e;  
}q[20000];  
void dfs(int x)  
{  
    if(q[x].e=='m')  
    {  
        flag=1;  
        return ;  
    }  
    for(int i=0;i<cot;i++)  
    {  
        if(q[x].e==q[i].s&&!vis[i])  
        {  
            vis[i]=1;  
            dfs(i);  
        }  
     }   
}  
int main()  
{  
    int i,len;  
    cot=0;  
    while(scanf("%s",str)!=EOF)  
    {  
        if(strcmp(str,"0")!=0)  
        {  
            len=strlen(str);  
            q[cot].s=str[0];  
            q[cot++].e=str[len-1];  
        }  
        else  
        {  
            flag=0;  
            memset(vis,0,sizeof(vis));  
            for(i=0;i<cot;i++)  
            {  
                if(q[i].s=='b')  
                {  
                    vis[i]=1;  
                    dfs(i);  
                }  
            }  
            if(flag)  
            printf("Yes.\n");  
            else  
            printf("No.\n");  
            cot=0;  
        }  
    }  
    return 0;  
 }   

写法二 ：不用结构体

#include<iostream>
#include<string>
#include<cstring>
using namespace std;

string s[1009];//看的别人的最大1000
int vis[1009];
int t = 0;
bool dfs(char a)
{
	if (a == 'm')
		return true;
	for (int i = 0; i < t; i++)
		if (s[i][0] == a && vis[i] == 0)
		{
			vis[i] = 1;
			if (dfs(s[i][s[i].size() - 1]))
				return true;
			vis[i] = 0;
		}			
	return false;
}

int main()
{
	while (cin >> s[t])
	{
		if (s[t] == "0")
		{
			memset(vis, 0, sizeof(vis));
			int state = 0;
			for (int i = 0; i < t; i++)
				if (s[i][0] == 'b')
				{
					vis[i] = 1;
					if (dfs(s[i][s[i].size() - 1]))
					{
						cout << "Yes." << endl;
						state = 1;
						break;
					}
					vis[i] = 0;
				}
			if (state == 0)
				cout << "No." << endl;
			t = 0;
		}
		else
			t++;
	}
	return 0;
}