华为机试：单词接龙

题目描述

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

输入描述

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

输出描述

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

示例1

输入

0
6
word
dd
da
dc
dword
d

输出

worddwordda

说明 先确定起始单词word 在接dword
剩余dd da dc 则取da

示例2

输入

4
6
word
dd
da
dc
dword
d

输出

dwordda

单词个数1<N<20
单个单词的长度  1~30 

思路分析

没有复杂的知识点，注意接龙几点规则： 

• 可用于接龙的单词 首字母必须要与前一个单词的尾字母相同。
• 当存在多个首字母相同的单词时，取长度最长的单词。
• 如果长度也相等，则取字典序最小的单词。
• 已经参与接龙的单词不能重复使用。
• 输出单词串是单词拼接而成的中间没有空格。

参考代码

注：题目网上找的，参考代码是练习用，仅供参考，并不保证用例通过率。

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * 单词接龙
 */
public class Test14 {
    public static void main(String[] args) {
        // 输入
        Scanner scanner = new Scanner(System.in);
        int k = scanner.nextInt();
        int n = scanner.nextInt();
        List<String> words = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            words.add(scanner.next());
        }

        // 获取接龙
        // 不重复
        String word = words.remove(k);
        List<String> longWords = new ArrayList<>();
        while (word != null) {
            longWords.add(word);
            word = getNextWords(word, words);
        }

        // 单词串是单词拼接而成的中间没有空格
        System.out.println(String.join("", longWords));
    }

    private static String getNextWords(String startWord, List<String> words) {
        if (words == null || words.isEmpty()) {
            return null;
        }

        String result = null;
        String next;
        int index = -1;
        for (int i = 0; i < words.size(); i++) {
            next = words.get(i);
            // 可用于接龙的单词 首字母必须要与前一个单词的尾字母相同
            if (startWord.charAt(startWord.length() - 1) == next.charAt(0)) {
                if (result == null) {
                    result = next;
                    index = i;
                } else {
                    if (next.length() > result.length()) {
                        //当存在多个首字母相同的单词时，取长度最长的单词
                        result = next;
                        index = i;
                    } else if (next.length() == result.length()) {
                        //如果长度也相等，则取字典序最小的单词
                        if (next.compareTo(result) < 0) {
                            result = next;
                            index = i;
                        }
                    }
                }
            }
        }
        if (index == -1) {
            return null;
        }
        //已经参与接龙的单词不能重复使用
        return words.remove(index);
    }
}