LeetCode 861. 翻轉矩陣後的得分
阿新 • • 發佈:2018-10-02
clas 開頭 color 簡單的 sum 移動 etc 貪心策略 span
有一個二維矩陣 A
其中每個元素的值為 0
或 1
。
移動是指選擇任一行或列,並轉換該行或列中的每一個值:將所有 0
都更改為 1
,將所有 1
都更改為 0
。
在做出任意次數的移動後,將該矩陣的每一行都按照二進制數來解釋,矩陣的得分就是這些數字的總和。
返回盡可能高的分數。
示例:
輸入:[[0,0,1,1],[1,0,1,0],[1,1,0,0]] 輸出:39 解釋: 轉換為 [[1,1,1,1],[1,0,0,1],[1,1,1,1]] 0b1111 + 0b1001 + 0b1111 = 15 + 9 + 15 = 39
簡單的貪心策略。先保證每一行開頭都為0,然後保證每列盡可能多一些1。
class Solution {public: int matrixScore(vector<vector<int>>& A) { if(A.size() == 0 || A[0].size() == 0) { return 0; } int row = A.size(), column = A[0].size(), sum = 0; for(int i = 0; i < row; i++) { if(A[i][0] == 0) { for(int j = 0; j < column; j++) { A[i][j] = 1 - A[i][j]; } } } for(int j = 0; j < column; j++) { int zero = 0, one = 0; for(int i = 0; i < row; i++) { if(A[i][j] == 0) { zero++; }else { one++; } } if(zero > one) { for(int i = 0; i < row; i++) { A[i][j] = 1 - A[i][j]; } } } for(int i = 0; i < row; i++) { for(int j = 0; j < column; j++) { sum += A[i][j] * pow(2, column - j - 1); } } return sum; } };
LeetCode 861. 翻轉矩陣後的得分