算法-螺旋矩阵II

86 阅读2分钟

1,LeetCode59

给你一个正整数 n ,生成一个包含 1n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] result = new int[n][n];
        int left = 0;
        int right = n - 1;
        int top = 0;
        int button = n - 1;
        int data = 1;
        while(data <= n * n){
            for(int i = left;i <= right;i++){
                result[top][i] = data++;
            }
            top++;
            for(int i = top;i <= button;i++){
                result[i][right] = data++;
            }
            right--;
            for(int i = right;i >= left;i--){
                result[button][i] = data++;
            }
            button--;
            for(int i = button;i >= top;i--){
                result[i][left] = data++;
            }
            left++;
        }
        return result;
    }
}

2,LeetCode54

给你一个 mn 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。

public class Solution1 {
    public List<Integer> spiralOrder(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        List<Integer> result = new ArrayList();
        int left = 0;
        int right = n - 1;
        int top = 0;
        int button = m - 1;
        int data = 1;
        while (true) {
            if(data > m * n){
                break;
            }
            for (int i = left; i <= right; i++) {
                result.add(matrix[top][i]);
                data++;
            }
            if(data > m * n){
                break;
            }
            top++;
            for (int i = top; i <= button; i++) {
                result.add(matrix[i][right]);
                data++;
            }
            if(data > m * n){
                break;
            }
            right--;
            for (int i = right; i >= left; i--) {
                result.add(matrix[button][i]);
                data++;
            }
            if(data > m * n){
                break;
            }
            button--;
            for (int i = button; i >= top; i--) {
                result.add(matrix[i][left]);
                data++;
            }
            if(data > m * n){
                break;
            }
            left++;
        }
        return result;
    }
}

3,LeetCode146

给定一个二维数组 array,请返回「螺旋遍历」该数组的结果。

螺旋遍历:从左上角开始,按照 向右向下向左向上 的顺序 依次 提取元素,然后再进入内部一层重复相同的步骤,直到提取完所有元素。

class Solution{
    public int[] spiralArray(int[][] array) {
        if(array.length == 0){
            return new int[]{};
        }
        int m = array.length;
        int n = array[0].length;
        List<Integer> result = new ArrayList();
        int left = 0;
        int right = n - 1;
        int top = 0;
        int button = m - 1;
        int data = 1;
        while (true) {
            if(data > m * n){
                break;
            }
            for (int i = left; i <= right; i++) {
                result.add(array[top][i]);
                data++;
            }
            if(data > m * n){
                break;
            }
            top++;
            for (int i = top; i <= button; i++) {
                result.add(array[i][right]);
                data++;
            }
            if(data > m * n){
                break;
            }
            right--;
            for (int i = right; i >= left; i--) {
                result.add(array[button][i]);
                data++;
            }
            if(data > m * n){
                break;
            }
            button--;
            for (int i = button; i >= top; i--) {
                result.add(array[i][left]);
                data++;
            }
            if(data > m * n){
                break;
            }
            left++;
        }
        int[] arr = new int[result.size()];
        for(int i = 0;i < result.size();i++){
            arr[i] = result.get(i);
        }
        return arr;
    }
}