Étant donné un entier positif n
, générez une matrice de taille n x n
remplie avec les éléments de 1
à n^2
dans un ordre en spirale.
Exemple 1:
Input: n = 3
Output: [[1,2,3],[8,9,4],[7,6,5]]
Exemple 2:
Input: n = 1
Output: [[1]]
1 <= n <= 20
Le principe est similaire à l'exercice 54. Spiral Matrix, mais ici nous devons remplir une matrice plutôt que de lister le parcours en spirale.
// Direction pour se déplacer dans la matrice : droite, bas, gauche, haut
vector<vector<int>> dir = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
vector<vector<int>> generateMatrix(int n) {
int size = n * n;
// Initialiser une matrice n x n avec des zéros
vector<vector<int>> ans = vector<vector<int>>(n, vector<int>(n, 0));
// Définir la position de départ et la direction initiale
int row = 0;
int col = -1;
int d = 0;
// Remplir les nombres de 1 à n*n dans la matrice
for(int i = 1; i <= size; i++){
int currD = d % 4;
// Déplacer dans la direction actuelle
row += dir[currD][0];
col += dir[currD][1];
// Vérifier si nous avons besoin de changer de direction
if(row < 0 || row == n || col < 0 || col == n || ans[row][col] != 0){
// Revenir à la position précédente
row -= dir[currD][0];
col -= dir[currD][1];
// Changer de direction
d++;
currD = d % 4;
// Déplacer dans la nouvelle direction
row += dir[currD][0];
col += dir[currD][1];
}
// Placer le nombre actuel dans la matrice
ans[row][col] = i;
}
return ans;
}
- Complexité Temporelle:
O(n * m)
- Complexité Spatiale:
O(n * m)