Spiral Matrix
程序员文章站
2022-07-12 09:30:09
...
好久没写文章了...
自罚30大板
在刷leetcode
就继续更新吧
Q:Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.
eg 1:
Input:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
Output: [1,2,3,6,9,8,7,4,5]
eg 2:
Input:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
Output: [1,2,3,4,8,12,11,10,9,5,6,7]
题目很好理解,就是把一个矩阵螺旋着输出。
看到这个题目,我首先就想着使用深度优先进行搜索。这有点像迷宫的搜索,我们可以定义搜索规则如下:
定义一个辅助矩阵,判断这个cell是否被搜索过。
eg
[
[1, 1, 1],
[1, 1, 1],
[1, 1, 1]
]
在这里,1代表未被搜索过,0代表已经被搜索过。
当我们搜索完第一行,matrix变成
[
[0, 0, 0 ],
[1, 1, 1 ],
[1, 1, 1]
]
这时候我们只有一个方向可以继续往下走,就是朝下,因为上下左右都是无法走的(0也是无法走的),类似地,可以一直走到这个状态
[
[0, 0, 0 ],
[0, 1, 0 ],
[0, 0, 0 ]
]
然后就只剩1可以走啦,也就是我们的搜索结束了ORZ
也就是说我们需要定义的变量有以下几个:
1 现在前进的方向direction当只有一个方向可以前行的时候进行改变
2 保存答案的数组
3 现在的位置,以便于下一次进行Move的时候方便找到初始位置
4 状态数组
然后就开始愉快地撸码。
/**
direction: 方向
cur_i,cur_j:现在的位置
recMatrix: 状态数组
ans: 保存答案的数组
**/
class Solution {
void move(int direction,int cur_i, int cur_j, vector<vector<int>>& matrix, vector<vector<int>>& recMatrix, vector<int>& ans){
if(direction == 0){
if( cur_j >= matrix[0].size() || recMatrix[cur_i][cur_j] == 0 ) return; //先判断是否还有路可走,没有的话返回
for( int j = cur_j; ;j++){ //然后开始走,并且记录答案
if( j < matrix[cur_i].size() && recMatrix[cur_i][j] != 0 ){
cout << matrix[cur_i][j] << endl;
ans.push_back(matrix[cur_i][j]);
recMatrix[cur_i][j] = 0;
cur_j =j;
}
else break;
}
direction = 1;
move(direction, cur_i + 1, cur_j , matrix, recMatrix, ans);
}
else if(direction == 1){
if( cur_i >= matrix.size() || recMatrix[cur_i][cur_j] == 0 ) return;
for( int i = cur_i; ;i++){
if( i < matrix.size() && recMatrix[i][cur_j] != 0 ){
ans.push_back(matrix[i][cur_j]);
recMatrix[i][cur_j] = 0;
cur_i =i;
}
else break;
}
direction = 2;
move(direction, cur_i , cur_j - 1, matrix, recMatrix, ans);
}
else if(direction == 2){
if( cur_j < 0 || recMatrix[cur_i][cur_j] == 0 ) return;
for( int j= cur_j;; j--){
if( j >= 0 && recMatrix[cur_i][j] != 0 ){
ans.push_back(matrix[cur_i][j]);
recMatrix[cur_i][j] = 0;
cur_j =j;
}
else break;
}
direction = 3;
move(direction, cur_i - 1, cur_j , matrix, recMatrix, ans);
}
else if(direction == 3){
if( cur_i < 0 || recMatrix[cur_i][cur_j] == 0 ) return;
for( int i = cur_i; ;i--){
if( i >= 0 && recMatrix[i][cur_j] != 0 ){
ans.push_back(matrix[i][cur_j]);
recMatrix[i][cur_j] = 0;
cur_i =i;
}
else break;
}
direction = 0;
move(direction, cur_i , cur_j + 1, matrix, recMatrix, ans);
}
}
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
vector<int> res;
// 定义的状态举证
vector<vector<int>> recMatrix;
if(matrix.size() ==0 ) return res;
// 初始化
for( int i = 0; i< matrix.size(); i++){
vector<int> tempMatrix;
for( int j = 0; j<matrix[i].size(); j++ ){
tempMatrix.push_back(1);
}
recMatrix.push_back(tempMatrix);
}
// direction 0:right 1:down 2:left 3:up
int direction = 0;
move(0, 0, 0, matrix, recMatrix, res);
return res;
}
};
提交结果 AC beats 100% 看来只有这一个解了。其实这个状态数组完全是不需要的,但是我太懒了,AC了就不想改了,读者也可是实现抛弃这个状态数组进行实现。
上一篇: Spiral Matrix
下一篇: 剑指Offer算法