今天是2022年的第一天,祝大家新年快乐,天天好心情!

题目

这是力扣2022年的第一题,没有什么难度,用来开开心心的开启2022再适合不过啦。题号也是2022,力扣有心了。

将一维数组转变成二维数组,力扣2022第一题,祝大家新年快乐_二维数组

思路

这道题十分简单,只需要注意一下入参的校验。
我们只需要定义好二维数组,然后遍历二维数组,因为二维数据的元素个数等于一维数组的元素个数,在遍历的时候实现赋值操作就好。

代码

class Solution {
    public int[][] construct2DArray(int[] original, int m, int n) {
        if (Objects.isNull(original) || original.length == 0 || m < 1 || n < 1 || original.length != m * n) {
            return new int[0][0];
        }
        final int[][] result = new int[m][n];
        for (int i = 0, index = 0;i < m;i++) {  
            for (int j = 0;j<n;j++) {
                result[i][j] = original[index++];
            }
        }
        return result;
    }
}

时间复杂度:O(m * n)
空间复杂度:O(m * n)