LeeCode 861. 翻转矩阵后的得分

文章目录

题目内容

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

解题思路

首先，对于一个二进制数其第n位的权值等于n-1位到0位的权值的累加和再加上一；所以我们通过行转换保证第一列所有的元素为一，对于之后的每一列，我们保证1的数量大于0的数量。

解题代码

class Solution {
public:
    int matrixScore(vector<vector<int>>& A) {
        int i=0;
        while(i<A.size())
        {
            if(!A[i][0])
            {
                int j=0;
                while(j<A[i].size())
                {
                    if(A[i][j])A[i][j]=0;
                    else    A[i][j]=1;
                    j++;
                }
            }
            i++;
        }
        i=1;
        while(i<A[0].size())
        {
            int j=0;
            int tn=0;
            int fn=0;
            while(j<A.size())
            {
                if(A[j][i])tn++;
                else    fn++;
                j++;
            }
            if(tn<fn)
            {
                j=0;
                while(j<A.size())
                {
                    if(A[j][i])A[j][i]--;
                    else    A[j][i]++;
                    j++;
                }
            }
            i++;
        }
        int sum=0;
        i=0;
        while(i<A.size())
        {
            int j=0;
            int tem=0;
            while(j<A[0].size())
            {
                tem=tem*2+A[i][j];
                j++;
            }
            sum+=tem;
            i++;
        }
        return sum;
    }
};