问题描述
在 R 行 C 列的矩阵上,我们从 (r0, c0) 面朝东面开始
这里,网格的西北角位于第一行第一列,网格的东南角位于最后一行最后一列。
现在,我们以顺时针按螺旋状行走,访问此网格中的每个位置。
每当我们移动到网格的边界之外时,我们会继续在网格之外行走(但稍后可能会返回到网格边界)。
最终,我们到过网格的所有 R * C 个空间。
按照访问顺序返回表示网格位置的坐标列表。
示例 1:
输入:R = 1, C = 4, r0 = 0, c0 = 0
输出:[[0,0],[0,1],[0,2],[0,3]]
示例 2:
输入:R = 5, C = 6, r0 = 1, c0 = 4
输出:[[1,4],[1,5],[2,5],[2,4],[2,3],[1,3],[0,3],[0,4],[0,5],[3,5],[3,4],[3,3],[3,2],[2,2],[1,2],[0,2],[4,5],[4,4],[4,3],[4,2],[4,1],[3,1],[2,1],[1,1],[0,1],[4,0],[3,0],[2,0],[1,0],[0,0]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
- 1、可以把虚线部分也想象成一个矩阵,即把当前的矩阵补全了,那么所有的遍历就在一个完整的矩阵里边了
- 2、按照四个方向:向右、向下、向左、向上来遍历矩阵,每次走一格,通过每个方向走的步长即可调整方向,步长的规律:1、1、2、2、3、3、… 每两次换方向,步长+1
- 3、要判断每次遍历的元素是不是在原矩阵中,如果在则把对应的坐标存入result矩阵即可
代码实现
public int[][] spiralMatrixIII(int R, int C, int r0, int c0) {
int resultTotalLine = R * C;
int[][] visitResult = new int[resultTotalLine][2];
int step = 1;
visitResult[0][0] = r0;
visitResult[0][1] = c0;
int currentLineNo = r0;
int currentColNo = c0;
int resultLineNo = 1;
while (resultLineNo < resultTotalLine) {
//向右
for (int rightStep = 1; rightStep <= step; rightStep++) {
currentColNo += 1;
if (isInOriginMatrix(R, C, currentLineNo, currentColNo)) {
visitResult[resultLineNo][0] = currentLineNo;
visitResult[resultLineNo][1] = currentColNo;
resultLineNo++;
}
}
// 向下
for (int downStep = 1; downStep <= step; downStep++) {
currentLineNo += 1;
if (isInOriginMatrix(R, C, currentLineNo, currentColNo)) {
visitResult[resultLineNo][0] = currentLineNo;
visitResult[resultLineNo][1] = currentColNo;
resultLineNo++;
}
}
//向右 向下调整方向两次之后,step+1
step++;
// 向左
for (int leftStep = 1; leftStep <= step; leftStep++) {
currentColNo -= 1;
if (isInOriginMatrix(R, C, currentLineNo, currentColNo)) {
visitResult[resultLineNo][0] = currentLineNo;
visitResult[resultLineNo][1] = currentColNo;
resultLineNo++;
}
}
//向上
for (int upStep = 1; upStep <= step; upStep++) {
currentLineNo -= 1;
if (isInOriginMatrix(R, C, currentLineNo, currentColNo)) {
visitResult[resultLineNo][0] = currentLineNo;
visitResult[resultLineNo][1] = currentColNo;
resultLineNo++;
}
}
step++;
}
return visitResult;
}
private boolean isInOriginMatrix(int originLineNum, int originColNum, int currentLineNo, int currentColNo) {
if (currentLineNo >= 0 && currentLineNo < originLineNum && currentColNo >= 0 && currentColNo < originColNum) {
return true;
}
return false;
}