题目描述:
有一个二维矩阵 A 其中每个元素的值为 0 或 1 。移动是指选择任一行或列,并转换该行或列中的每一个值:将所有 0 都更改为 1,将所有 1 都更改为 0。在做出任意次数的移动后,将该矩阵的每一行都按照二进制数来解释,矩阵的得分就是这些数字的总和。返回尽可能高的分数。
提示:
- 1 <= A.length <= 20
- 1 <= A[0].length <= 20
- A[i][j] 是 0 或 1
示例:
输入:[[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 的数目,就翻转该列,其他的列则保持不变。
class Solution {
public int matrixScore(int[][] A) {
int m = A.length;//行
int n = A[0].length;//列
for (int i = 0; i < m; i++) {
if (A[i][0] == 0) {
for (int j = 0; j < n; j++) {
if (A[i][j] == 0) {
A[i][j] = 1;
} else {
A[i][j] = 0;
}
}
}
}
for (int i = 1; i < n; i++) {//列
int count = 0;
for (int j = 0; j < m; j++) {//行
if (A[j][i] == 1) {
count++;
}
}
if (count <= m / 2) {
for (int k = 0; k < m; k++) {
if (A[k][i] == 0) {
A[k][i] = 1;
} else {
A[k][i] = 0;
}
}
}
}
int sum = 0;
for (int i = 0; i < m; i++) {
int ans = 0;
for (int j = 0; j < n; j++) {
ans += Math.pow(2, n - 1 - j) * A[i][j];
}
sum += ans;
}
return sum;
}
}
执行结果: