Leetcode 每日一题 - 2078. 两栋颜色不同且距离最远的房子

非标准解法提醒:

这道题其实考的是贪心来着

题面

街上有 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 栋颜色不同的房子

代码

class Solution {
public:
    int maxDistance(vector<int>& colors) {
        unordered_map<int, int> map_min;
        unordered_map<int, int> map_max;
        for(int i=0;i<colors.size();i++){
            if(!map_min.count(colors[i])){  // 仅记录一次
                // printf("found minnum %d at %d\n",colors[i],i);
                map_min[colors[i]] = i;
            }
            map_max[colors[i]] = i;
        }
        int ans = 0;
        for(const pair<int, int>& it:map_min){
            for(const pair<int, int>& _it:map_max){
                if(it.first == _it.first){  // 避免重复计算算出岔子
                    continue;
                }
                // printf("map_min[%d]=%d, map_max[%d]=%d.\n",it.first,it.second,_it.first,_it.second);
                ans = max(abs(it.second - _it.second), ans);
            }
        }
        return ans;
    }
};
Just Programming With ♥️ & Peace
使用 Hugo 构建
主题 Stack 由 Jimmy 设计