题目描述与示例

题目描述

单词接龙的规则是:

可用于接龙的单词首字母必须要前一个单词的尾字母相同

当存在多个首字母相同的单词时,取长度最长的单词,如果长度也相等,则取字典序最小的单词;已经参与接龙的单词不能重复使用

现给定一组全部由小写字母组成单词数组,并指定其中的一个单词作为起始单词,进行单词接龙,

请输出最长的单词串,单词串是单词拼接而成,中间没有空格

输入描述

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

备注:

单词个数N的取值范围为[1,20];

单个单词的长度的取值范围为[1,30]

输出描述

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

示例一

输入

0
6
word
dd
da
dc
dword
d

输出

worddwordda

说明

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

示例二

输入

4
6
word
dd
da
dc
dword
d

输出

暂时无法在飞书文档外展示此内容

说明

先确定起始单词dword,剩余以d开头且长度最长的有dd、da.

dc`,则取字典序最小的`da`,所以最后输出`dwordda。

解题思路

代码

Python

# 题目:2023B-单词接龙
# 分值:100
# 作者:许老师-闭着眼睛学数理化
# 算法:哈希表/排序
# 代码看不懂的地方,请直接在群上提问


from collections import defaultdict

# 输入起始索引
startIdx = int(input())
# 输入单词个数
n = int(input())

# 构建一个哈希表,用于按照单词首字母储存单词列表
# key为某一个首字母first_ch
# value为字母first_ch为首字母的单词列表
dic = defaultdict(list)
# 初始化答案变量ans
ans = ""
# 循环n次,输入每一个单词并储存在哈希表dic中
for i in range(n):
    # 输入单词word
    word = input()
    # 如果是起始单词,则将word储存在ans中
    if i == startIdx:
        ans += word
    else:
        # 获得单词word的首字母
        first_ch = word[0]
        # 把word储存在哈希表dic中,
        dic[first_ch].append(word)


# 需要对dic中,value储存的每一个单词列表进行排序
# 先按照单词长度从小到大排序,再按照字典序逆序排序
# 譬如以'd'为首字母的单词列表应该排序为
# 'd' : ['d', 'dd', 'dc', 'da', 'dword']
for ch in dic:
    dic[ch].sort(key = lambda x: (-len(x), x), reverse = True)


# 进行while循环
# 退出循环的条件为,ans的末尾字母ans[-1],在dic中对应的单词列表长度为0
# 即无法找到进一步单词接龙的单词
while len(dic[ans[-1]]) > 0:
    # 弹出dic[ans[-1]]的最后一个单词,作为接下来延长的单词
    following_word = dic[ans[-1]].pop()
    # 对ans进行延长
    ans += following_word

print(ans)

Java

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int startIdx = scanner.nextInt();
        int n = scanner.nextInt();

        // 读取换行符
        scanner.nextLine(); 

        HashMap<Character, List<String>> dic = new HashMap<>();
        StringBuilder ans = new StringBuilder();

        for (int i = 0; i < n; i++) {
            String word = scanner.nextLine();
            if (i == startIdx) {
                ans.append(word);
            } else {
                char firstCh = word.charAt(0);
                dic.putIfAbsent(firstCh, new ArrayList<>());
                dic.get(firstCh).add(word);
            }
        }

        for (List<String> words : dic.values()) {
            words.sort((a, b) -> {
                if (a.length() != b.length()) {
                    return Integer.compare(a.length(), b.length());
                } else {
                    return b.compareTo(a);
                }
            });
        }

        while (dic.get(ans.charAt(ans.length() - 1)) != null && !dic.get(ans.charAt(ans.length() - 1)).isEmpty()) {
            String followingWord = dic.get(ans.charAt(ans.length() - 1)).remove(dic.get(ans.charAt(ans.length() - 1)).size() - 1);
            ans.append(followingWord);
        }

        System.out.println(ans.toString());
    }
}

C++

“`C++
#include <iostream>
#include <unordered_map>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
int startIdx;
cin >> startIdx;
int n;
cin >> n;
cin.ignore();

<pre><code>unordered_map<char, vector<string>> dic;
string ans = "";

for (int i = 0; i < n; i++) {
string word;
getline(cin, word);
if (i == startIdx) {
ans += word;
} else {
char firstCh = word[0];
dic[firstCh].push_back(word);
}
}

for (auto& entry : dic) {
sort(entry.second.begin(), entry.second.end(), [](const string& a, const string& b) {
if (a.length() != b.length()) {
return a.length() < b.length();
} else {
return a > b;
}
});
}

while (!dic[ans.back()].empty()) {
string followingWord = dic[ans.back()].back();
dic[ans.back()].pop_back();
ans += followingWord;
}

cout << ans << endl;

return 0;
</code></pre>

}

“`

时空复杂度

时间复杂度:O(``NlogM + N``)。排序所花费的时间复杂度为O(N/M * MlogM) = O(NlogM) ,接龙过程的时间复杂度为O(N)

空间复杂度:O(``N``)。哈希表所需要的额外空间。

N为单词个数,M是以某个字母为首字母的单词个数。

说明

华为OD机试有三道题⽬,第⼀道和第⼆道属于简单或中等题,分值为 100 分,第三道为中等或困难题,分值为 200分,总分为 400 分。

机试分数越⾼评级越⾼,⼯资也就越⾼。

关于华为 OD 机试更加详细的介绍可以查看这篇⽂章:华为OD机考须知

关于机考题目汇总可以看这篇文章:华为OD机试真题 2023 A+B+C+D卷 + 2024新卷(Python&Java&C++)⽬录汇总(每⽇更新)