0
点赞
收藏
分享

微信扫一扫

[leetcode每日一题]2.28

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)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

举报

相关推荐

0 条评论