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 ]
]
Solution
class Solution
{
public:
vector<vector<int> > generateMatrix(int n)
{
vector<vector<int>> res;
for(int i=0; i<n; i++)
{
vector<int> temp(n,0);
res.push_back(temp);
}
bool left_up = true;
bool left_down = false;
bool right_up = false;
bool right_down = false;
int min_i = 0;
int max_i = n-1;
int min_j = 0;
int max_j = n-1;
int count = 1;
while(min_i<=max_i && min_j<=max_j)
{
if(left_up == true)
{
for(int j = min_j; j<=max_j; j++)
{
res[min_i][j] = count;
count++;
}
left_up = false;
right_up = true;
min_i = min_i+1;
}
else if(right_up == true)
{
for(int i=min_i; i<=max_i; i++)
{
res[i][max_j] = count;
count++;
}
max_j = max_j-1;
right_up = false;
right_down = true;
}
else if(right_down == true)
{
for(int j=max_j; j>=min_j; j--)
{
res[max_i][j] = count;
count++;
}
max_i = max_i-1;
left_down = true;
right_down = false;
}
else
{
for(int i=max_i; i>=min_i; i--)
{
res[i][min_j] = count;
count++;
}
min_j = min_j+1;
left_up = true;
left_down = false;
}
}
return res;
}
};