题目:
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 ] ]
思路:
与
http://blog.csdn.net/lanxu_yy/article/details/17528505完全类似。
代码:
class Solution {
public:
vector<vector<int>> r;
int cur;
vector<vector<int>> generateMatrix(int n) {
if(n==0)
{
return *(new vector<vector<int>>());
}
r.resize(n);
for(int i=0;i<n;i++)
{
r[i].resize(n);
}
int x1=0;
int y1=0;
int y2=r.size()-1;
int x2=r[0].size()-1;
cur=1;
getOutside(true, x1, y1, x2, y2, r);
return r;
}
void getOutside(bool upside, int x1, int y1, int x2, int y2,vector<vector<int> > &matrix)
{
if(x1 == x2 && y1 == y2)
{
matrix[y1][x1] = cur++;
}
else
{
if(upside)
{
for(int i=x1;i<=x2;i++)
{
matrix[y1][i] = cur++;
}
if(y2>y1)
{
for(int i=y1+1;i<=y2;i++)
{
matrix[i][x2] = cur++;
}
}
int x3 = x2-1;
int y3 = y2;
int x4 = x1;
int y4 = y1+1;
if(x3>=x4 && y3>=y4)
{
getOutside(!upside, x3, y3, x4, y4,matrix);
}
}
else
{
for(int i=x1;i>=x2;i--)
{
matrix[y1][i] = cur++;
}
if(y1>y2)
{
for(int i=y1-1;i>=y2;i--)
{
matrix[i][x2] = cur++;
}
}
int x3 = x2+1;
int y3 = y2;
int x4 = x1;
int y4 = y1-1;
if(x3<=x4 && y3<=y4)
{
getOutside(!upside, x3, y3, x4, y4,matrix);
}
}
}
}
};