2363. 合并相似的物品
提示
简单
12
相关企业
给你两个二维整数数组 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
都是 唯一的
Solution
2月的最后一天,希望能de完代码明天去清华园玩。
class Solution:
def mergeSimilarItems(self, items1: List[List[int]], items2: List[List[int]]) -> List[List[int]]:
items1.sort()
items2.sort()
res = []
p1 = 0
p2 = 0
n1 = len(items1)
n2 = len(items2)
while p1 < n1 or p2 < n2:
if p1 == n1:
for i in range(p2, n2):
res.append(items2[i])
p2 += 1
elif p2 == n2:
for i in range(p1, n1):
res.append(items1[i])
p1 += 1
elif items1[p1][0] == items2[p2][0]:
res.append([items1[p1][0], items1[p1][1] + items2[p2][1]])
p1 += 1
p2 += 1
elif items1[p1][0] > items2[p2][0]:
res.append(items2[p2])
p2 += 1
else:
res.append(items1[p1])
p1 += 1
return res
然后一看题解用的是Counter,好吧,我无语了。
顺便学习一下C++的map函数。map函数会自动进行排序,所以可以省去排序步骤。
当然我还是很无语,一是题解又偷懒,而是自己写这种东西还是会犯一大堆错,按理说也做过这么多题了,也不应该,但自己就是比较蠢,比较糊。
就这样吧,唉。二月更文也完成了,希望奖励早点寄出来。
class Solution {
public:
vector<vector<int>> mergeSimilarItems(vector<vector<int>>& items1, vector<vector<int>>& items2) {
map<int, int> mp;
for (auto &v : items1) {
mp[v[0]] += v[1]; // 甚至自动初始化成0,66666
}
for (auto &v : items2) {
mp[v[0]] += v[1];
}
vector<vector<int>> res;
for (auto &[k, v] : mp) {
res.push_back({k, v});
}
return res;
}
};
作者:力扣官方题解
链接:https://leetcode.cn/problems/merge-similar-items/solutions/2136734/he-bing-xiang-si-de-wu-pin-by-leetcode-s-ywx0/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。