题目描述
有一个二维矩阵 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
提示:
- <= A.length <= 20
- <= A[0].length <= 20
- A[i][j] 是 0 或 1
我的解
思路:
- 这是一个有权重的数组,从左到右权重降低,尽可能让左边多一点1
- (贪心)
通过行变换完成第一列全部置为1
通过列变换有尽可能多的1
class Solution {
public:
int matrixScore(vector<vector<int>>& A) {
int line = A[0].size();
int column = A.size();
for(int i = 0; i < column; i++ ){ //通过行变换完成第一列全部置为1
if(A[i][0] == 0){
for(int j = 0; j < line; j++){
if(A[i][j] == 0) A[i][j] = 1;
else A[i][j] = 0;
}
}
}
int onecount = 0, zerocount = 0; //通过列变换有尽可能多的1
for(int i = 0; i < line; i++){
onecount = 0, zerocount = 0;
for(int j = 0; j < column; j++){
if(A[j][i] == 0) zerocount++;
else onecount++;
}
if(zerocount > onecount){
for(int j = 0; j < column; j++){
if(A[j][i] == 0) A[j][i] = 1;
else A[j][i] = 0;
}
}
}
int sum = 0; //计算
for(int i = 0; i < column; i++){
for(int j = 0; j < line; j++){
int weight = line - j - 1;
sum += A[i][j] * (int)pow(2,weight);
}
}
return sum;
}
};
实在是太拉跨了,看到思路差不多的难兄难弟,在第二部分的时候可以少一个zerocount,直接用onecount和总长度的一半比较就好了,算onecount的时候可以直接数组求和