Question

2078. 两栋颜色不同且距离最远的房子

English Version

题目描述

街上有 n 栋房子整齐地排成一列，每栋房子都粉刷上了漂亮的颜色。给你一个下标从 0 开始且长度为 n 的整数数组 colors ，其中 colors[i] 表示第  i 栋房子的颜色。

返回 两栋 颜色 不同 房子之间的 最大 距离。

第 i 栋房子和第 j 栋房子之间的距离是 abs(i - j) ，其中 abs(x) 是 x 的绝对值。

 

示例 1：

输入：colors = [_1_,1,1,_6_,1,1,1]
输出：3
解释：上图中，颜色 1 标识成蓝色，颜色 6 标识成红色。
两栋颜色不同且距离最远的房子是房子 0 和房子 3 。
房子 0 的颜色是颜色 1 ，房子 3 的颜色是颜色 6 。两栋房子之间的距离是 abs(0 - 3) = 3 。
注意，房子 3 和房子 6 也可以产生最佳答案。

示例 2：

输入：colors = [_1_,8,3,8,_3_]
输出：4
解释：上图中，颜色 1 标识成蓝色，颜色 8 标识成黄色，颜色 3 标识成绿色。
两栋颜色不同且距离最远的房子是房子 0 和房子 4 。
房子 0 的颜色是颜色 1 ，房子 4 的颜色是颜色 3 。两栋房子之间的距离是 abs(0 - 4) = 4 。

示例 3：

输入：colors = [_0_,_1_]
输出：1
解释：两栋颜色不同且距离最远的房子是房子 0 和房子 1 。
房子 0 的颜色是颜色 0 ，房子 1 的颜色是颜色 1 。两栋房子之间的距离是 abs(0 - 1) = 1 。

 

提示：

• n == colors.length
• 2 <= n <= 100
• 0 <= colors[i] <= 100
• 生成的测试数据满足 至少 存在 2 栋颜色不同的房子

解法

方法一：暴力枚举

时间复杂度 $O(n^2)$。

class Solution:
  def maxDistance(self, colors: List[int]) -> int:
    ans, n = 0, len(colors)
    for i in range(n):
      for j in range(i + 1, n):
        if colors[i] != colors[j]:
          ans = max(ans, abs(i - j))
    return ans

class Solution {
  public int maxDistance(int[] colors) {
    int ans = 0, n = colors.length;
    for (int i = 0; i < n; ++i) {
      for (int j = i + 1; j < n; ++j) {
        if (colors[i] != colors[j]) {
          ans = Math.max(ans, Math.abs(i - j));
        }
      }
    }
    return ans;
  }
}

class Solution {
public:
  int maxDistance(vector<int>& colors) {
    int ans = 0, n = colors.size();
    for (int i = 0; i < n; ++i)
      for (int j = i + 1; j < n; ++j)
        if (colors[i] != colors[j])
          ans = max(ans, abs(i - j));
    return ans;
  }
};

func maxDistance(colors []int) int {
  ans, n := 0, len(colors)
  for i := 0; i < n; i++ {
    for j := i + 1; j < n; j++ {
      if colors[i] != colors[j] {
        ans = max(ans, abs(i-j))
      }
    }
  }
  return ans
}

func abs(x int) int {
  if x >= 0 {
    return x
  }
  return -x
}

方法二：贪心

class Solution:
  def maxDistance(self, colors: List[int]) -> int:
    n = len(colors)
    if colors[0] != colors[-1]:
      return n - 1
    i, j = 1, n - 2
    while colors[i] == colors[0]:
      i += 1
    while colors[j] == colors[0]:
      j -= 1
    return max(n - i - 1, j)

class Solution {
  public int maxDistance(int[] colors) {
    int n = colors.length;
    if (colors[0] != colors[n - 1]) {
      return n - 1;
    }
    int i = 0, j = n - 1;
    while (colors[++i] == colors[0])
      ;
    while (colors[--j] == colors[0])
      ;
    return Math.max(n - i - 1, j);
  }
}

class Solution {
public:
  int maxDistance(vector<int>& colors) {
    int n = colors.size();
    if (colors[0] != colors[n - 1]) return n - 1;
    int i = 0, j = n;
    while (colors[++i] == colors[0])
      ;
    while (colors[--j] == colors[0])
      ;
    return max(n - i - 1, j);
  }
};

func maxDistance(colors []int) int {
  n := len(colors)
  if colors[0] != colors[n-1] {
    return n - 1
  }
  i, j := 1, n-2
  for colors[i] == colors[0] {
    i++
  }
  for colors[j] == colors[0] {
    j--
  }
  return max(n-i-1, j)
}