首页 > 其他分享 >力扣---2363. 合并相似的物品

力扣---2363. 合并相似的物品

时间:2023-02-28 11:56:54浏览次数:40  
标签:力扣 arr weight int temList --- items1 items2 2363

给你两个二维整数数组 items1 和 items2 ,表示两个物品集合。每个数组 items 有以下特质:
    items[i] = [valuei, weighti] 其中 valuei 表示第 i 件物品的 价值 ,weighti 表示第 i 件物品的 重量 。
    items 中每件物品的价值都是 唯一的 。
请你返回一个二维数组 ret,其中 ret[i] = [valuei, weighti], weighti 是所有价值为 valuei 物品的 重量之和 。
注意:ret 应该按价值 升序 排序后返回。

示例 1:
输入:items1 = [[1,1],[4,5],[3,8]], items2 = [[3,1],[1,5]]
输出:[[1,6],[3,9],[4,5]]
解释:
value = 1 的物品在 items1 中 weight = 1 ,在 items2 中 weight = 5 ,总重量为 1 + 5 = 6 。
value = 3 的物品再 items1 中 weight = 8 ,在 items2 中 weight = 1 ,总重量为 8 + 1 = 9 。
value = 4 的物品在 items1 中 weight = 5 ,总重量为 5 。
所以,我们返回 [[1,6],[3,9],[4,5]] 。

示例 2:
输入:items1 = [[1,1],[3,2],[2,3]], items2 = [[2,1],[3,2],[1,3]]
输出:[[1,4],[2,4],[3,4]]
解释:
value = 1 的物品在 items1 中 weight = 1 ,在 items2 中 weight = 3 ,总重量为 1 + 3 = 4 。
value = 2 的物品在 items1 中 weight = 3 ,在 items2 中 weight = 1 ,总重量为 3 + 1 = 4 。
value = 3 的物品在 items1 中 weight = 2 ,在 items2 中 weight = 2 ,总重量为 2 + 2 = 4 。
所以,我们返回 [[1,4],[2,4],[3,4]] 。

示例 3:
输入:items1 = [[1,3],[2,2]], items2 = [[7,1],[2,2],[1,4]]
输出:[[1,7],[2,4],[7,1]]
解释:
value = 1 的物品在 items1 中 weight = 3 ,在 items2 中 weight = 4 ,总重量为 3 + 4 = 7 。
value = 2 的物品在 items1 中 weight = 2 ,在 items2 中 weight = 2 ,总重量为 2 + 2 = 4 。
value = 7 的物品在 items2 中 weight = 1 ,总重量为 1 。
所以,我们返回 [[1,7],[2,4],[7,1]] 。

提示:
    1 <= items1.length, items2.length <= 1000
    items1[i].length == items2[i].length == 2
    1 <= valuei, weighti <= 1000
    items1 中每个 valuei 都是 唯一的 。
    items2 中每个 valuei 都是 唯一的 。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/merge-similar-items
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


 

由于题目明确了valuei都是唯一的,很容易就能联想到用哈希表来做。

由于题目要求按照升序排序返回,有两种方法来实现这个效果:

1. 先用HashMap将所有结果存入list中,然后利用Collections.sort()函数以及重写比较器来进行排序。

2. 利用TreeMap可以默认按照key字典序的特点,从而可以省略最后对结果进行排序的步骤。

方法一代码:

class Solution {
    public List<List<Integer>> mergeSimilarItems(int[][] items1, int[][] items2) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int[] arr : items1) {
            map.put(arr[0], arr[1]);
        }
        for (int[] arr : items2) {
            map.put(arr[0], map.getOrDefault(arr[0], 0) + arr[1]);
        }
        List<List<Integer>> list = new ArrayList<>();
        for (Map.Entry<Integer, Integer> temMap : map.entrySet()) {
            List<Integer> temList = new ArrayList<>();
            temList.add(temMap.getKey());
            temList.add(temMap.getValue());
            list.add(temList);
        }
        Collections.sort(list, (a, b) -> (a.get(0) - b.get(0)));
        return list;
    }
}

 

TreeMap 代码:

class Solution {
    public List<List<Integer>> mergeSimilarItems(int[][] items1, int[][] items2) {
        Map<Integer, Integer> map = new TreeMap<>();
        for (int[] arr : items1) {
            map.put(arr[0], arr[1]);
        }
        for (int[] arr : items2) {
            map.put(arr[0], map.getOrDefault(arr[0], 0) + arr[1]);
        }
        List<List<Integer>> list = new ArrayList<>();
        for (Map.Entry<Integer, Integer> temMap : map.entrySet()) {
            List<Integer> temList = new ArrayList<>();
            temList.add(temMap.getKey());
            temList.add(temMap.getValue());
            list.add(temList);
        }
        return list;
    }
}

 

 由于本题数据量不大,还可以直接利用数组来做。

class Solution {
    public List<List<Integer>> mergeSimilarItems(int[][] items1, int[][] items2) {
        int[] arr = new int[1001];
        for (int[] temArr : items1) {
            arr[temArr[0]] += temArr[1];
        }
        for (int[] temArr : items2) {
            arr[temArr[0]] += temArr[1];
        }
        List<List<Integer>> list = new ArrayList<>();
        for (int i = 0; i < arr.length; i ++) {
            if (arr[i] != 0) {
                List<Integer> temList = new ArrayList<>();
                temList.add(i);
                temList.add(arr[i]);
                list.add(temList);
            }
        }
        return list;
    }
}

 

 

标签:力扣,arr,weight,int,temList,---,items1,items2,2363
From: https://www.cnblogs.com/allWu/p/17163525.html

相关文章