1. 程式人生 > >LeetCode 861. 翻轉矩陣後的得分

LeetCode 861. 翻轉矩陣後的得分

clas 開頭 color 簡單的 sum 移動 etc 貪心策略 span

有一個二維矩陣 A 其中每個元素的值為 01

移動是指選擇任一行或列,並轉換該行或列中的每一個值:將所有 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. 翻轉矩陣後的得分