lecode 2610. 转换二维数组-medium

1 题目：转换二维数组

官方标定难度：中

给你一个整数数组 nums 。请你创建一个满足以下条件的二维数组：

二维数组应该 只 包含数组 nums 中的元素。
二维数组中的每一行都包含 不同 的整数。
二维数组的行数应尽可能 少 。
返回结果数组。如果存在多种答案，则返回其中任何一种。

请注意，二维数组的每一行上可以存在不同数量的元素。

示例 1：

输入：nums = [1,3,4,1,2,3,1]
输出：[[1,3,4,2],[1,3],[1]]
解释：根据题目要求可以创建包含以下几行元素的二维数组：

• 1,3,4,2
• 1,3
• 1
  nums 中的所有元素都有用到，并且每一行都由不同的整数组成，所以这是一个符合题目要求的答案。
  可以证明无法创建少于三行且符合题目要求的二维数组。

示例 2：

输入：nums = [1,2,3,4]
输出：[[4,3,2,1]]
解释：nums 中的所有元素都不同，所以我们可以将其全部保存在二维数组中的第一行。

提示：

1 <= nums.length <= 200
1 <= nums[i] <= nums.length

2 solution

这道题相对比较简单，只需要统计每个数字出现的次数，然后每次将剩余的数字放入新的行，不重复即可。直到没有数字为止。
其实可以稍微优化一下，但是数据量不大，感觉没太必要。

代码

class Solution {
public:
vector<vector<int>> findMatrix(vector<int> &nums) {
    vector<int> cnt(nums.size() + 1, 0);
    for (int i: nums) cnt[i]++;
    
    auto p = max_element(cnt.begin(), cnt.end());

    vector<vector<int>> result;
    
    while (*p) {
        vector<int> row;
        for (int j = 1; j <= nums.size(); j++) {
            if (cnt[j] > 0) {
                row.push_back(j);
                cnt[j]--;
            }
        }
        result.push_back(row);
    }
    return result;
}
};

结果