LeetCode 中级 - 翻转矩阵后的得分(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

提示:

  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;
    }
};

猜你喜欢

转载自www.cnblogs.com/-xinxin/p/10630980.html