变形课

问题描述：

呃……变形课上Harry碰到了一点小麻烦,因为他并不像Hermione那样能够记住所有的咒语而随意的将一个棒球变成刺猬什么的,但是他发现了变形咒语的一个统一规律:如果咒语是以a开头b结尾的一个单词,那么它的作用就恰好是使A物体变成B物体.
Harry已经将他所会的所有咒语都列成了一个表,他想让你帮忙计算一下他是否能完成老师的作业,将一个B(ball)变成一个M(Mouse),你知道,如果他自己不能完成的话,他就只好向Hermione请教,并且被迫听一大堆好好学习的道理.

Input

测试数据有多组。每组有多行，每行一个单词,仅包括小写字母,是Harry所会的所有咒语.数字0表示一组输入结束.

Output

如果Harry可以完成他的作业,就输出”Yes.”,否则就输出”No.”(不要忽略了句号)

问题分析：

这是一道简单的dfs题目。先看下一个例子：

so
soon
river
goes
them
got
moon
begin
big
0

我们可以有big-got-them得到最终答案。我的想法是只要记录每一个字母的首字母和最后一个字母，然后直接dfs遍历查找是否有’m’。

AC代码：

/*
author:Manson
date:8.19.2018
theme:dfs
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 10005;

vector<int>a[N];
int used[N];
int flag;
char s[N];
void dfs(int x){

    if(x == 'm'-'a'){
        flag = 1;
        return;
    }
    if(flag){
        return;
    }
    for(int i:a[x]){
        if(!used[i]){
            used[i] = 1;
            dfs(i);
            used[i] = 0;
        }
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);

    while(cin>>s){
        if(s[0] == '0'){
            memset(used,0,sizeof(used));
            flag = 0;
            used[1] = 1;
            dfs(1);
            if(flag){
                cout<<"Yes."<<endl;
            }
            else{
                cout<<"No."<<endl;
            }
            for(int i = 0;i <26;i++){
                a[i].clear();
            }
        }
        else{
            a[s[0]-'a'].push_back(s[strlen(s)-1]-'a');
        }

    }

    return 0;
}