566. Reshape the Matrix

Question 1

Can you set up variables for mat.length and mat[0].length?

Answer 1

let m = mat.length, n = mat[0].length;

Question 2

Can you set up the restrict condition for return the original mat?

Answer 2

let m = mat.length, n = mat[0].length;
if (m * n !== r * c) return mat;

Question 3

Can you set up variable res for empty two dimensional arrays and stack for empty array?

Answer 3

let m = mat.length, n = mat[0].length;
if (m * n !== r * c) return mat;
let res = new Array(r).fill(0).map(() => new Array(c).fill(0)), stack = [];

Question 4

Can you loop through mat and push each element to the empty stack?

Answer 4

let m = mat.length, n = mat[0].length;
if (m * n !== r * c) return mat;
let res = new Array(r).fill(0).map(() => new Array(c).fill(0)), stack = [];
for (let i = 0; i < m; i++) {
for (let j = 0; j < n; j++) {
stack.push(mat[i][j]);
}
}

Question 5

Can you loop through from 0 to kth elements in stack and assign it to the empty two dimensional array res?

Answer 5

let m = mat.length, n = mat[0].length;
if (m * n !== r * c) return mat;
let res = new Array(r).fill(0).map(() => new Array(c).fill(0)), stack = [];
for (let i = 0; i < m; i++) {
for (let j = 0; j < n; j++) {
stack.push(mat[i][j]);
}
}
let k = 0;
for (let i = 0; i < r; i++) {
for (let j = 0; j < c; j++) {
res[i][j] = stack[k];
k++;
}
}

Question 6

Can you return the two dimensional array?

Answer 6

return res;