先看leetcode题目:
Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.
For example,
Given n = 3
,
[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
大意是说跟定n*n个整数,将他们螺旋的放入一个n*n的矩阵中,如下图,形成一个由外到内的顺时针螺旋:
比较容易想到的一种思路是按照螺旋方向行列行列交替的走,直到走到行的末尾换列,到列的末尾换行,最终每个位置都填上就结束,例如,第一行,添上1, 2, 3, 然后第三列添上4,5,让后三行6,7,一列8,二行9,结束。
>>>
>>>
>>>
>>>
按照这个逻辑就可以写代码了,一个到循环负责整个矩阵,4个小循环负责每个方向,关键地方请参照代码注释。
vector<vector<int>> generateSpiralMatrix(int n) {
vector<vector<int>> retVec(n, vector<int>(n, 1));//数据矩阵
vector<vector<int>> retVecFlag(n, vector<int>(n, 0));//辅助矩阵,标识该位置是否处理过0:未处理,1:处理了
if ((1 == n) || (0 == n))//边界判断
{
return retVec;
}
size_t count = 0;
int i = 0;
int j = 0;
while (count < n * n)//外层矩阵循环
{
while ((j < n) && (0 == retVecFlag[i][j]))//列增加方向循环,对应上面步骤1和5
{
retVec[i][j] = ++count;//对应位置填充数字
retVecFlag[i][j] = 1;//置已处理标识
++j;//继续填充改行的下一个位置
}
--j;//循环退出时,j==n,需要回退一个位置
++i;//改行已经处理完成,继续下一行
while ((i < n) && (0 == retVecFlag[i][j]))//行增加方向循环,对应上面步骤2
{
retVec[i][j] = ++count;
retVecFlag[i][j] = 1;
++i;
}
--i;
--j;
while ((j >=0) && (0 == retVecFlag[i][j]))//列减少方向循环,对应上面步骤3
{
retVec[i][j] = ++count;
retVecFlag[i][j] = 1;
--j;
}
++j;
--i;
while ((i >= 0) && (0 == retVecFlag[i][j]))//行减少方向循环,对应上面步骤4
{
retVec[i][j] = ++count;
retVecFlag[i][j] = 1;
--i;
}
++i;
++j;
}
return retVec;
}
这个就是比较容易想到的一种思路,后来在网上查了查,发现一种更加通用易懂的方法,明天再详细写一写。