英文输入法 E100

题目描述

主管期望你来实现英文输入法单词联想功能。需求如下：

依据用户输入的单词前缀，从已输入的英文语句中联想出用户想输入的单词，按字典序输出联想到的单词序列，如果联想不到，请输出用户输入的单词前缀。

注意：

1. 英文单词联想时，区分大小写

2. 缩略形式如”don’t”，判定为两个单词，”don”和”t”

3. 输出的单词序列，不能有重复单词，且只能是英文单词，不能有标点符号

输入描述

输入为两行。

首行输入一段由英文单词word和标点符号组成的语句str；

接下来一行为一个英文单词前缀pre。

• 0 < word.length() <= 20

• 0 < str.length <= 10000

• 0 < pre <= 20

输出描述

输出符合要求的单词序列或单词前缀，存在多个时，单词之间以单个空格分割

示例1

输入：

I love you
He

输出：

He

说明：

从用户已输入英文语句”I love you”中提炼出“I”、“love”、“you”三个单词，接下来用户输入“He”，从已输入信息中无法联想到任何符合要求的单词，因此输出用户输入的单词前缀。

示例2

输入：

The furthest distance in the world, Is not between life and death, But when I stand in front of you, Yet you don't know that I love you.
f

输出：

front furthest

说明：

从用户已输入英文语句”The furthestdistance in the world, Is not between life and death, But when I stand in frontof you, Yet you dont know that I love you.”中提炼出的单词，符合“f”作为前缀的，有“furthest”和“front”，按字典序排序并在单词间添加空格后输出，结果为“front furthest”。

题解

题目中的输入较短，使用 Map存储数据，使用倒排索引，将内容单词放入 Map 中
key 为单词首字母
value 为该字母开头的单词的集合

源码Java

public class AssociatedWords {

	static Map<Character, TreeSet<String>> map = new HashMap<>();

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		String str = scanner.nextLine();
		String[] split = str.split("[^a-zA-Z]");
		for (String word : split) {
			if (word.length() > 0) {
				// 将单词放入集合
				map.computeIfAbsent(word.charAt(0), k -> new TreeSet<>()).add(word);
			}
		}
		String input = scanner.nextLine();
		TreeSet<String> strings = map.get(input.charAt(0));
		List<String> result = new ArrayList<>();
		for (String word : strings) {
			if (word.startsWith(input)) {
				result.add(word);
			}
		}
		// TreeSet 为已经排序的集合，无需再次处理
		if (result.size() > 0) {
			System.out.println(String.join(" ", result));
		} else {
			System.out.println(input);
		}
	}
}