1. 程式人生 > 其它 >選擇某行某列的元素_演算法題——[leetcode1072]翻轉列使得矩陣有儘可能多的有相同元素的行...

選擇某行某列的元素_演算法題——[leetcode1072]翻轉列使得矩陣有儘可能多的有相同元素的行...

技術標籤:選擇某行某列的元素

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;    }};

d98c44a7fb777e41e1d95767f607e415.png