50道算法题
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
修改一下昨天的算法即可
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> Matrix(n,vector<int>(n));
int u = 0; //赋值上下左右边界
int d = Matrix.size() - 1;
int l = 0;
int r = Matrix[0].size() - 1;
int n_value=1;
while(true)
{
for(int i = l; i <= r; ++i) {Matrix[u][i]=n_value;n_value++;} //向右移动直到最右
if(++ u > d) break; //重新设定上边界,若上边界大于下边界,则遍历遍历完成,下同
for(int i = u; i <= d; ++i) {Matrix[i][r]=n_value;n_value++;} //向下
if(-- r < l) break; //重新设定有边界
for(int i = r; i >= l; --i) {Matrix[d][i]=n_value;n_value++;} //向左
if(-- d < u) break; //重新设定下边界
for(int i = d; i >= u; --i) {Matrix[i][l]=n_value;n_value++;} //向上
if(++ l > r) break; //重新设定左边界
}
return Matrix;
}
};