写点什么

【LeetCode】单词距离 Java 题解

作者:HQ数字卡
  • 2022 年 5 月 27 日
  • 本文字数:1271 字

    阅读完需:约 4 分钟

题目描述

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


示例:
输入:words = ["I","am","a","student","from","a","university","in","a","city"], word1 = "a", word2 = "student"输出:1
来源:力扣(LeetCode)链接:https://leetcode.cn/problems/find-closest-lcci著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
复制代码

思路分析

  • 今天的算法题目是单词距离问题,题目以数组的形式展现,需要求解的是两个单词之间的最短距离。我们观察单词之间示例,最短距离即位当前的两个单词之间的索引的差值。

  • 很容易想到的解法是朴素解法,我们依次遍历 words 中的每一个单词,使用 hashMap 记录每一个 word1, word2 每一个单词的索引位置。把每个位置记录之后,我们可以采用循环遍历的方式,求出最小的单词距离。

  • 朴素算法通过之后,发现算法时间复杂度较高,很多的重复计算。我们可以优化一下思路,从 0 位置开始遍历数组,分别使用 idx1, idx2 记录 word1, word2 的位置,当找到两个位置的时候,开始比较记录最短的距离。实现代码如下,供参考。

通过代码

  • 朴素解法


class Solution {    public int findClosest(String[] words, String word1, String word2) {        Map<String, List<Integer>> map = new HashMap<>();        int n = words.length;        for (int i = 0; i < n; i++) {            List<Integer> idxs = map.getOrDefault(words[i], new ArrayList<>());            idxs.add(i);            map.put(words[i], idxs);        }
int ans = Integer.MAX_VALUE;
List<Integer> word1Idxs = map.getOrDefault(word1, new ArrayList<>()); List<Integer> word2Idxs = map.getOrDefault(word2, new ArrayList<>());
for (int j = 0; j < word1Idxs.size(); j++) { for (int k = 0; k < word2Idxs.size(); k++) { ans = Math.min(ans, Math.abs(word1Idxs.get(j) - word2Idxs.get(k))); } } return ans; }}
复制代码


  • 优化解法


class Solution {    public int findClosest(String[] words, String word1, String word2) {        int n = words.length;        int ans = Integer.MAX_VALUE;        int idx1 = -1;        int idx2 = -1;        for (int i = 0; i < n; i++) {            if (words[i].equals(word1)) {                idx1 = i;            } else if (words[i].equals(word2)) {                idx2 = i;            }            if (idx1 >= 0 && idx2 >= 0) {                ans = Math.min(ans, Math.abs(idx1 - idx2));            }        }
return ans; }}
复制代码

总结

  • 朴素算法的时间复杂度是 O(n * n), 空间复杂度是 O(n)

  • 朴素算法的时间复杂度是 O(n), 空间复杂度是 O(1)

  • 坚持算法每日一题,加油!

发布于: 刚刚阅读数: 4
用户头像

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

评论

发布
暂无评论
【LeetCode】单词距离Java题解_LeetCode_HQ数字卡_InfoQ写作社区