LeetCode 中级 - 翻转矩阵后的得分(861)

Posted 2021-11-24 -xinxin

有一个二维矩阵 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. 1 <= A.length <= 20
2. 1 <= A[0].length <= 20
3. A[i][j] 是 0 或 1

 

解题思路：对每一行，如果该行第一个数为0，那么翻转这一行。

　　　　　 对每一列，比较该列0和1的个数，若0的个数大于行数一半，则翻转该列。

　　　　　　先翻转列，再翻转行。

class Solution {
public:
    int matrixScore(vector<vector<int>>& A) {
       int rowindex=A.size();
       int columnindex=A[0].size();
       for(int i=0;i<rowindex;i++)
       {
           if(A[i][0]==0)
           {
               for(int j=0;j<columnindex;j++)
               {
                    A[i][j]=1-A[i][j];
               }
           }
       }
       for(int j=0;j<columnindex;j++)
       {
           int cnt=0;
           for(int i=0;i<rowindex;i++)
           {
               if(A[i][j]==1)
                   cnt++;
           }
           if(cnt<=rowindex/2)
           {
               for(int i=0;i<rowindex;i++)
              {
                  A[i][j]=1-A[i][j];
              }
           }
       }
    
        int res=0;
        for(int i=0;i<rowindex;i++)
        {
            for(int j=0;j<columnindex;j++)
                res=res+A[i][j]*pow(2,columnindex-1-j);
        }
        return res;
    }
};