給定一個包含 m x n 個元素的矩陣(m 行, n 列),請按照順時針螺旋順序,返回矩陣中的所有元素。
示例1:
輸入: [ [ 1, 2, 3 ], [ 4, 5, 6 ], [ 7, 8, 9 ] ] 輸出: [1,2,3,6,9,8,7,4,5]
示例2:
輸入: [ [1, 2, 3, 4], [5, 6, 7, 8], [9,10,11,12] ] 輸出: [1,2,3,4,8,12,11,10,9,5,6,7]
解決思路:首先讀取矩陣最外一圈的元素,接着依次向內。
對於M行N列的矩陣,每讀一圈余下的矩陣為(M-2)行(N-2)列,這個特點可以作為循環的終止條件。
如示例2中,首先讀取1,2,3,4,8,12,11,10,9,5 接着6,7
實現代碼如下:
public static List<Integer> spiralOrder(int[][] matrix) { int M = matrix.length; if (M == 0) return new ArrayList<Integer>(); int N = matrix[0].length; List<Integer> list = new ArrayList<>(M*N); int i,j; // 余下矩陣的行數與列數 int p = M, q = N; // 讀取一圈元素的其實點 int m = 0; while(p>0 && q>0) { // 讀取矩陣的一圈 i = m; j= m; while(j<=N-m-1) list.add(matrix[i][j++]); i = i+1; j = N-m-1; while(i<=M-m-1) list.add(matrix[i++][j]); i = M-m-1; j=j-1; // 這里的i>m很有必要,矩陣的一圈可能為單行或單列 while(i>m && j>=m) list.add(matrix[i][j--]); i = i-1; j=m; // 這里的j<N-m-1很有必要,矩陣的一圈可能為單行或單列 while(i>m && j<N-m-1) list.add(matrix[i--][j]); m++; p-=2; q-=2; } return list; }