最短单词距离

/*
 * 思路：
 * 1. 使用IntStream来遍历数组下标。
 * 2. 通过filter找到目标单词1和单词2的位置。
 * 3. 计算所有位置之间的距离，并找到最小值。
 */
 
import java.util.stream.IntStream;
 
public class Solution {
    public int shortestDistance(String[] words, String word1, String word2) {
        int[] indices1 = IntStream.range(0, words.length)
                                   .filter(i -> words[i].equals(word1))
                                   .toArray();
        int[] indices2 = IntStream.range(0, words.length)
                                   .filter(i -> words[i].equals(word2))
                                   .toArray();
        return IntStream.of(indices1)
                        .flatMap(i1 -> IntStream.of(indices2).map(i2 -> Math.abs(i1 - i2)))
                        .min()
                        .orElse(Integer.MAX_VALUE);
    }
}