leetcode 5077. 按列翻转得到最大值等行数(Flip Columns For Maximum Number of Equal Rows)

目录

• 题目描述：
• 示例 1：
• 示例 2：
• 示例 3：
• 解法：

题目描述：

给定由若干 0 和 1 组成的矩阵 matrix，从中选出任意数量的列并翻转其上的 每个 单元格。翻转后，单元格的值从 0 变成 1，或者从 1 变为 0 。

返回经过一些翻转后，行上所有值都相等的最大行数。

示例 1：

输入：[[0,1],[1,1]]
输出：1
解释：不进行翻转，有 1 行所有值都相等。

示例 2：

输入：[[0,1],[1,0]]
输出：2
解释：翻转第一列的值之后，这两行都由相等的值组成。

示例 3：

输入：[[0,0,0],[0,0,1],[1,1,0]]
输出：2
解释：翻转前两列的值之后，后两行由相等的值组成。

提示：

• 1 <= matrix.length <= 300
• 1 <= matrix[i].length <= 300
• 所有 matrix[i].length 都相等
• matrix[i][j] 为 0 或 1

---

解法：

class Solution {
public:
    bool valid(vector<int>& lst1, vector<int>& lst2){
        int sz = lst1.size();
        int pre = lst1[0]^lst2[0];
        for(int i = 1; i < sz; i++){
            if((lst1[i]^lst2[i]) != pre){
                return false;
            }
        }
        return true;
    }
    
    int maxEqualRowsAfterFlips(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        vector<int> tag(m, 0);
        for(int i = 1; i < m; i++){
            tag[i] = i;
            for(int j = 0; j < i; j++){
                if(valid(matrix[j], matrix[i])){
                    tag[i] = tag[j];
                    break;
                }
            }
        }
        vector<int> lst(300, 0);
        for(int val : tag){
            // cout<<val<<endl;
            lst[val]++;
        }
        int res = 0;
        for(int val : lst){
            res = max(res, val);
        }
        return res;
    }
};