leetcode每日一题:转换二维数组

诗远

关注

阅读 13

03-19 12:00

leetcode每日一题:转换二维数组_二维数组

题目

2610. 转换二维数组

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

思路

题目中有一个点没明确,就是生成的二维数组,需要包含原始整数数组 nums中的所有元素。做法可以直接贪心,二维数组中,第1行包含原始数组中所有不重复的元素,第2行包含去掉第1行元素后,剩余元素中所有不重复的元素,以此类推,直到原始数组中不包含任何元素。

具体操作上,可以利用一个map记录下原始数组中,每个元素出现的次数,然后构造二维数组中的每1行,贪心的,每1行要尽可能多的包含元素,把1个元素分配掉以后,需要从原始map中把这个数的出现次数-1,如果减到0,就删除对应的键值对,直到map为空,那二维数据也就构造完成。

图解

leetcode每日一题:转换二维数组_算法_02

代码

leetcode每日一题:转换二维数组_数组_03

异常

上述代码运行会报ConcurrentModificationException,原因是我们在for循环遍历map 的时候,不能做remove操作,否则可能导致数据不一致或者底层数据结构破坏。

leetcode每日一题:转换二维数组_List_04

我们可以对上述代码做一些调整,使用一个Set记录下要删除的key,for循环后统一删除;或者可以直接使用迭代器来遍历Map。

代码--使用set保存待删除的key

leetcode每日一题:转换二维数组_二维数组_05

public List<List<Integer>> findMatrix(int[] nums) {
    // key = num value, value = cnt
    Map<Integer, Integer> valueToCntMap = new HashMap<Integer, Integer>();
    for (int num : nums) {
        valueToCntMap.merge(num, 1, Integer::sum);
    }
    List<List<Integer>> ans = new ArrayList<List<Integer>>();
    while (!valueToCntMap.isEmpty()) {
        List<Integer> list = new ArrayList<>();
        Set<Integer> removeKeySet = new HashSet<>();
        for (Map.Entry<Integer, Integer> entry : valueToCntMap.entrySet()) {
            list.add(entry.getKey());
            if (entry.getValue() == 1) {
                removeKeySet.add(entry.getKey());
            } else {
                valueToCntMap.put(entry.getKey(), entry.getValue() - 1);
            }
        }
        for (Integer key : removeKeySet) {
            valueToCntMap.remove(key);
        }
        ans.add(list);
    }
    return ans;
}

代码--使用迭代器

leetcode每日一题:转换二维数组_二维数组_06

public List<List<Integer>> findMatrix(int[] nums) {
    // key = num value, value = cnt
    Map<Integer, Integer> valueToCntMap = new HashMap<Integer, Integer>();
    for (int num : nums) {
        valueToCntMap.merge(num, 1, Integer::sum);
    }
    List<List<Integer>> ans = new ArrayList<List<Integer>>();
    while (!valueToCntMap.isEmpty()) {
        List<Integer> list = new ArrayList<>();
        Iterator<Map.Entry<Integer, Integer>> iterator = valueToCntMap.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<Integer, Integer> entry = iterator.next();
            list.add(entry.getKey());
            if (entry.getValue() == 1) {
                iterator.remove();
            } else {
                valueToCntMap.put(entry.getKey(), entry.getValue() - 1);
            }
        }
        ans.add(list);
    }
    return ans;
}

耗时

leetcode每日一题:转换二维数组_数组_07

精彩评论(0)

0 0 举报