【Daily Code】leetcode 2078. 两栋颜色不同且距离最远的房子

class Solution {
public:
    int maxDistance(vector<int>& colors) {
        int n = colors.size();
        map<int, pair<int, int>> idx_pair;
        // map<int, vector<int>> idx;
        vector<int> cols;
        int len = 0;
        for(int i = 0; i < n; i ++){
            // idx[colors[i]].push_back(i);
            if(idx_pair[colors[i]].first == 0 && colors[0] == colors[i]) {
                ;
            }
            else if(idx_pair[colors[i]].first == 0 && colors[0] != colors[i]) { 
                idx_pair[colors[i]].first = i;
            }
            else {
                idx_pair[colors[i]].first = min(i, idx_pair[colors[i]].first);
            }
            idx_pair[colors[i]].second = max(i, idx_pair[colors[i]].second);
            vector<int>::iterator iter=std::find(cols.begin(),cols.end(), colors[i]);
            if(iter == cols.end()) {
                cols.push_back(colors[i]);
            }
        }
        // 有多个颜色，两两之间进行运算
        int s = cols.size();
        // for(int i = 0; i < s; i ++) {
        //     cout << cols[i] << endl;
        // }
        for(int i = 0; i < s - 1; i ++) {
            for(int j = i + 1; j < s; j ++) {
                // 两个区间重合或者是完全没有交集？
                int col_i = cols[i], col_j = cols[j];
                int mini = idx_pair[col_i].first, maxi = idx_pair[col_i].second;
                int minj = idx_pair[col_j].first, maxj = idx_pair[col_j].second;
                // cout << mini << " " << maxi << " " << minj << " " << maxj << endl;
                len = max(len, abs(mini - maxj));
                len = max(len, abs(minj - maxi));
                len = max(len, abs(mini - minj));
                len = max(len, abs(maxi - maxj));
            }
        }

        return len;
        
    }
};

作者：seedlesscute
链接：https://leetcode.cn/problems/two-furthest-houses-with-different-colors/solutions/2771346/problem-2078-liang-dong-yan-se-bu-tong-q-99gb/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。