题目描述
有一个二维矩阵 A 其中每个元素的值为 0 或 1 。
移动是指选择任一行或列,并转换该行或列中的每一个值:将所有 0 都更改为 1,将所有 1 都更改为 0。
在做出任意次数的移动后,将该矩阵的每一行都按照二进制数来解释,矩阵的得分就是这些数字的总和。
返回尽可能高的分数。
力扣:861. 翻转矩阵后的得分
输入:[[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,尤其是数字的高位,最好就是 1。
我们想下,不论如何翻转,第一列全为 1 的结果一定大于其它任何结果的。所以我们先把最高位为 0 的行翻转。
对行翻转后,实现了所有的行第一列全为 1 后,再对行进行翻转结果肯定会变小,所以此时要想让数字的高位为 1 ,只能进行列翻转,考虑到要使结果最大,那么 当且仅当翻转的列中 0 的个数大于 1 的个数时,翻转才会使结果变得更大 ( 想想为什么 ? )
在求和时有个技巧,例如上面示例进行翻转后可得到
1 1 1 1
1 0 0 1
1 1 1 1
sum = ( 3 * 8 ) + ( 2 * 4 ) + ( 2 * 2 ) + ( 3 * 1 ) = 39
class Solution {
public int matrixScore(int[][] A) {
int m = A.length, n = A[0].length;
int result = 0;
for (int i = 0; i < m; i++) {
// 如果第 i 行以 0 开头
if (A[i][0] == 0) {
// 行翻转
for (int j = 0; j < n; j++) {
A[i][j] ^= 1;
}
}
}
for (int j = 0; j < n; j++) {
int cnt = 0;
// 统计第 j 列有多少个 1。
for (int i = 0; i < m; i++) {
cnt += A[i][j];
}
// max(cnt, m - cnt) 表示这个位置最多有多少个 1。 1<<(n-j-1) 表示这个位置的1的大小。
result += Math.max(cnt, m - cnt) * (1 << (n - j - 1));
}
return result;
}
}