題目概述
給定一個二進制矩陣 A，我們想先水平翻轉影像，然后反轉影像并回傳結果
水平翻轉圖片就是將圖片的每一行都進行翻轉，即逆序，例如，水平翻轉 [1, 1, 0] 的結果是 [0, 1, 1]
反轉圖片的意思是圖片中的 0 全部被 1 替換， 1 全部被 0 替換，例如，反轉 [0, 1, 1] 的結果是 [1, 0, 0]
示例如下：
輸入：[[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]]
解法思路
最直觀的做法：首先對矩陣 A 的每一行進行水平翻轉操作，然后對矩陣中的每個元素進行反轉操作，該做法需要遍歷矩陣兩次
class Solution {
public int[][] flipAndInvertImage(int[][] A) {
for(int i = 0; i < A.length; i++) {
reverse(A[i]);
for(int j = 0; j < A[i].length; j++) {
A[i][j] ^= 1;
}
}
return A;
}
private void reverse(int[] array) {
int low = 0, high = array.length - 1;
while(low < high) {
int temp = array[low];
array[low] = array[high];
array[high] = temp;
low++;
high--;
}
}
}
如果只希望遍歷一次，就要做到在翻轉行的同時反轉元素，在翻轉行時，我們可以同時修改了左右對稱的兩個元素，其規律如下：
- 如果兩個元素相同，那么同時進行反轉，例如兩個元素都為 1 或 0，很顯然，在進行了翻轉和反轉操作后，兩個元素都將變為 0 或 1
- 如果兩個元素不相同，那么保持不變，例如兩個元素分別為 [1,0] 或 [0,1]，很顯然，在進行了翻轉和反轉操作后，兩個元素的值不會發生變化
class Solution {
public int[][] flipAndInvertImage(int[][] A) {
for (int i = 0; i < A.length; i++) {
int low = 0, high = A[i].length - 1;
while (low < high) {
if (A[i][low] == A[i][high]) {
A[i][low] ^= 1;
A[i][high] ^= 1;
}
low++;
high--;
}
if (low == high) {
A[i][low] ^= 1;
}
}
return A;
}
}
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/263249.html
標籤：其他
上一篇：圖示加代碼 搞懂線性表（一）