1. 程式人生 > >[leetcode]翻轉影象

[leetcode]翻轉影象

832. 翻轉影象

給定一個二進位制矩陣 A,我們想先水平翻轉影象,然後反轉影象並返回結果。

水平翻轉圖片就是將圖片的每一行都進行翻轉,即逆序。例如,水平翻轉 [1, 1, 0] 的結果是 [0, 1, 1]。

反轉圖片的意思是圖片中的 0 全部被 1 替換, 1 全部被 0 替換。例如,反轉 [0, 1, 1] 的結果是 [1, 0, 0]。

示例 1:

輸入: [[1,1,0],[1,0,1],[0,0,0]]
輸出: [[1,0,0],[0,1,0],[1,1,1]]
解釋: 首先翻轉每一行: [[0,1,1],[1,0,1],[0,0,0]];
     然後反轉圖片: [[1,0,0],[0,1,0],[1,1,1]]

示例 2:

輸入: [[1,1,0,0],[1,0,0,1],[0,1,1,1],[1,0,1,0]]
輸出: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]]
解釋: 首先翻轉每一行: [[0,0,1,1],[1,0,0,1],[1,1,1,0],[0,1,0,1]];
     然後反轉圖片: [[1,1,0,0],[0,1,1,0],[0,0,0,1],[1,0,1,0]]

說明:

  • 1 <= A.length = A[0].length <= 20
  • 0 <= A[i][j] <= 1

C++解法:

class Solution {
public:
    vector<
vector<int>> flipAndInvertImage(vector<vector<int>>& A) { int rowNum = A.size(); int column = A[0].size(); for (int i = 0; i < rowNum; i++) { for (int j = 0; j < column/2; ++j) { swap(A[i][j], A[i][column -
1 - j]); } } for (int i = 0; i < rowNum; ++i) { for (int j = 0; j < column; ++j) { if (A[i][j] == 0) { A[i][j] = 1; } else if (A[i][j] == 1) { A[i][j] = 0; } } } return A; } };