LeetCode 54. 螺旋矩陣 Spiral Matrix(C語言)
阿新 • • 發佈:2018-12-27
題目描述:
給定一個包含 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;
}