2022-05-27每日一题

面试题 17.11. 单词距离

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

示例：

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

words.length <= 100000

思路：水题，直接模拟过去就行，可以先用一个HashMap建立每个单词的位置列表，然后再循环两个key值所出现的位置，计算距离，取最小值

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++){
            if(map.containsKey(words[i])){
               map.get(words[i]).add(i);
            }
            else{
                List<Integer> list = new ArrayList<>();
                list.add(i);
                map.put(words[i],list);
            }
        }
        int d = Integer.MAX_VALUE;
        List<Integer> w1 = map.get(word1);
        List<Integer> w2 = map.get(word2);
        for(int i = 0; i < w1.size(); i++){
            for(int j = 0; j < w2.size(); j++){
                d = Math.min(Math.abs(w1.get(i) - w2.get(j)), d);
            }
        }
        return d;
    }
}