题目描述与示例

题目描述

单词接龙的规则是：

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

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

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

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

输入描述

输入的第一行为一个非负整数，表示起始单词在数组中的索引K，0 <= 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

输出

dwordda

说明

先确定起始单词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):# 输入单词wordword = input()# 如果是起始单词，则将word储存在ans中if i == startIdx:ans += wordelse:# 获得单词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_wordprint(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++

#include <iostream>
#include <unordered_map>
#include <vector>
#include <algorithm>using namespace std;int main() {int startIdx;cin >> startIdx;int n;cin >> n;cin.ignore();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;
}

时空复杂度

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

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

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

---

华为OD算法/大厂面试高频题算法练习冲刺训练

• 华为OD算法/大厂面试高频题算法冲刺训练目前开始常态化报名！目前已服务100+同学成功上岸！

• 课程讲师为全网50w+粉丝编程博主@吴师兄学算法 以及小红书头部编程博主@闭着眼睛学数理化

• 每期人数维持在20人内，保证能够最大限度地满足到每一个同学的需求，达到和1v1同样的学习效果！

• 60+天陪伴式学习，40+直播课时，300+动画图解视频，300+LeetCode经典题，200+华为OD真题/大厂真题，还有简历修改、模拟面试、专属HR对接将为你解锁

• 可上全网独家的欧弟OJ系统练习华子OD、大厂真题

• 可查看链接 大厂真题汇总 & OD真题汇总(持续更新)

• 绿色聊天软件戳 od1336了解更多