力扣---6363. 转换二维数组

发布时间 2023-04-02 15:35:30作者: Owlwu

给你一个整数数组 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

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/convert-an-array-into-a-2d-array-with-conditions
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


 

周赛第二题,见注释。

class Solution {
    public List<List<Integer>> findMatrix(int[] nums) {
        // 大致思路是:题目要求实际上是让我们求最少不包含相同元素的数组的数量,即我们很容易就能想到实际上该数量由 nums 中最多数量的相同元素决定。然后我们可以想到先将最多的元素放到不同数组的开头,然后将第二多的元素分别放到不同数组的第二个,不够的话就空着,以此类推。
        // 有了这个方向,我们可以利用排序来完成这个思路
        Arrays.sort(nums);
        // 记录返回答案的长度(实际上是 nums 中最多相同元素的数量。)
        int len = 1;
        // 计数
        int count = 1;
        // 存储不同元素在nums中开始的下标,往答案里面添加数据后,直接该坐标加一即可。
        List<Integer> list = new ArrayList<>();
        // 第一个元素必定包含。
        list.add(0);
        // 数数,往list中添加不同元素开始时的下标,并且记录最多相同元素的数量。
        for (int i = 1; i < nums.length; i ++) {
            if (nums[i - 1] == nums[i]) {
                count ++;
            } else {
                list.add(i);
                len = Math.max(len, count);
                count = 1;
            }
        }
        len = Math.max(len, count);
        // 我们需要判断是否越界,所以弄一个备份。
        List<Integer> list1 = new ArrayList<>(list);
        // 存储答案。
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < len; i ++) {
            List<Integer> ans = new ArrayList<>();
            for (int j = 0; j < list.size(); j ++) {
                if (j + 1 < list.size() && list.get(j) < list1.get(j + 1)) { 
                    // 避免 j + 1 越界,并且判断当前是否还可以添加下一个元素。
                    ans.add(nums[list.get(j)]);
                    // 该元素已经被拿掉了,下次该放下一个元素了(如果有的话。)
                    list.set(j, list.get(j) + 1);
                } else if (j + 1 == list.size()) {
                    // 避免越界。
                    if (list.get(j) < nums.length) {
                        ans.add(nums[list.get(j)]);
                        list.set(j, list.get(j) + 1);
                    }
                }
            }
            res.add(ans);
        }
        return res;
    }
}