Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.
For example,
Given n =
You should return the following matrix:Given n =
3,[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]Understand the problem:
The problem gives an integer n, generate a square matrix filled in spiral order.
For example, n = 0, return empty matrix.
For n = 1, return [1]
For n = 2, return [ [1, 2]
[4, 3]
]
For n = 3, return the matrix as shown in the problem.
So the solution would be the very closed to the Spiral matrix I.
Solution:
public class Solution {
    public int[][] generateMatrix(int n) {
        int[][] result = new int[n][n];
        if (n == 0) {
            return result;
        }
        
        int start = 1;
        int row = 0;
        int col = 0;
        
        while (n > 0) {
            if (n == 1) {
                result[row][col] = start;
                break;
            }
            
            // Move right
            for (int i = 0; i < n - 1; i++) {
                result[row][col++] = start++;
            }
            
            // Move down
            for (int i = 0; i < n - 1; i++) {
                result[row++][col] = start++;
            }
            
            // Move left
            for (int i = 0; i < n - 1; i++) {
                result[row][col--] = start++;
            }
            
            // Move up
            for (int i = 0; i < n - 1; i++) {
                result[row--][col] = start++;
            }
            
            row++;
            col++;
            n -= 2;
        }
        
        return result;
    }
}
No comments:
Post a Comment