1. 程式人生 > >leetcode832:翻轉影象 解題

leetcode832:翻轉影象 解題

題目: 給定一個二進位制矩陣 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

解法一

用一個list儲存陣列A中的元素,用的時候依次取出來,其中關係為A[i][A.length - 1 - j] = ((la.get((i*A.length)+j))+1)%2

這種解法中我犯了個錯。我本想直接用一個與A等大小的二維陣列B來儲存計算後的值,故使用了int[][] B=A,這樣導致AB指向的同一個陣列,修改了B的值後其實A的值也發生了變動。

class Solution {
    public int[][] flipAndInvertImage(int[][] A) {
        ArrayList<Integer> la = new ArrayList<Integer>();
        for (int i = 0; i < A.length; i++) 
            for (int j = 0
; j < A.length; j++) la.add(A[i][j]); for (int i = 0; i < A.length; i++) for (int j = 0; j < A.length; j++) A[i][A.length - 1 - j] = ((la.get((i*A.length)+j))+1)%2; return A; } }

解法二

思路與一相同,均是先將A的值儲存下來,再進行計算。由於題目給定了A最大為20*20的陣列,所以陣列b的大小是可以確定的(定義一個長度不定的二維陣列比較麻煩)。不過最後還是要將b賦值給A,返回陣列A,不太好。

class Solution {
    public int[][] flipAndInvertImage(int[][] A) {
        int b[][] = new int[20][20];
        for (int i = 0; i < A.length; i++)
            for (int j = 0; j < A.length; j++)
                b[i][A.length - 1 - j] = (A[i][j]+1)%2;
        for (int i = 0; i < A.length; i++)
            for (int j = 0; j < A.length; j++)
                A[i][j] = b[i][j];
        return A;
    }
}

解法三

網上看到的解法,我就沒有自己敲了。 我想了想,第一二行可以繼續優化,兩行用int half = (A.length-1)/2;即可代替。

class Solution {
    public int[][] flipAndInvertImage(int[][] A) {
        int half = A[0].length >> 1;//矩陣的列的長度
        half = (A[0].length & 1) == 0 ? half-1:half;
        for(int i = 0; i < A.length; i++) {
            for(int j = 0; j <= half; j++) {
                int temp = A[i][j];
                A[i][j] = 1 - A[i][A[i].length-j-1];
                A[i][A[i].length-j-1] = 1 - temp;
            }
        }
        return A;
    }
}