[英]use of capture in lambda comparator
我創建了一個類型為pair的priority_queue,該索引將索引分別存儲在兩個向量(nums1 []和nums2 [])上。
nums1和nums2已排序。
我希望priority_queue頂部保持對p,以便nums1 [p.first] + nums2 [p.second]在此priority_queue中的其他元素中是最小的。
我編寫了以下代碼,但是pq top給了我最大化 nums1 [] + nums2 []的對p。 我不知道為什么。 有人可以給我一個提示嗎? 我知道可以使用用戶定義的類/結構的pq來解決此問題,但我很好奇如何在此處使用lambda函數。 謝謝。
priority_queue<pair<int, int>, vector<pair<int, int>>, function<bool(const pair<int,int>&, const pair<int,int>&)>> pq([&](const pair<int,int>&a, const pair<int,int>&b){
return nums1[a.first] + nums2[a.second] > nums1[b.first] + nums2[b.second];
});
要提供完整的背景信息:
我正在解決的問題如下:
您將獲得兩個按升序排列的整數數組nums1和nums2,以及一個整數k。 定義一個對(u,v),該對由第一個數組中的一個元素和第二個數組中的一個元素組成。 找出總和最小的k對(u1,v1),(u2,v2)...(uk,vk)。
我的代碼是:
vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
vector<pair<int, int>> ans;
int m = nums1.size();
if(m == 0) return ans;
int n = nums2.size();
if(n == 0) return ans;
priority_queue<pair<int, int>, vector<pair<int, int>>, function<bool(const pair<int,int>&, const pair<int,int>&)>> pq([&]](const pair<int,int>&a, const pair<int,int>&b){
return nums1[a.first] + nums2[a.second] > nums1[b.first] + nums2[b.second];
});
pq.push({nums1[0], nums2[0]}); // THIS LINE SHOULD BE pq.push({0, 0});
unordered_set<string> visited;
visited.emplace("0,0");
while(!pq.empty() && k-- > 0) {
auto top = pq.top();
pq.pop();
int index1 = top.first, index2 = top.second;
ans.push_back({nums1[index1], nums2[index2]});
if(index1 + 1 < m && !visited.count(to_string(index1 + 1) + "," + to_string(index2))) {
visited.emplace(to_string(index1 + 1) + "," + to_string(index2));
pq.push({index1 + 1, index2});
}
if(index2 + 1 < n && !visited.count(to_string(index1) + "," + to_string(index2 + 1))) {
visited.emplace(to_string(index1) + "," + to_string(index2 + 1));
pq.push({index1, index2 + 1});
}
}
return ans;
}
輸入nums1 = {1,7,11},nums2 = {2,4,6},k = 3
我的錯誤輸出是ans = {{7,6},{11,6}}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.