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

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

输入：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) {
        vector<int> pos(101, -1); // 记录每种颜色的最左边的位置
        int ans = 0;
        for(int i = 0; i < colors.size(); ++i)
        {
            if(pos[colors[i]] == -1)
                pos[colors[i]] = i;
            for(int j = 0; j <= 100; ++j)
            {
                if(j != colors[i] && pos[j] != -1)
                // 颜色不相同，且 存在
                    ans = max(ans, i-pos[j]);
            }
        }
        return ans;
    }
};

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