单词距离

/*
 * 思路：
 * 1. 使用流遍历单词数组，记录两个单词出现的位置。
 * 2. 使用一个列表来存储两个单词的位置。
 * 3. 使用流计算所有位置对的距离，并找出最小距离。
 */
import java.util.*;
import java.util.stream.*;

public class Solution {
    public int shortestDistance(String[] words, String word1, String word2) {
        List<Integer> positions1 = IntStream.range(0, words.length)
                                             .filter(i -> words[i].equals(word1))
                                             .boxed()
                                             .collect(Collectors.toList());
        List<Integer> positions2 = IntStream.range(0, words.length)
                                             .filter(i -> words[i].equals(word2))
                                             .boxed()
                                             .collect(Collectors.toList());
        return positions1.stream()
                         .flatMapToInt(p1 -> positions2.stream().mapToInt(p2 -> Math.abs(p1 - p2)))
                         .min()
                         .orElse(Integer.MAX_VALUE);
    }
}