HDU 1181--变形课【DFS】

变形课

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

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".

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;

string str[1100];
bool vis[1100];
int ans, flag;

void dfs(int pos){
    int len = str[pos].length();
    if(str[pos][len - 1] == 'm'){
        flag = 1;
        return ;
    }
    for(int i = 0; i < ans; ++i){
        if(str[pos][len - 1] == str[i][0] && !vis[i]){
            vis[i] = 1;
            dfs(i);
        }
    }
}

int main (){
    string s;
    ans = 0;
    flag = 0;
    while(cin >> s){
        if(s[0] == '0'){
            for(int i = 0; i < ans; ++i)
                if(str[i][0] == 'b')
                    dfs(i);
                if(flag)
                    printf("Yes.\n");
                else
                    printf("No.\n");
                flag = 0;
                ans = 0;
            }
        else {
            str[ans++] = s;
        }
    }
    return 0;
}