给你一个下标从 0 开始的一维整数数组 original 和两个整数 m 和 n 。你需要使用 original 中 所有 元素创建一个 m 行 n 列的二维数组。
original 中下标从 0 到 n - 1 (都 包含 )的元素构成二维数组的第一行,下标从 n 到 2 * n - 1 (都 包含 )的元素构成二维数组的第二行,依此类推。
请你根据上述过程返回一个 m x n 的二维数组。如果无法构成这样的二维数组,请你返回一个空的二维数组。
示例1:
输入:original = [1,2,3,4], m = 2, n = 2
输出:[[1,2],[3,4]]
示例2:
输入:original = [1,2,3,4], m = 1, n = 2
输出:[]
示例3:
输入:original = [1,2,3,4], m = 3, n = 2
输出:[]
分析:只有当原数组长度等于m*n时才能构成,判断之后遍历赋值即可
class Solution {
public int[][] construct2DArray(int[] original, int m, int n) {
//判断长度是否够, 不够返回空数组
if(original.length != m * n){
return new int[0][];
}
//建立新数组
int[][] res = new int[m][n];
//遍历赋值
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
res[i][j] = original[j + i * n];
}
}
return res;
}
}
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-1d-array-into-2d-array