leetcode Spiral Matrix 构造螺旋矩阵

Given an integer n, 
generate a square matrix filled with 
elements from 1 to n2 in spiral order.
For example,
Given n = 3,
You should return the following matrix:
[[ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]]
思路：同上篇 螺旋输出矩阵，代码稍改

#include<iostream>
#include<vector>
using namespace std;
vector<vector<int> > build_spiral(int n)
{
    int left=0;
    int right=n-1;
    int up=0;
    int down=n-1;
    int num=1;
    vector<vector<int> > matrix(n, vector<int>(n));
    while(left<=right && up<=down)
    {
        //从左到右遍历一行
        for(int i=left;i<=right;i++)
        {
            matrix[up][i]=num++;
        }
        up++;//行下移
        //从上往下遍历一列
        for(int j=up;j<=down;j++)
        {
            matrix[j][right]=num++;
        }
        right--;//列左移
        //从右向左遍历一行
        if(up<down)//避免行重复
        {
            for(int p=right;p>=left;p--)
            {
                matrix[down][p]=num++;
            }
        }
        down--;//行上移
        //从下往上遍历一列
        if(left<right)//避免列重复
        {
            for(int q=down;q>=up;q--)
            {
                matrix[q][left]=num++;
            }
        }
        left++;//列右移
    }
    return matrix;
}

int main()
{
    int n=3;
    vector<vector<int> > v(n,vector<int>(n));
    v=build_spiral(n);
    for(auto i:v)
    {
        for(auto j:i)
        {
            cout<<j<<" ";
        }
        cout<<endl;
    }
}