861. 翻转矩阵后的得分

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 class Solution {
 4 public:
 5     //贪心思想 
 6     //若每行的第一个数都是1 才会得分最高
 7     //所以先贪心每行的第一位数 都让它为1
 8     //此时就不能改变行了 因为改变行会导致第一位不为1
 9     //接着从第二列开始贪心列 使得每一列的1的个数最多
10     void inverseRow(vector<vector<int>>& A, int row)//翻转行
11     {
12         int col_size = A[0].size();
13         for (int j = 0; j < col_size; j++)
14             A[row][j] = A[row][j] ^ 1;
15     }
16     void inverseCol(vector<vector<int>>& A, int col)//翻转列
17     {
18         int row_size = A.size();
19         for (int i = 0; i < row_size; i++)
20             A[i][col] = A[i][col] ^ 1;
21     }
22     int computeRes(vector<vector<int>>& A)//计算结果
23     {
24         int res = 0;
25         int row = A.size(), col = A[0].size();
26         int temp = 0;
27         for (int i = 0; i < row; i++)//计算每一行
28         {
29             temp = 0;//每一行的二进制值
30             for (int j = 0; j < col; j++)
31                 temp = temp * 2 + A[i][j];
32             res += temp;
33         }
34         return res;
35     }
36     int matrixScore(vector<vector<int>>& A) {
37         int row = A.size(), col = A[0].size();
38         for (int i = 0; i < row; i++)if (!A[i][0])inverseRow(A, i);//对第一列进行贪心 翻转每行 使得第一列为全1
39         int count_1;
40         for (int j = 1; j < col; j++)//对之后的每一列进行贪心 翻转列
41         {
42             count_1 = 0;//记录当前列为1的个数
43             for (int i = 0; i < row; i++)if (A[i][j])count_1++;
44             if (count_1 <= row / 2)inverseCol(A, j);//若为1的个数小于行数的一半 则需要列翻转
45         }
46         return computeRes(A);
47     }
48 };
49 int main()
50 {
51     int m, n, temp, data;
52     cin >> m >> n;
53     vector<vector<int>> A;
54     while (m--)
55     {
56         temp = n;
57         vector<int> v;
58         while (temp--)
59         {
60             cin >> data;
61             v.push_back(data);
62         }
63         A.push_back(v);
64     }
65     cout << Solution().matrixScore(A);
66     return 0;
67 }