最短单词距离 II

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
 
/**
 * 设计一个类，该类包含以下两个函数：
 * 1. WordDistance(String[] words) 初始化对象，传入单词列表。
 * 2. int shortest(String word1, String word2) 返回两个单词之间的最短距离。
 */
public class WordDistance {
    private Map<String, List<Integer>> wordIndices;
 
    /**
     * 初始化对象，存储每个单词的所有出现位置
     */
    public WordDistance(String[] words) {
        wordIndices = new HashMap<>();
        for (int i = 0; i < words.length; i++) {
            wordIndices.computeIfAbsent(words[i], k -> new ArrayList<>()).add(i);
        }
    }
 
    /**
     * 返回两个单词之间的最短距离
     */
    public int shortest(String word1, String word2) {
        List<Integer> indices1 = wordIndices.get(word1);
        List<Integer> indices2 = wordIndices.get(word2);
        return indices1.stream()
            .flatMapToInt(index1 -> indices2.stream().mapToInt(index2 -> Math.abs(index1 - index2)))
            .min()
            .orElse(Integer.MAX_VALUE);
    }
}