题目
输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字,例如,如果输入如下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.
思路
- 顺时针打印矩阵,粗略的想象可以分为四步,对于每一个圈,
- 从左到右打印第一行。
- 从上到下打印最后一列。
- 从右到左打印最后一行。
- 从下到上打印第一列。
- 开始下一圈,继续上面四步。
- 所以关键在于圈的开始坐标。循环何时停止。
犯的错
没有考虑几种情况。
- 每次考虑一个圈。
- 如果起始地址是合法的,那么第一行是肯定能被打印的,所以不需要条件判断。
- 打印最后一列,如果只有一行,那么最后一列就一个元素,那么不打印。
- 打印最后一行,保证行数大于1,避免重复打印一行的元素。
- 打印第一列,保证列数大于1,避免重复打印一列的元素。
代码
class Solution {
public:
vector<int> printMatrix(vector<vector<int> > matrix) {
if ( matrix.size() == 0 ) return {};
vector<int> res;
int rows = matrix.size();
int columns = matrix[0].size();
int start = 0;
int minSize = (rows < columns ? rows : columns);
while ( start < (minSize+1)/2 ) {
// 通过start确定end
int endX = rows - start - 1;
int endY = columns - start - 1;
// 从左到右
for ( int y = start; y <= endY; ++y ) {
res.push_back( matrix[start][y] );
}
if ( start < endX ) { // 如果只有一行,不运行
// 从上到下
for ( int x = start + 1; x <= endX; ++ x ) {
res.push_back( matrix[x][endY] );
}
}
if ( start < endX && start < endY ) {
// 从右向左
for ( int y = endY - 1; y >= start; --y ) {
res.push_back( matrix[endX][y] );
}
}
if ( start < endY && start < endX - 1 ) {
// 从下到上
for ( int x = endX - 1; x > start; --x ) {
res.push_back( matrix[x][start] );
}
}
++start;
}
return res;
}
};
第二次写的
class Solution {
public:
vector<int> printMatrix(vector<vector<int> > matrix) {
vector<int> res;
int Y = matrix.size();
if ( Y == 0 ) return res;
int X = matrix[0].size();
int limit = min( (X+1)/2, (Y+1)/2 );
for ( int start = 0; start < limit; ++start ) {
int endX = X - start - 1;
int endY = Y - start - 1;
for ( int x = start; x <= endX; ++x )
res.push_back( matrix[start][x] );
for ( int y = start + 1; y <= endY; ++y )
res.push_back( matrix[y][endX] );
if ( start < endY ) {
for ( int x = endX - 1; x >= start; --x )
res.push_back( matrix[endY][x] );
}
if ( start < endX && start < endY - 1 ) {
for ( int y = endY - 1; y > start; --y )
res.push_back( matrix[y][start] );
}
}
return res;
}
};