螺旋算法
算法题目
题来源–LeetCode(https://leetcode-cn.com/problems/spiral-matrix)
给定一个包含 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]
算法分析
既然矩形,我们先抛开题目提到的螺旋,想象成下面这样一层层去遍历
取到左上点(row1,column1)和右下点(row2,column2),然后根据题目要求的顺序,依次遍历一圈后,将左上点和右下点向圈全移动成(row1+1,column1+1)和(row2-1,column2-1),再次遍历知道遍历完
算法
class Solution {
public static List<Integer> spiralOrder(int[][] matrix) {
List<Integer> result = new ArrayList<>();
//先拦截异常情况
if (matrix == null || matrix.length == 0)
return result;
//初始化外圈左上点(row1,column1)和右下点(row2,column2)
int row1 = 0;
int row2 = matrix.length - 1;
int column1 = 0;
int column2 = matrix[0].length - 1;
//层层遍历
while (row1 <= row2 && column1 <= column2) {
//上行
for (int i = column1; i <= column2; i++) {
result.add(matrix[row1][i]);
}
//右列
for (int i = row1 + 1; i <= row2; i++) {
result.add(matrix[i][column2]);
}
//防止剩下最后一圈时,重复输出的问题
if(column1<column2 && row1<row2) {
//下行
for (int i = column2 - 1; i >= column1; i--) {
result.add(matrix[row2][i]);
}
//左列
for (int i = row2 - 1; i >= row1 + 1; i--) {
result.add(matrix[i][column1]);
}
}
//将左上点和右下点向圈全移动成(row1+1,column1+1)和(row2-1,column2-1)
row1++;
column1++;
row2--;
column2--;
}
return result;
}
}
后言
还有一种解法是根据观察方向-顺时针转换方案,记录转换方向和是否遍历过,到达遍历过的点就切换方向,算法我这里就暂时不贴出来了,有兴趣的可以自己尝试写写