HDoj 1181変形課【DFS】

変形レッスン
Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 131072/65536 K (Java/Others)
Total Submission(s): 18035    Accepted Submission(s): 6478
Problem Description
ええと、変形の授業でハリーはちょっと困ったことがありました.彼はHermioneのようにすべての呪文を覚えることができないので、勝手に野球をハリネズミに変えるなんて、変形呪文の統一的な法則を発見しました.もし呪文がaの冒頭のbで終わる単語であれば、その役割はちょうどA物体をB物体にすることです.
Harryはすでに彼のできるすべての呪文を1つの表に並べて、彼はあなたに彼が先生の宿題を完成できるかどうかを計算してもらいたいと思って、1つのB(ball)を1つのM(Mouse)に変えて、あなたは知っていて、もし彼自身が完成できないならば、彼はHermioneに教えてもらうしかなくて、しかもたくさんのよく勉強する道理を聞かなければなりません.
 
Input
テストデータには複数のグループがあります.各グループには複数の行があり、各行に1つの単語があり、小文字のみを含み、Harryができるすべての呪文である.数字0は入力の終わりを表す.
 
Output
ハリーが彼の宿題を完成できるなら「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 <algorithm>
using namespace std;
char str[1010][2];
int N;
bool vis[1010];
bool flag;
void DFS(int pos)
{
    if(str[pos][1] == 'm')
    {
        flag = true;
        return ;
    }
    for(int i = 0; i < N; i++)
    {
        if(!vis[i] && str[pos][1] == str[i][0])
        {
            vis[i] = true;
            DFS(i);
        }
    }
}
int main()
{
    int len;
    char s[1010];
    while(scanf("%s", s) != EOF)
    {
        N = 0;
        len = strlen(s);
        str[N][0] = s[0]; str[N++][1] = s[len-1];
        while(scanf("%s", s), strcmp(s, "0"))
        {
            len = strlen(s);
            str[N][0] = s[0]; str[N++][1] = s[len-1];
        }
        flag = false;
        for(int i = 0; i < N; i++)
        {
            if(str[i][0] == 'b')
            {
                memset(vis, false, sizeof(vis));
                vis[i] = true;
                DFS(i);
                if(flag)
                    break;
            }
        }
        if(flag)
            printf("Yes.
");
        else
            printf("No.
");
    }
    return 0;
}