leetcode2363.合并相似的物品

Posted on | In |
Words count in article: 480 | Reading time ≈ 2

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

items[i] = [valuei, weighti] 其中 valuei 表示第 i 件物品的 价值 ，weighti 表示第 i 件物品的 重量 。
items 中每件物品的价值都是 唯一的 。
请你返回一个二维数组 ret，其中 ret[i] = [valuei, weighti]， weighti 是所有价值为 valuei 物品的 重量之和 。

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

示例：

1
2
3
4
5
6
7
输入：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
简单题无

代码

  • C++

  • 归并
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class Solution {
public:
    vector<vector<int>> mergeSimilarItems(vector<vector<int>>& items1, vector<vector<int>>& items2) {
        sort(items1.begin(), items1.end());
        sort(items2.begin(), items2.end());

        vector<vector<int>> ans;

        int i = 0, j = 0;
        while (i < items1.size() || j < items2.size()) {
            if (j == items2.size()) { 
                ans.push_back(items1[i]);
                i++;
            } else if (i == items1.size()) {
                ans.push_back(items2[j]);
                j++;
            } else {
                if (items1[i][0] < items2[j][0]) {
                    ans.push_back(items1[i]);
                    i++;
                } else if (items1[i][0] > items2[j][0]) {
                    ans.push_back(items2[j]);
                    j++;
                } else {
                    ans.push_back({items1[i][0], items1[i][1] + items2[j][1]});
                    i++;
                    j++;
                }
            }
        }

        return ans;
    }
};


  • 哈希表
1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution {
public:
    vector<vector<int>> mergeSimilarItems(vector<vector<int>>& items1, vector<vector<int>>& items2) {
        map<int, int> cnt; // map自带排序
        for (auto& v : items1) cnt[v[0]] += v[1];
        for (auto& v : items2) cnt[v[0]] += v[1];

        vector<vector<int>> res;
        for (auto& [k, v] : cnt) res.push_back({k, v});

        return res;
    }
};

[原题链接](2363. 合并相似的物品 - 力扣（Leetcode）)

0%