Completa el cuadrado (matriz)

Desafortunadamente, el mundo matemático está sesgado a favor de las matrices cuadradas. Por ello, este desafío ayudará a las matrices rectangulares convirtiéndolas en cuadradas.

Por ejemplo, para la matriz:

[
  [1, 2],
  [3, 4],
  [5, 6]
]

Esto se puede hacer rellenándola con una columna de ceros a la derecha para obtener:

[
  [1, 2, 0],
  [3, 4, 0],
  [5, 6, 0]
]

Mientras que para la matriz:

[
  [1, 2, 3, 4],
  [5, 6, 7, 8]
]

Se puede rellenar con dos filas de ceros en la parte inferior para obtener:

[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [0, 0, 0, 0],
  [0, 0, 0, 0]
]

Escribe una función que rellene una matriz rectangular con ceros a la derecha o en la parte inferior para convertirla en cuadrada.

Ejemplos

complete_square([
  [2, 5]
]) ➞ [
  [2, 5],
  [0, 0]
]

complete_square([
  [2, 5],
  [1, 7]
]) ➞ [
  [2, 5],
  [1, 7]
]

complete_square([
  [1, 2],
  [3, 4],
  [5, 6]
 ]) ➞ [
  [1, 2, 0],
  [3, 4, 0],
  [5, 6, 0]
]

Notas

  • Las matrices deben rellenarse a la derecha o en la parte inferior, pero no en ambas direcciones simultáneamente (es decir, el tamaño de la dimensión mayor no debe cambiar).
  • Si la matriz de entrada ya es cuadrada, simplemente devuelve esa matriz.