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

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

思路一:另用一個空陣列 B 陣列用於儲存翻轉後的陣列,將原序陣列的每一行倒序輸出到 B 陣列的每一行,這樣就實現了倒序輸出,然後再遍歷陣列將 0 變成 1, 1 變成 0。

程式碼:

class Solution {
    public int[][] flipAndInvertImage(int[][] A) {
        int len = A.length,k = 0;
        int[][] B = new int[len][len];
        for(int i = 0;i < A.length;i++){
            k=0;
            for(int j = A[i].length-1;j >=0 &&k < len ;j--){
                B[i][k++] = A[i][j];
            }
        }
        for(int i = 0;i < B.length;i++){
            for(int j = 0;j < B[i].length;j++){
                if(B[i][j]==1){
                    B[i][j] = 0;
                }else{
                    B[i][j] = 1;
                }
            }
        }
        return B;
    }
}

思路二:對矩陣的每一行進行操作,用兩個指標 left 和 right,利用二分的思想進行交換,然後用 1 減去每一行的元素,就實現了1 變成 0,0 變成 1。 

程式碼:

class Solution {
    public int[][] flipAndInvertImage(int[][] A) {
        
        for(int i=0;i<A.length;i++) {
            
            int left = 0;
            int right = A[i].length-1;
            
            while(left < right) {
                int temp = 1-A[i][left];
                A[i][left] = 1-A[i][right];
                A[i][right] = temp;
                left++;
                right--;
                
            }
            if(left == right) A[i][left] = 1-A[i][left];
        }
        
        return A;
    }
}