Question

我创建了一个priority_queue，其类型为pair，它将索引分别存储在两个向量（nums1 []和nums2 []）中。

nums1和nums2已经排序。

我希望priority_queue top保持对p，使得nums1 [p.first] + nums2 [p.second]是此priority_queue中其他元素中的最小值。

我写了下面的代码，但pq top给了我一对最大化 nums1 [] + nums2 []。我无法弄清楚为什么。有人能给我一个暗示吗？我理解这个问题可以使用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}}

Answer 1

您的类型输入错误：pq应该存储索引，而不是值，替换

pq.push({nums1[0], nums2[0]});

通过

pq.push({0, 0});

Demo

在lambda比较器中使用捕获

1 个答案: