1. 程式人生 > >LeetCode 54. 螺旋矩陣 Spiral Matrix(C語言)

LeetCode 54. 螺旋矩陣 Spiral Matrix(C語言)

題目描述:

給定一個包含 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]

題目解答:

方法1:遍歷

直接按照順時針順序儲存,按順時針儲存矩形的上邊、右邊、下邊、左邊元素。注意

只剩一行或者一列的情況。
執行時間4ms,程式碼如下。

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* spiralOrder(int** matrix, int matrixRowSize, int matrixColSize) {
    int i = 0, j = 0, offset = 0, offset_j = matrixColSize;
    int len_i = matrixRowSize, len_j = matrixColSize;
    int index =
0; int* result = (int*)malloc(matrixRowSize * matrixColSize * sizeof(int)); while(len_i > 0 && len_j > 0) { for(j = offset; j < offset + len_j; j++) result[index++] = matrix[offset][j]; if(len_i == 1) break; for(i = offset + 1; i <
offset + len_i - 1; i++) result[index++] = matrix[i][offset + len_j - 1]; for(j = offset + len_j - 1; j >= offset; j--) result[index++] = matrix[offset + len_i - 1][j]; if(len_j == 1) break; for(i = offset + len_i - 2; i > offset; i--) result[index++] = matrix[i][offset]; offset++; len_i -= 2; len_j -= 2; } return result; }