Convert an Array Into a 2D Array With

lcpy gen -n 2610   # by problem number
lcpy gen -s convert_an_array_into_a_2d_array_with_conditions   # by problem name

Input: nums = [1,3,4,1,2,3,1]
Output: [[1,3,4,2],[1,3],[1]]
Explanation: We can create a 2D array that contains the following rows:
- 1,3,4,2
- 1,3
- 1
All elements of nums were used, and each row of the 2D array contains distinct integers, so it is a valid answer.
It can be shown that we cannot have less than 3 rows in a valid array.

Input: nums = [1,2,3,4]
Output: [[4,3,2,1]]
Explanation: All elements of the array are distinct, so we can keep all of them in the first row of the 2D array.

from collections import Counter


class Solution:
    # Time: O(n)
    # Space: O(n)
    def find_matrix(self, nums: list[int]) -> list[list[int]]:
        counts = Counter(nums)
        rows: list[list[int]] = [[] for _ in range(max(counts.values()))]
        for value, count in counts.items():
            for row in rows[:count]:
                row.append(value)
        return rows