LC2363-合并相似的物品

题目名称：合并相似的物品

给你两个二维整数数组 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]] 。

提示：

• 1 <= items1.length, items2.length <= 1000
• items1[i].length == items2[i].length == 2
• 1 <= valuei, weighti <= 1000
• items1 中每个 valuei 都是 唯一的 。
• items2 中每个 valuei 都是 唯一的 。

思路分析

把所有价值相同的物品求重量和,然后按照价值大小从低到高进行排序. 根据数据集来看:物品的最大价值只有1000,可以参考有限的数据集使用基数排序. 中间需要注意一下就是如果某个价值对应的重量和为0,表示这个物品没有出现过,需要从结果中过滤掉.

Code实现

public List < List < Integer >> mergeSimilarItems(int[][] items1, int[][] items2) {
    int[] ws = new int[1001];
    for (int[] ints: items1) {
        ws[ints[0]] += ints[1];
    }
    for (int[] ints: items2) {
        ws[ints[0]] += ints[1];
    }

    List < List < Integer >> result = new ArrayList < > ();
    for (int i = 0; i < ws.length; i++) {
        if (ws[i] > 0) {
            result.add(Arrays.asList(i, ws[i]));
        }
    }

    return result;
}

结果

算法复杂度分析

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$