Medium级别的题: 给定一个01矩阵,你可以任意选择翻转若干列(该列0变为1,1变为0),要求翻转后使得具有相同元素的行(即全0行或者全1行)尽可能多,返回最多有多少行可以拥有相同元素。
数据范围: 输入矩阵metrix的行数和列数都不超过300并且只由0和1组成。
例如:输入{{0, 1}, {1, 1}}, 返回1。
我们可以不做变换,则第1行(下标从0开始)是全1行。
再如,输入{{0, 1}, {1, 0}}, 返回2,我们任选一列翻转,得到一个全0行和1个全1行。
分析: 无聊题。要把一行变成全1或者全0,我们要么翻转其中为0的列,要么翻转其中为1的列。即如果我们要用0和1表示某列翻转不翻转,则要把某行变为相同的值,对应的翻转字符串其实就是这行本身,或者这行取反。于是,我们把每行对应的2个翻转字符串都统计出来“投票”,出现次数最多的那个可以使得最多行变为相同值。
代码“
class Solution {public: int maxEqualRowsAfterFlips(vector<vector<int>>& matrix) { unordered_map<string, int> have; int r = 0; for (const auto& row : matrix) { string t1, t2; for (int x : row) { t1 += x + '0'; t2 += (x ^ 1) + '0'; } r = max(r, ++have[t1]); r = max(r, ++have[t2]); } return r; }};
小优化:可以看到一个字符串和它的反出现的次数必然相同,于是我们可以只统计一个——例如让所有的操作字符串都以1开头。
代码:
class Solution {public: int maxEqualRowsAfterFlips(vector<vector<int>>& matrix) { unordered_map<string, int> have; int r = 0; for (const auto& row : matrix) { string t; for (int x : row) { t += (x == row[0] ? 1 : 0) + '0'; } r = max(r, ++have[t]); } return r; }};