Skip to content

Latest commit

 

History

History
165 lines (129 loc) · 4.03 KB

File metadata and controls

165 lines (129 loc) · 4.03 KB

English Version

题目描述

有一个二维矩阵 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,则将该行每个元素进行翻转,即 grid[i][j] ^= 1

接着,遍历每一列,统计列中元素为 1 的个数 cnt,若 cnt(1 的个数) 比 m - cnt(0 的个数) 小,则将该列进行翻转。实际过程中,并不需要对列进行翻转,只需要取 max(cnt, m - cnt),即表示 1 的个数,再乘上该位的大小 n - j - 1,即求得当前列的大小。累加每一列大小即可。

Python3

class Solution:
    def matrixScore(self, grid: List[List[int]]) -> int:
        m, n = len(grid), len(grid[0])
        for i in range(m):
            if grid[i][0] == 0:
                for j in range(n):
                    grid[i][j] ^= 1

        res = 0
        for j in range(n):
            cnt = 0
            for i in range(m):
                cnt += grid[i][j]
            res += max(cnt, m - cnt) * (1 << (n - j - 1))
        return res

Java

class Solution {
    public int matrixScore(int[][] grid) {
        int m = grid.length, n = grid[0].length;
        for (int i = 0; i < m; ++i) {
            if (grid[i][0] == 0) {
                for (int j = 0; j < n; ++j) {
                    grid[i][j] ^= 1;
                }
            }
        }
        int res = 0;
        for (int j = 0; j < n; ++j) {
            int cnt = 0;
            for (int i = 0; i < m; ++i) {
                cnt += grid[i][j];
            }
            res += Math.max(cnt, m - cnt) * (1 << (n - j - 1));
        }
        return res;
    }
}

C++

class Solution {
public:
    int matrixScore(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size();
        for (int i = 0; i < m; ++i)
        {
            if (grid[i][0] == 0)
            {
                for (int j = 0; j < n; ++j) grid[i][j] ^= 1;
            }
        }
        int res = 0;
        for (int j = 0; j < n; ++j)
        {
            int cnt = 0;
            for (int i = 0; i < m; ++i) cnt += grid[i][j];
            res += max(cnt, m - cnt) * (1 << (n - j - 1));
        }
        return res;
    }
};

Go

func matrixScore(grid [][]int) int {
	m, n := len(grid), len(grid[0])
	for i := 0; i < m; i++ {
		if grid[i][0] == 0 {
			for j := 0; j < n; j++ {
				grid[i][j] ^= 1
			}
		}
	}
	res := 0
	for j := 0; j < n; j++ {
		cnt := 0
		for i := 0; i < m; i++ {
			cnt += grid[i][j]
		}
		res += max(cnt, m-cnt) * (1 << (n - j - 1))
	}
	return res
}

func max(a, b int) int {
	if a > b {
		return a
	}
	return b
}

...