1,题目要求
输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字,例如,如果输入如下4 X 4矩阵: 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.
2,题目思路
对于这道题,要求将一个矩阵按照螺旋顺时针顺序展开为一维数组。
在展开数组时,我们可以看到,我们按照的顺序是:
- 向右->向下->向左->向上
假如说一个矩阵是m*n的,那么,首先向右走m步,然后向下走n-1步;接着向左走m-1步,之后向上走n-2步。。。
这里需要注意的是,逻辑上,第一次向右走时,走的m步,但是实际上,走的是m-1步,为了保持一致性,我们设置第一行第一次走时,并不是从matrix[0][0]开始的,而是从matrix[-1][0],当然,这个节点是不存在的,只是从逻辑上方便理解。
当我们发现需要走的步数为0时,自然就结束了。
int nr = matrix.size();
int nc = matrix[0].size();
//设定走的步数
//初始先向右走nc步,再向下走nr-1步
vector<int> nSteps {nc, nr-1};
int iDir = 0; //方向指示的下标
int ir = 0, ic = -1; //初值设置为-1,是因为要走nc步
//iDir偶数下标,说明是横向的走;奇数下标,说明是竖向走
//只要走的步骤为0,说明就结束了
//nSteps[iDir%2]表示当前要走的步骤数
while(nSteps[iDir%2]){
for(int i = 0;i<nSteps[iDir%2];i++){
ir += dirs[iDir][0]; //更新横坐标
ic += dirs[iDir][1]; //更新纵坐标
res.push_back(matrix[ir][ic]);
}
nSteps[iDir%2]--; //减少下一次该方向(实际是反方向)所要走的步数
iDir = (iDir + 1)%4;
}
return res;
}
private:
//定义方向,分别是向右、向下、向左、向上
vector<vector<int>> dirs{{0,1},{1,0},{0,-1},{-1,0}};
3,代码实现
class Solution {
public:
vector<int> printMatrix(vector<vector<int> > matrix) {
vector<int> res;
int nr = matrix.size();
if(nr == 0)
return {};
int nc = matrix[0].size();
if(nc == 0)
return {};
vector<int> nStep {nc, nr-1};
int iDir = 0;
int ir = 0, ic = -1;
while(nStep[iDir%2]){
for(int i = 0;i<nStep[iDir%2];i++){
ir += dirs[iDir][0];
ic += dirs[iDir][1];
res.push_back(matrix[ir][ic]);
}
nStep[iDir%2]--;
iDir = (iDir+1)%4;
}
return res;
}
private:
vector<vector<int>> dirs {{0,1},{1,0},{0,-1},{-1,0}};
};