leetcode832:翻轉影象 解題
阿新 • • 發佈:2018-12-10
題目: 給定一個二進位制矩陣 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
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;
}
}