HDU 1181 变形课 （DFS）

题目：

呃……变形课上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.

---

题目分析：

给出一串单词，要求我们找找能否从b为开头的单词找到m为结尾的单词，例如：big – got – them 。这样就算从b找到了m。

考虑在输入单词的时候把首字母作为父节点，尾字母作为子节点，将26个字母转化成数字，输入所有单词之后，进行深度优先搜索，以b为开头，去寻找是否有m这个子节点，同时要注意防止死循环的发生要加入访问数组。

---

代码如下：

#include <iostream>
#include <cstdio>
#include <set>
#include <algorithm>
#include <cstring>
using namespace std;
set<int> ar[30];

bool f = 0;        //全局变量
bool vis[30];        //访问数组，如果搜索过就不进行搜索

void dfs(int i)
{
    vis[i] = 1;        //标记下来这个已经被搜索过
    for(auto &n : ar[i])
    {
        if(n == 12)        //如果找到了m，就改变全局变量
            f = 1;
        if(!vis[n])        //如果子节点没有被搜索过就搜索子节点
            dfs(n);
    }
}

int main()
{
    string a;
    while(cin >> a)
    {
        while(a[0] != '0')
        {
            int b, l;
            l = a.length();
            //cout << l << endl;
            b = a[l - 1] - 'a';        //尾字母转成数字
            //cout<<a[0] - 'a'<<endl;
            //cout << b << endl;
            ar[a[0] - 'a'].insert(b);        //存入节点，单向存入
            cin >> a;
        }
        dfs(1);        //从b开始查找
        if(f)
            cout << "Yes." << endl;
        else
            cout << "No." << endl;
        for(int i = 0; i < 30; i++) ar[i].clear();
        memset(vis, 0, sizeof(vis));
        f = 0;        //三行全都是初始化
    }
    return 0;
}

---

这里说一下为什么要是用全局变量f来控制输出结果，如果在dfs中使用了return true;和return false;的话，我们不能保证最后一个一定是true，可能会发生中间出现了true，最后一次是一个false覆盖掉之前的true，因此使用全局变量f。

另外，访问数组是为了避免重复搜索，出现死循环的情况。

---

笔者水平有限，如有错误，欢迎指出。