leetcode 2363. 合并相似的物品

99 阅读2分钟

1. 题目与解析

给你两个二维整数数组 items1 和 items2 ,表示两个物品集合。每个数组 items 有以下特质:

  • items[i] = [valuei, weighti] 其中 valuei 表示第 i 件物品的 价值 ,weighti 表示第 i 件物品的 重量 。
  • items 中每件物品的价值都是 唯一的 。

请你返回一个二维数组 ret,其中 ret[i] = [valuei, weighti], weighti 是所有价值为 valuei 物品的 重量之和 。

注意: ret 应该按价值 升序 排序后返回。

输入: 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]] 。

输入: 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]] 。

输入: 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 都是 唯一的 。

2. 题解

class Solution {
    public List<List<Integer>> mergeSimilarItems(int[][] items1, int[][] items2) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int[] a: items1) {
            map.put(a[0], map.getOrDefault(a[0], 0) + a[1]);
        }
        for (int[] a: items2) {
            map.put(a[0], map.getOrDefault(a[0], 0) + a[1]);
        }

        List<List<Integer>> ans = new ArrayList<>();
        for (Map.Entry<Integer, Integer> e: map.entrySet()) {
            ans.add(Arrays.asList(e.getKey(), e.getValue()));
        }

        Collections.sort(ans, new Comparator<List<Integer>>() {
            public int compare(List<Integer> a, List<Integer> b) {
                return a.get(0) - b.get(0);
            }
        });

        return ans;
    }
}