【中等】力扣算法题解析LeetCode245：最短单词距离Ⅲ

关注文末推广名片，即可免费获得本题测试源码！

题目来源：🔒LeetCode245：最短单词距离Ⅲ

问题抽象：在单词距离问题基础上扩展相同单词处理，给定字符串数组和两个单词（允许相同），计算满足特定条件的最短索引距离，满足以下核心需求：

1. 距离计算规则：

   • 当 word1 和 word2 不同时：计算两单词任意索引差的最小绝对值（同 LeetCode243）；
   • 当 word1 和 word2 相同时：计算该单词两个不同位置（i ≠ j）索引差的最小绝对值。

2. 计算约束：

   • 时间复杂度 O(n)（单次遍历数组）；
   • 空间复杂度 O(1)（仅需常数变量）；
   • 遍历时动态维护两单词的最近索引：
     • 若单词相同，需额外存储上一个有效索引；
     • 若单词不同，则分别存储各自最近索引。

3. 边界处理：

   • 输入数组非空（长度至少为 2），单词存在性由题目保证；
   • 相同单词场景：
     • 至少出现两次才可计算距离（否则无有效输出，但题目保证有解）；
     • 最小距离为 1（相邻相同单词）；
   • 特殊值：["a","b","a"] 中 ("a","a") 的最短距离为 2（非 0）。

输入：字符串数组 words，字符串 word1，字符串 word2（可相同）
输出：整数（最短距离，值域 [1, len(words)-1]）

解题思路

核心是处理两种场景：

1. word1 和 word2 不同：
   使用双指针 idx1 和 idx2 分别记录 word1 和 word2 最近出现的索引。每次更新指针后计算距离并更新最小值。
2. word1 和 word2 相同：
   使用单指针 prev 记录上一个匹配位置。当再次遇到该单词时，用当前位置与 prev 的差值更新最小距离，然后更新 prev。

亮点：

• 单次遍历：时间复杂度为 $O(n)$，$n$ 为数组长度。
• 常量空间：仅使用固定变量，空间复杂度为 $O(1)$。
• 避免重复计算：合并分支逻辑，减少冗余判断。

代码实现（Java版）🔥点击下载源码

class Solution {
    public int shortestWordDistance(String[] wordsDict, String word1, String word2) {
        int minDist = Integer.MAX_VALUE;
        int idx1 = -1; // 记录 word1 最近出现的位置
        int idx2 = -1; // 记录 word2 最近出现的位置
        
        // 判断 word1 和 word2 是否相同
        boolean isSame = word1.equals(word2);
        
        for (int i = 0; i < wordsDict.length; i++) {
            String word = wordsDict[i];
            if (isSame) {
                // 相同单词场景：用 idx1 记录连续出现的索引
                if (word.equals(word1)) {
                    if (idx1 != -1) {
                        minDist = Math.min(minDist, i - idx1);
                    }
                    idx1 = i; // 更新为当前索引
                }
            } else {
                // 不同单词场景：分别更新索引
                if (word.equals(word1)) {
                    idx1 = i;
                } else if (word.equals(word2)) {
                    idx2 = i;
                }
                // 当两个索引均有效时更新最小距离
                if (idx1 != -1 && idx2 != -1) {
                    minDist = Math.min(minDist, Math.abs(idx1 - idx2));
                }
            }
        }
        return minDist;
    }
}

代码说明

1. 变量初始化：
   • minDist 初始化为 Integer.MAX_VALUE 确保首次比较有效。
   • idx1 和 idx2 初始为 -1 表示尚未找到单词。
2. 相同单词处理：
   • 当 word1 == word2 时，只需维护一个指针 idx1。每次遇到该单词时，用当前索引与 idx1 的差值更新 minDist，再更新 idx1 = i。
3. 不同单词处理：
   • 独立更新 idx1（对应 word1）和 idx2（对应 word2）。
   • 当两者均非 -1 时，计算绝对距离并更新 minDist。
4. 边界处理：
   • 题目保证 word1 和 word2 一定在数组中，无需额外判空。