LeetCode 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]] 。

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

直接模拟即可:

class Solution {
public:
    vector<vector<int>> mergeSimilarItems(vector<vector<int>>& items1, vector<vector<int>>& items2) {
        map<int, int> mapAns;
        for (vector<int>& item : items1) {
            mapAns[item[0]] += item[1];
        }
        for (vector<int> &item : items2) {
            mapAns[item[0]] += item[1];
        }

        vector<vector<int>> ans;
        for (pair<const int, int> item : mapAns) {
            vector<int> oneAns = {item.first, item.second};
            ans.push_back(oneAns);
        }

        return ans;
    }
};

如果输入数组items1中有n个元素,items2中有m个元素,此算法时间复杂度为O((n+m)lg(n+m)),空间复杂度为O(n+m)。

posted @ 2023-02-19 15:26  epiphanyy  阅读(19)  评论(0)    收藏  举报  来源