2482. Difference Between Ones and Zeros in Row and Column

發表於 | 分類於

problem

solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
class Solution {
public:
    vector<vector<int>> onesMinusZeros(vector<vector<int>>& grid) {
        // 0   1   1
        // 1   0   1
        // 0   0   1
        // rows = 2 2   2
        // cols 1   1   3
        int n = grid.size(), m = grid[0].size();
        vector<int> rows(n), cols(m);
        vector<vector<int>> ret(n, vector<int>(m));
        for(int i=0;i<n;++i)
        {
            for(int j=0;j<m;++j){
                 if(grid[i][j]==1) rows[i]++;
            }
        }
        for(int j=0;j<m;++j)
        {
            for(int i=0;i<n;++i) 
                if(grid[i][j]==1) cols[j]++;
        }
        for(int i=0;i<n;++i)
        {
            for(int j=0;j<m;++j)
            {
                ret[i][j] = rows[i] + cols[j] - (m-rows[i]) - (n-cols[j]);
            }
        }
        return ret;
    }
};

analysis

  • time complexity O(nm)
  • space complexity max(O(n), O(m))