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