Convert 1D Array Into 2D Array LeetCode solution in Java

1
class Solution {
2
  public int[][] construct2DArray(int[] original, int m, int n) {
3
    if (m * n != original.length) {
4
      return new int[0][0];
5
    }
6
    int[][] answer = new int[m][n];
7
    int rCount = 0, cCount = 0, len = original.length;
8
    for (int i = 0; i < len; i++) {
9
      answer[rCount][cCount++] = original[i];
10
      if (cCount == n) {
11
        rCount++;
12
        cCount = 0;
13
      }
14
    }
15
    return answer;
16
  }
17
}