欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【LeetCode】861. 翻转矩阵后的得分

程序员文章站 2022-07-12 12:34:17
...

原题

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

代码

class Solution {
public:
    int matrixScore(vector<vector<int>>& A) {
        int row = A.size();
        int col = A[0].size();
        double sum = row * pow(2,col-1);
        for (int i = 0; i < row; ++i)
        {
            if (A[i][0] == 0)
            {
                for (int j = 0; j < col; ++j)
                {
                    A[i][j] = 1 - A[i][j];
                }
            }
        }
        
        for (int j = 1; j < col; ++j)
        {
            int n = 0;
            for(int i = 0; i < row; ++i)
            {
                if (A[i][j] == 1)
                    ++n;
            }
            if (n > row/2)
                sum += n * pow(2,col-j-1);
            else
                sum += (row-n) * pow(2,col-j-1);
        }     
        return(sum);        
    }
};

执行用时:0 ms

相关标签: LeetCode C++