面试题 17.11. 单词距离

题目表述

有个内含单词的超大文本文件，给定任意两个不同的单词，找出在这个文件中这两个单词的最短距离(相隔单词数)。如果寻找过程在这个文件中会重复多次，而每次寻找的单词不同，你能对此优化吗?

示例：

输入：words = ["I","am","a","student","from","a","university","in","a","city"], word1 = "a", word2 = "student"
输出：1

一次遍历

顺序访问words数组，找到word1 和 word2 分别将他们下标记录下来，如果word1 和 word2下标均不是初始值，则找到在数组中的位置，更新两个字符串的位置差，找到最小的即可。

class Solution {
    public int findClosest(String[] words, String word1, String word2) {
        int index1 = -1;
        int index2 = -1;
        int res = Integer.MAX_VALUE;
        for(int i = 0; i < words.length;i++){
            if(words[i].equals(word1)){
                index1 = i;
            }else if(words[i].equals(word2)){
                index2 = i;
            }
            if(index2 != -1 && index1 != -1){
                res = Math.min(res,Math.abs(index2 - index1));
            }
        }
        return res;
    }
}

进阶问题

如果寻找过程在这个文件中会重复多次，而每次寻找的单词不同，则可以维护一个哈希表记录每个单词的下标列表。遍历一次文件，按照下标递增顺序得到每个单词在文件中出现的所有下标。在寻找单词时，只要得到两个单词的下标列表，使用双指针遍历两个下标链表，即可得到两个单词的最短距离。

class Solution {
    public int findClosest(String[] words, String word1, String word2) {
        Map<String,List<Integer>> map=new HashMap<>();
        for(int i=0;i<words.length;i++){
            map.putIfAbsent(words[i],new ArrayList<>());
            map.get(words[i]).add(i);
        }
        List<Integer> list1=map.get(word1),list2=map.get(word2);
        int i=0,j=0,ans=(int)1e5+5;
        while(i<list1.size()&&j<list2.size()){
            int a=list1.get(i),b=list2.get(j);
            ans=Math.min(ans,Math.abs(a-b));
            if(a<b){i++;}
            else{j++;}
        }
        return ans;
    }
}