HDU 1181 变形课

http://acm.hdu.edu.cn/showproblem.php?pid=1181

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.

 

时间复杂度：$O(n+m)$，其中 $n$ 表示节点个数，$m$ 表示有向边的条数。  

代码：

#include <bits/stdc++.h>
using namespace std;

const int maxn = 30;
int g[maxn][maxn];
char s[100000];
int len;

/*
  g[i][j] == 0 (i -> j 走不通)
  g[i][j] == 1 (i -> j 可以走)
*/

void bfs() {
  int flag[30];
  for(int i = 0; i < 26; i ++) {
    flag[i] = 0;
  }
  queue<int> Q;
  Q.push(1);
  flag[1] = 1;
  while(!Q.empty()) {
    int t = Q.front();
    Q.pop();
    for(int i = 0; i < 26; i ++) {
      if(g[t][i] && flag[i] == 0) {
        Q.push(i);
        flag[i] = 1;
      }
    }
  }

  if(flag['m' - 'a']) {
    printf("Yes.
");
  } else {
    printf("No.
");
  }
}

int main() {
    while(~scanf("%s", s)) {
      len = strlen(s);
      if(len == 1 && s[0] == '0') {
        bfs();
        memset(g, 0, sizeof g);
        continue;
      }
      g[s[0] - 'a'][s[len - 1] - 'a'] = 1;
    }
    return 0;
}