C++中翻轉矩陣後的得分


假設我們有一個二維矩陣A,其中每個值都是0或1。這裡的一次移動包括選擇任何一行或一列,並切換該行或列中的每個值:將所有0更改為1,將所有1更改為0。現在,在進行任意次數的移動後,將此矩陣的每一行解釋為二進位制數,矩陣的分數是這些數字的總和。因此,我們的任務是找到可能獲得的最高分數。如果輸入如下:

0011
1010
1100

輸出將是39,因為在切換後,矩陣將變為:

1111
1001
1111

所以數字是 15 + 9 + 15 = 39

為了解決這個問題,我們將遵循以下步驟:

  • n := 行數,m := 列數

  • ret := n x (2^(m – 1))

  • 對於 j 從 1 到 m – 1

    • cnt := 0

    • 對於 i 從 0 到 n – 1

      • cnt := cnt + A[i, j] == A[i, 0]

    • temp := 2^(m – j – 1) * max(cnt, n – cnt)

    • ret := ret + temp

  • 返回 ret

讓我們看看下面的實現,以便更好地理解:

示例

 線上演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   int matrixScore(vector<vector<int>>& A) {
      int n = A.size();
      int m = A[0].size();
      int ret = (1 << (m - 1)) * n;
      for(int j = 1; j < m; j++){
         int cnt = 0;
         for(int i = 0; i < n; i++){
            cnt += (A[i][j] == A[i][0]);
         }
         int temp = ((1 << (m - (j + 1))) * max(cnt, n - cnt));
         ret += temp;
      }
      return ret;
   }
};
main(){
   vector<vector<int>> v = {{0,0,1,1},{1,0,1,0},{1,1,0,0}};
   Solution ob;
   cout << (ob.matrixScore(v));
}

輸入

[[0,0,1,1],[1,0,1,0],[1,1,0,0]]

輸出

39

更新於:2020年4月30日

瀏覽量:119

開啟您的職業生涯

完成課程獲得認證

開始學習
廣告