题目描述
输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字。
分析:
和spiral matrix类似(leetcode spiral matrix )。
假设矩阵为方阵(行列数为n),设置打印位置点M(xstart,ystart),每次都打印当前矩阵的最外围元素,从第一个元素开始:
1.向左打印,xstart不变,ystart+1,打印n-1个;
2.向下打印,ystart为当前矩阵的最后一列,xstart+1,打印n-1个;
3.向右打印,xstart不变,且为当前矩阵的最后一行,ystart-1,打印n-1个;
4.向上打印,ystart为当前矩阵的第一列即初始值,xstart从最后一行开始上移。
然后M(xstart,ystart)沿对角线向右下方移动,行列分别减2,此时算是打印一个新的矩阵的最外围元素,重复上述1-4步骤。
注:
打印矩阵时,虽然行列可能不一样,但和方阵同理,不过需要考虑几种特殊情况:
1. 不论方阵还是矩阵,只剩一个点时,这个时候行列都为1,四个循环一个都进不去,需要写上。
2. 行为1列大于1,循环时会出现向左重复打印,需要把这种情况去除。如:[[1,2,3,4]],四层循环打印下来为[1,2,3,4,3,2]。所以向左打印的循环中设置判断条件,如果行不为1,就正常打印,如果行为1,打印此时需打印的第一个元素即可。
3. 列为1行不为1,循环时会出现向上重复打印。如[[1],[2]],打印结果为[1,2,1]。
AC代码:
class Solution {
public:
vector<int> printMatrix(vector<vector<int> > matrix) {
int n=matrix.size();
int m=matrix[0].size();
vector<int> rst;
int xstart=0, ystart=0;
while(n>0&&m>0){
if(n==1&&m==1)
rst.push_back(matrix[xstart][ystart]);
for(int i=0;i<m-1;i++){
rst.push_back(matrix[xstart][ystart+i]);
}
for(int i=0;i<n-1;i++){
rst.push_back(matrix[xstart+i][ystart+m-1]);
}
for(int i=0;i<m-1;i++){
if(n!=1)
rst.push_back(matrix[xstart+n-1][ystart+m-1-i]);
else{
rst.push_back(matrix[xstart+n-1][ystart+m-1]);
break;
}
}
for(int i=0;i<n-1;i++){
if(m!=1)
rst.push_back(matrix[xstart+n-1-i][ystart]);
else
{
rst.push_back(matrix[xstart+n-1][ystart]);
break;
}
}
xstart++;
ystart++;
n=n-2;
m=m-2;
}
return rst;
}
};