LeetCode——861. 翻转矩阵后的得分

题目描述:

有一个二维矩阵 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;
    }   
}

执行结果:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/FYPPPP/article/details/114676199