螺旋矩阵II

螺旋矩阵II

//螺旋矩阵II
/*给定一个整数n，将数字1到n^2按螺旋的顺序填入n×n的矩阵
例如：
给出的n=3，
你应该返回如下矩阵：
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]*/
#include<vector>
using namespace std;
class Solution {
public:
	/**
	*
	* @param n int整型
	* @return int整型vector<vector<>>
	*/
	vector<vector<int> > generateMatrix(int n) {
		vector<vector<int>> res(n, vector<int>(n));
		if (n < 1)
			return res;
		int index = 1, rowStart = 0, colStart = 0, rowEnd = n - 1, colEnd = n - 1;
		while (index <= (n*n))
		{
			for (int i = colStart; i <= colEnd; ++i)
			{
				res[rowStart][i] = index++;
			}
			for (int i = rowStart + 1; i <= rowEnd; ++i)
			{
				res[i][colEnd] = index++;
			}
			for (int i = colEnd - 1; i >= colStart; --i)
			{
				res[rowEnd][i] = index++;
			}
			for (int i = rowEnd - 1; i > rowStart; --i)
			{
				res[i][colStart] = index++;
			}
			rowStart++;
			colStart++;
			rowEnd--;
			colEnd--;
		}
		return res;
	}
};