Difficulty: Medium
Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.
For example,
Given n = 3
,
[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
vector<vector<int> > generateMatrix(int n) {
vector<vector<int> > matrix;
vector<int> temp;
if(n==0)
return matrix;
for(int i = 0;i<n;i++)
temp.push_back(0);
for(int i = 0;i<n;i++)
matrix.push_back(temp);
int left = 0, up = 0, cou = 1;
int right = n-1, down = n-1;
//cout<<left<<right<<up<<down<<endl;
while(right>left&&down>up)
{
for(int i = left;i<=right;i++){
matrix[up][i] = cou;
cou++;
}
for(int i = up+1; i<down;i++){
matrix[i][right] = cou;
cou++;
}
for(int i = right;i>=left;i--){
matrix[down][i] = cou;
cou++;
}
for(int i = down-1; i>up;i--){
matrix[i][left] = cou;
cou++;
}
left++;right--;
up++;down--;
}
if(left==right&&up==down)
matrix[up][up] = cou;
return matrix;
}