1098: PIPI的变形课

变形课上PIPI碰到了一点小麻烦,因为他并不能够记住所有的咒语而随意的将一个棒球变成刺猬什么的,但是他发现了变形咒语的一个统一规律:如果咒语是以a开头b结尾的一个单词,那么它的作用就恰好是使A物体变成B物体.

PIPI已经将他所会的所有咒语都列成了一个表,他想让你帮忙计算一下他是否能完成老师的作业,将一个B(ball)变成一个M(Mouse)。

输入：

测试数据有多组。每组有多行，每行一个单词,仅包括小写字母,是PIPI所会的所有咒语,每个单词长度不超过15.

数字0表示一组输入结束.

输出：

如果PIPI可以完成他的作业,就输出"Yes.",否则就输出"No."

样例输入

so

soon

river

goes

them

got

moon

begin

big

0

样例输出

Yes.

#include<iostream>
#include <iomanip>
#include<cstring>
#include<algorithm>
#include<stack>
using namespace std;
int a[27][27];
int vis[27];
int flag;
void find(int x,int y){
  if(flag==1)return;
  vis[x]=1;
  for(int i=0;i<=26;i++){
    if(a[x][i]==1&&vis[i]==0){
      if(i==y){
        flag=1;
        //printf("yes.\n");
        return;
      }
      find(i,y);
    }
  }
}
int main(){
  string str;
  //str.resize(20);
  memset(a,0,sizeof(a));
  memset(vis,0,sizeof(vis));
  flag=0;
  while(cin>>str){
    if(str=="0"){
      find('b'-'a','m'-'a');
      if(flag==0)printf("No.\n");
      else printf("Yes.\n");
      memset(a,0,sizeof(a));
      memset(vis,0,sizeof(vis));
      flag=0;
      continue;
    }
    char ch1=str[0];
    char ch2=str[str.length()-1];
    a[ch1-'a'][ch2-'a']=1;
    
  }
}