跳转至

3696. 不同单词间的最大距离 I 🔒

题目描述

给你一个字符串数组 words。

找到两个 不同 下标 i 和 j 之间的 最大距离 ，且满足以下条件：

  • words[i] != words[j]，并且
  • 距离定义为 j - i + 1。

返回所有满足条件的下标对中最大的距离。如果不存在有效的下标对，返回 0。

 

示例 1：

输入： words = ["leetcode","leetcode","codeforces"]

输出： 3

解释：

在此示例中，words[0] 和 words[2] 不相等，并且它们的最大距离为 2 - 0 + 1 = 3。

示例 2：

输入： words = ["a","b","c","a","a"]

输出： 4

解释：

在此示例中，words[1] 和 words[4] 的最大距离为 4 - 1 + 1 = 4。

示例 3：

输入： words = ["z","z","z"]

输出： 0

解释：

在此示例中，所有单词都相等，因此答案为 0。

 

提示：

  • 1 <= words.length <= 100
  • 1 <= words[i].length <= 10
  • words[i] 由小写英文字母组成。

解法

方法一：一次遍历

我们可以发现，最大距离的两个单词中至少有一个单词在数组的两端（即下标为 \(0\) 或 \(n - 1\)）。否则，假设最大距离的两个单词分别在下标 \(i\) 和 \(j\) 处，即 \(0 < i < j < n - 1\)，那么单词 \(\textit{words}[0]\) 和 \(\textit{words}[j]\) 相同，而单词 \(\textit{words}[n - 1]\) 和 \(\textit{words}[i]\) 也相同（否则距离会更大），因此单词 \(\textit{words}[0]\) 和 \(\textit{words}[n - 1]\) 不同，且它们的距离 \(n - 1 - 0 + 1 = n\) 一定大于 \(j - i + 1\)，与假设矛盾。因此，最大距离的两个单词中至少有一个单词在数组的两端。

所以，我们只需要遍历数组，计算每个单词与数组两端单词的距离，并更新最大距离。

时间复杂度 \(O(n)\)，其中 \(n\) 是数组 \(\textit{words}\) 的长度。空间复杂度 \(O(1)\)。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
class Solution:
    def maxDistance(self, words: List[str]) -> int:
        n = len(words)
        ans = 0
        for i in range(n):
            if words[i] != words[0]:
                ans = max(ans, i + 1)
            if words[i] != words[-1]:
                ans = max(ans, n - i)
        return ans

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
class Solution {
    public int maxDistance(String[] words) {
        int n = words.length;
        int ans = 0;
        for (int i = 0; i < n; ++i) {
            if (!words[i].equals(words[0])) {
                ans = Math.max(ans, i + 1);
            }
            if (!words[i].equals(words[n - 1])) {
                ans = Math.max(ans, n - i);
            }
        }
        return ans;
    }
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
class Solution {
public:
    int maxDistance(vector<string>& words) {
        int n = words.size();
        int ans = 0;
        for (int i = 0; i < n; ++i) {
            if (words[i] != words[0]) {
                ans = max(ans, i + 1);
            }
            if (words[i] != words[n - 1]) {
                ans = max(ans, n - i);
            }
        }
        return ans;
    }
};

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
func maxDistance(words []string) int {
    n := len(words)
    ans := 0
    for i := 0; i < n; i++ {
        if words[i] != words[0] {
            ans = max(ans, i+1)
        }
        if words[i] != words[n-1] {
            ans = max(ans, n-i)
        }
    }
    return ans
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
function maxDistance(words: string[]): number {
    const n = words.length;
    let ans = 0;
    for (let i = 0; i < n; i++) {
        if (words[i] !== words[0]) {
            ans = Math.max(ans, i + 1);
        }
        if (words[i] !== words[n - 1]) {
            ans = Math.max(ans, n - i);
        }
    }
    return ans;
}

评论