华为od 机试真题 C++ 实现【单词接龙】

题目

单词接龙的规则是：

• 可用于接龙的单词首字母必须要前一个单词的尾字母相同；
• 当存在多个首字母相同的单词时，取长度最长的单词，如果长度也相等，则取字典序最小的单词；已经参与接龙的单词不能重复使用。
• 现给定一组全部由小写字母组成单词数组，并指定其中的一个单词作为起始单词，进行单词接龙，
• 请输出最长的单词串，单词串是单词拼接而成，中间没有空格。

输入描述:

• 输入的第一行为一个非负整数，表示起始单词在数组中的索引K，0 <= K < N ；
• 输入的第二行为一个非负整数，表示单词的个数N；
• 接下来的N行，分别表示单词数组中的单词。

输出描述:

• 输出一个字符串，表示最终拼接的单词串。

示例1  输入输出示例仅供调试，后台判题数据一般不包含示例

输入

0

6

word

dd

da

dc

dword

d

输出

worddwordda

说明

先确定起始单词word，再接以d开头的且长度最长的单词dword，剩余以d开头且长度最长的有dd、da、dc，则取字典序最小的da，所以最后输出worddwordda。

示例2  输入输出示例仅供调试，后台判题数据一般不包含示例

输入

4

6

word

dd

da

dc

dword

d

输出

dwordda

说明

• 先确定起始单词dword，剩余以d开头且长度最长的有dd、da、dc，则取字典序最小的da，所以最后输出dwordda。

备注:

• 单词个数N的取值范围为[1, 20]；
• 单个单词的长度的取值范围为[1, 30]；

思路

1：回溯 广搜

输入不对
#include <iostream>
#include <sstream>
#include <math.h>
#include <algorithm>
#include <vector>
#include <unordered_set>
#include <unordered_map>
#include <queue>

using namespace std;

int ladderLength(string beginWord, string endWord, vector<string>& wordList) {
    // 将vector转成unordered_set，提高查询速度
    unordered_set<string> wordSet(wordList.begin(), wordList.end());
    // 如果endWord没有在wordSet出现，直接返回0
    if (wordSet.find(endWord) == wordSet.end()) return 0;
    // 记录word是否访问过
    unordered_map<string, int> visitMap; // <word, 查询到这个word路径长度>
    // 初始化队列
    queue<string> que;
    que.push(beginWord);
    // 初始化visitMap
    visitMap.insert(pair<string, int>(beginWord, 1));

    while(!que.empty()) {
        string word = que.front();
        que.pop();
        int path = visitMap[word]; // 这个word的路径长度
        for (int i = 0; i < word.size(); i++) {
            string newWord = word; // 用一个新单词替换word，因为每次置换一个字母
            for (int j = 0 ; j < 26; j++) {
                newWord[i] = j + 'a';
                if (newWord == endWord) return path + 1; // 找到了end，返回path+1
                // wordSet出现了newWord，并且newWord没有被访问过
                if (wordSet.find(newWord) != wordSet.end()
                        && visitMap.find(newWord) == visitMap.end()) {
                    // 添加访问信息
                    visitMap.insert(pair<string, int>(newWord, path + 1));
                    que.push(newWord);
                }
            }
        }
    }
    return 0;
}
    
int main() {
    string beginWord, endWord;
    getline(cin, beginWord);
    getline(cin, endWord);
    
    string str;
    vector<string> wordList;
    while(getline(cin, str)) {
        wordList.push_back(str);
    }
    
    int result = ladderLength(beginWord, endWord, wordList);
    cout << result << endl;
    system("pause");
    return 0;
}