面试题 17.11.单词距离

https://leetcode-cn.com/problems/find-closest-lcci

• 面试题 17.11.单词距离

  • 题目描述

  • 方法 1：双指针

    • 思路

    • 复杂度分析

    • 代码

  • 方法 2：哈希表

    • 思路

    • 复杂度分析

    • 代码

题目描述

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

示例：

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

words.length <= 100000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-closest-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法 1：双指针

思路

使用双指针去找目标词：

• 当 指针l 找到 word1 时，指针r 从 指针l 的右边出发去找 word1 或者 word2；

• 如果 指针r 找到了 word2，计算距离 r - l，同时记录一个最小的距离；

• 如果 指针r 找到的还是 word1，更新 指针l 到 指针r 的位置，指针r 继续右移寻找；

复杂度分析

• 时间复杂度：$O(N)$，N 为数组长度。

• 空间复杂度：$O(1)$。

代码

JavaScript Code

/**
 * @param {string[]} words
 * @param {string} word1
 * @param {string} word2
 * @return {number}
 */
var findClosest = function (words, word1, word2) {
    const len = words.length;

    // 找到 word1 或者 word2
    const foundTarget = word => [word1, word2].includes(word);

    // e.g. 如果当前是 word1，那下一个要找的是 word2
    const getNext = cur => (cur === word1 ? word2 : word1);

    let res = len;
    let l = -1,
        r = -1,
        next = ''; // 下一个目标词

    while (r < len) {
        // 找到 word1 或者 word2
        if (foundTarget(words[r])) {
            // 如果找到的是目标词就更新 res
            if (words[r] === next && r - l < res) res = r - l;

            // 获取下一个目标词
            next = getNext(words[r]);

            l = r;
            r = r + 1;
        } else {
            r++;
        }
    }
    return res;
};

方法 2：哈希表

思路

先用一个哈希表把每个词出现的位置坐标收集起来，再用两个指针分别遍历两个目标词的坐标数组，计算最短距离。

如果寻找过程在这个文件中会重复多次，而每次寻找的单词不同，哈希表的解法更适合。

ps. 下图中 a 和 student 的坐标数组不是题目中的真实结果。

复杂度分析

• 时间复杂度：$O(N)$，N 为数组长度，遍历一次数组记录单词出现位置的时间复杂度 $O(N)$，遍历两个目标单词的位置数组时间复杂度为 $O(N)$。

• 空间复杂度：$O(N)$，N 为数组长度，用了一个哈希表来记录每个单词出现的所有位置。

代码

JavaScript Code

/**
 * @param {string[]} words
 * @param {string} word1
 * @param {string} word2
 * @return {number}
 */
var findClosest = function (words, word1, word2) {
    // 记录所有单词出现的位置
    const dict = {};
    words.forEach((w, i) => {
        dict[w] || (dict[w] = []);
        dict[w].push(i);
    });

    const indices1 = dict[word1],
        indices2 = dict[word2];
    let p1 = 0,
        p2 = 0,
        res = words.length;

    while (p1 < indices1.length && p2 < indices2.length) {
        res = Math.min(Math.abs(indices2[p2] - indices1[p1]), res);
        indices2[p2] > indices1[p1] ? p1++ : p2++;
    }
    return res;
};