算法-Score After Flipping Matrix-翻转矩阵后的得分
1、题目描述
有一个二维矩阵 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
提示:
1 <= A.length <= 20
1 <= A[0].length <= 20
A[i][j] 是 0 或 1
2、题目分析
- 从题目描述可以看到,矩阵的每一行和每一列都可以整体的进行反转,也就是1-0的变化。
- 如果要得到最高的分数,第一列的值必须全为1,其他列的值应该1的数量大于0的数量/ —贪心
- 代码就可以从以上入手,先遍第一列全为1(需要整行整行的处理),再变每一列(每列每列的处理)
3、代码1,整合思路,步步清晰
class Solution {
public int matrixScore(int[][] A) {
for(int i = 0;i < A.length;i++){
if(A[i][0] == 0){
for(int j = 0;j < A[0].length;j++){
if(A[i][j] == 0) A[i][j] = 1;
else if(A[i][j] == 1) A[i][j] = 0;
}
}
}
for(int j = 0;j < A[0].length;j++){
int count = 0;
for(int i = 0;i < A.length;i++){
count += A[i][j];
}
if(count <= A.length / 2){
for(int i = 0;i < A.length;i++){
if(A[i][j] == 0) A[i][j] = 1;
else if(A[i][j] == 1) A[i][j] = 0;
}
}
}
int max = 0;
for(int j = 0;j < A[0].length;j++){
for(int i = 0;i < A.length;i++){
max += (A[i][j] * Math.pow(2,A[0].length - j - 1));
}
}
return max;
}
}
4、代码2,整合步骤,化简代码
- 从步骤3可以知道,在第二步的时候,可以不用真的反转,只要知道最后1的总数即可
class Solution {
public int matrixScore(int[][] A) {
for(int i = 0;i < A.length;i++){
if(A[i][0] == 0){
for(int j = 0;j < A[0].length;j++){
A[i][j]= A[i][j] == 0 ? 1 : 0;
}
}
}
int max = 0;
for(int j = 0;j < A[0].length;j++){
int count = 0;
for(int i = 0;i < A.length;i++){
count += A[i][j];
}
if(count <= A.length / 2) count = A.length - count;
max += (count * Math.pow(2,A[0].length - j - 1));
}
return max;
}
}
5、复杂度分析
- 时间复杂度:O(mn),其中 mm 为矩阵行数,nn 为矩阵列数。
- 空间复杂度:O(1)