螺旋矩阵II(数组篇)

        给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

螺旋矩阵II(数组篇)_第1张图片

 

思路:

  • 填充上行从左到右
  • 填充右列从上到下
  • 填充下行从右到左
  • 填充左列从下到上

        采用左闭右开的原则

代码:

class Solution
{
public:
    vector> generateMatrix(int n)
    {
        vector> res(n,vector(n,0));
        int startX = 0, startY = 0;
        int i, j;
        int mid = n / 2;
        int loop = n / 2;
        int count = 1;
        int offset = 1;

        while(loop--)
        {
            for(i = startX; i < n - offset; i++)
            {
                res[startY][i] = count++;
            }

            for(j = startY; j < n - offset; j++)
            {
                res[j][i] = count++;
            }

            for( ; i > startX; i--)
            {
                res[j][i] = count++;
            }

            for( ; j > startY; j--)
            {
                res[j][i] = count++;
            }

            startX++;
            startY++;
            offset++;
        }

        if(n % 2 == 1)
        {
            res[mid][mid] = count;
        }

        return res;
    }
};

你可能感兴趣的:(leetcode刷题,矩阵,算法,线性代数)