多重循环与边界条件。
遇到复杂问题时,可以用图形来帮助思考。由于是以从外圈到内圈的顺序依次打印,我们可以把矩阵想象成若干个圈。用一个循环来打印矩阵,每一次打印矩阵中的一个圈。
循环结束的条件
假设矩阵的行数是rows,列数十columns。打印第一圈的左上角的坐标是(1,1),第二圈左上角的坐标是(2,2),依此类推。注意到,左上角的坐标的行列坐标总是相同。所以可以在矩阵中选取最后一圈的左上角(start,start)作为分析对象。
对于一个5*5的矩阵而言,最后一圈只有一个数字,对应坐标为(2,2),发现5>2*2。对一个6*6的坐标而言,最后一圈有四个数字,左上角坐标仍然是(2*2),6>2*2依然成立。于是可以得出循环继续的条件是columns>2*start&&rows>2*start。
打印一圈
把打印一圈分为四步:第一步从左向右打印一行;第二步从上到下打印一列;第三步从右向左打印一行;第四部从下到上打印一列。每一步中,我们只需根据起始坐标与终止坐标,用一个循环就可以打印出一行或者一列。
但是,最后一圈可能退化成只有一行、只有一列甚至只有一个数字的情况。
所以我们要分析打印时,每一步的前提条件。第一步总是需要的,因为打印一圈至少有一步。如果只有一行,那么第二步就不需要了。也就是说,需要第二部的前提是终止行号大于起始行号。需要第三步打印的前提条件是圈内至少有两行两列,也就是说除了要求终止行号大于其实行号之外,还要求终止列号大于起始列号。同理,需要打印第四步的前提条件是至少有三行两列,因此要求终止行号比起始行号至少大2。
void PrintNumber(int num)
{
cout<<num<<" ";
}
void PrintMatrixOnce(const vector<vector<int> >& matrix,vector<int>& iVec,int start)
{
int endX = matrix[0].size() -start-1;
int endY = matrix.size()-start-1;
for(int i = start;i<=endX;++i)
{
iVec.push_back(matrix[start][i]);
}
if(start<endY)
{
for(int i = start+1;i<=endY;++i)
{
iVec.push_back(matrix[i][endX]);
}
}
if(start<endY && start<endX)
{
for(int i = endX-1;i>=start;--i)
{
iVec.push_back(matrix[endY][i]);
}
}
if(start<endX && endY-start>1)
{
for(int i = endY-1;i>=start+1;--i)
{
iVec.push_back(matrix[i][start]);
}
}
}
vector<int> printMatrix(vector<vector<int> > matrix) {
vector<int> ret;
if(matrix.size() == 0)
return ret;
int start = 0;
while(matrix.size()>2*start && matrix[0].size()>2*start)
{
PrintMatrixOnce(matrix,ret,start++);
}
return ret;
}