华为OD机试 - 单词接龙 - 数据结构map、list （Java 2023 B卷 100分）

在这里插入图片描述

- 专栏导读
- 一、题目描述
- 二、输入描述
- 三、输出描述
- 四、输入示例
- - 1、输入：
  - 2、输出
  - 3、说明
- 五、解题思路
- - 1、核心思想：
  - 2、核心算法是构建一个map：
- 六、Java算法源码
- 七、效果展示
- - 1、输入
  - 2、输出
  - 3、说明
  - 4、没有移除后再次拼接的情况，改造一下

华为OD机试 2023B卷题库疯狂收录中，刷题点这里

专栏导读

本专栏收录于《华为OD机试（JAVA）真题（A卷+B卷）》。

刷的越多，抽中的概率越大，每一题都有详细的答题思路、详细的代码注释、样例测试，发现新题目，随时更新，全天CSDN在线答疑。

一、题目描述

单词接龙的规则是：

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

现给定一组全部由小写字母组成单词数组，并指定其中的一个单词作为起始单词，进行单词接龙，请输出最长的单词串，单词串是单词拼接而成，中间没有空格。

二、输入描述

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

备注：

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

三、输出描述

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

四、输入示例

1、输入：

0
6
word
dd
da
dc
dword
d

2、输出

worddwordda

3、说明

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

五、解题思路

1、核心思想：

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

2、核心算法是构建一个map：

key：每个字符串的第一个字符；
value：以key开头的字符串，并按字符串长度、字典顺序排序

如果接龙字符串builder的末字符，在map中存在，表示有以builder末字符开头的字符串，继续循环；
获取以“末字符”开头的字符串集合；
获取排序后的第一个；
添加到接龙字符串builder；
在map中移除已经加入接龙的字符串；
如果此开头的字符串还有其他字符串，再加入map，否则移除；
获取接龙字符串builder末字符，进行下次判断；
输出接龙字符串builder

六、Java算法源码

package com.guor.od;import java.util.*;public class OdTest03 {/*** 单词接龙* 核心思想：** 1. 可用于接龙的单词首字母必须要与前一个单词的尾字母相同；* 2. 当存在多个首字母相同的单词时，取长度最长的单词；* 3. 如果长度也相等，则取字典序最小的单词；* 4. 已经参与接龙的单词不能重复使用。*/public static void main(String[] args) {Scanner scanner = new Scanner(System.in);// 起始单词在数组中的索引Kint K = Integer.parseInt(scanner.nextLine());// 单词的个数Nint N = Integer.parseInt(scanner.nextLine());// 接下来的N行，分别表示单词数组中的单词List<String> wordList = new ArrayList<>();for (int i = 0; i < N; i++) {wordList.add(scanner.nextLine());}// 获取第一个单词String first = wordList.get(K);// 已经参与接龙的单词不能重复使用，故移除wordList.remove(K);// 最终拼接的单词串StringBuilder builder = new StringBuilder();builder.append(first);/*** key：每个字符串的第一个字符* value：以key开头的字符串，并按字符串长度排序*/Map<String,List<String>> listMap = getListMap(wordList);System.out.println(listMap);// 获取接龙字符串的最后一个字符，作为待接龙字符串的第一个字符String last = builder.substring(builder.length() - 1);// 如果接龙字符串builder的末字符，在map中存在，表示有以builder末字符开头的字符串，继续循环while (listMap.containsKey(last)) {// 获取以“末字符”开头的字符串集合List<String> tempList = listMap.get(last);// 获取排序后的第一个String firstWord = tempList.get(0);// 添加到接龙字符串builderbuilder.append(firstWord);// 在map中移除已经加入接龙的字符串tempList.remove(firstWord);// 如果此开头的字符串还有其他字符串，再加入map，否则移除if(tempList.size()>0){listMap.put(last, tempList);}else {listMap.remove(last);}// 获取接龙字符串builder末字符，进行下次判断last = builder.substring(builder.length() - 1);}// 输出接龙字符串builderSystem.out.println(builder);}/*** key：每个字符串的第一个字符* value：以key开头的字符串，并按字符串长度排序*/public static Map<String,List<String>> getListMap(List<String> wordList) {Map<String,List<String>> listMap = new HashMap<>();for(String word : wordList){String first = String.valueOf(word.charAt(0));List<String> list = null;if(listMap.containsKey(first)){list = listMap.get(first);list.add(word);sortList(list);}else{list = new ArrayList<>();list.add(word);}listMap.put(first,list);}return listMap;}/*** 排序规则：* 1. 当存在多个首字母相同的单词时，取长度最长的单词* 2. 如果长度也相等，则取字典序最小的单词*/private static void sortList(List<String> wordList){wordList.sort(new Comparator<String>() {@Overridepublic int compare(String o1, String o2) {if(o1.length()>o2.length()){return -1;}else if(o1.length()<o2.length()){return 1;}else{// 字典排序return o1.compareTo(o2);}}});}
}