祝大家元旦快乐!!!
java:
这道题和566.重塑矩阵相似
/ 列数:可以取到每一行;%列数:可以取到每一列
class Solution {
public int[][] construct2DArray(int[] original, int m, int n) {
int l = original.length;
if (l != m * n) {
return new int[0][];
}
int [][] ans = new int[m][n];
for(int i = 0; i < l; i++){
ans[i / n][i % n] = original[i];
}
return ans;
}
}
python3:
class Solution:
def construct2DArray(self, original: List[int], m: int, n: int) -> List[List[int]]:
l = len(original)
if l != m * n:
return []
ans = [[0 for _ in range(n)] for _ in range(m)]
for i in range(l):
ans[i // n][i % n] = original[i]
return ans