LeetCode 861. 翻转矩阵后的得分(贪心)
程序员文章站
2022-07-12 12:34:23
...
1. 题目
有一个二维矩阵 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/score-after-flipping-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 贪心
- 第一列的比重是最大,且比后面所有的最大和还大1
- 先把所有的第一列翻成1
- 然后,后面的列,看0多还是1多,翻成1多的情况(不必真的翻,看计数即可)
class Solution {
public:
int matrixScore(vector<vector<int>>& A) {
int m = A.size(), n = A[0].size();
int i,j,sum = 0, count;
for(i = 0; i < m; ++i)
{
if(A[i][0] == 0)//把最高位翻成1,他的比重,比后面最大和还大1
{
for(j = 0; j < n; ++j)
A[i][j] = (A[i][j] == 0 ? 1 : 0);
}
}
sum += m*(1<<(n-1));
//除第0列外,按列,取1多的方法翻转
for(j = 1; j < n; ++j)
{
count = 0;
for(i = 0; i < m; ++i)
{
if(A[i][j]==0)
count++;
}
if(count >= (m+1)/2)//0比较多,直接改成1
sum += count*(1<<(n-j-1));
else
sum += (m-count)*(1<<(n-j-1));
}
return sum;
}
};
上一篇: Token