具有自定义类和 lambda 表达式的 C++ 中的优先级队列

Question

我正在学习如何在 C++ 中使用 STL。我正在尝试使用std::priority_queue 实现堆。这是我的尝试：

#include<vector>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stack>
#include<functional>
using namespace std;

struct data {
    int first;
    int second;
};

class compare {
public:
    bool operator()(data &a ,data &b)
    {
         if(a.first < b.first) return true;
         else false;
    }
};

int main()
{   
    priority_queue <data , vector<data>, compare> heap;
    data temp2[] = { {5,19},{2,7},{90,9},{12,6} };

    for(int i = 0; i < 4; ++i)
    {       
       heap.push(temp2[i]);
    }

    while(heap.empty() == false)
    {
        cout << heap.top().first << endl;;
        heap.pop();
    }       
}

当我在 Visual Studio 2012 上运行此代码时。它在推送第二个元素时崩溃。错误如下：

Program: C:\WINDOWS\system32\MSVCP110D.dll
File: c:\program files (x86)\microsoft visual studio 11.0\vc\include\algorithm
Line: 2463
Expression: invalid operator<

但是当我在 ideone 中运行相同的代码时，它可以工作，但输出错误。 Ideone link

1.谁能指出我哪里出错了？

在同一个问题中再添加一个问题：

2.如何写 this 的 lambda 表达式而不是使用 compare 函数？

Answer 1

您不应忽略编译器发出的警告：

warning C4715: 'compare::operator()' : not all control paths return a value

这是因为else false; 不返回任何内容，它只是计算表达式false 并丢弃结果。应该是else return false;。

---

当然，更简单的选择是直接说return a.first < b.first;。

---

回复您更新的问题：

在这里使用 lambda 不会给您带来太多好处，因为您需要在初始化 heap 时将仿函数的类型指定为 std::priority_queue 的模板参数，并将仿函数本身指定为构造函数的参数（在使用compare 的版本中，函子是默认构造的；lambdas 不是默认构造的）。

但是，既然你问了，那就是：

#include <iostream>
#include <queue>
#include <vector>

struct data
{
   int first;
   int second;
};

int main()
{
   auto cmp = [](const data& a, const data& b) { return a.first < b.first; };

   // For Visual C++ 2013 and beyond:
   // std::vector<data> temp = {{5, 19}, {2, 7}, {90, 9}, {12, 6}};
   // std::priority_queue<data, std::vector<data>, decltype(cmp)> heap{cmp, std::move(temp)};

   // For Visual C++ 2012 (doesn't support list-initialization for non-aggregates):
   data temp[] = {{5, 19}, {2, 7}, {90, 9}, {12, 6}};
   std::priority_queue<data, std::vector<data>, decltype(cmp)> heap(std::begin(temp), std::end(temp), cmp);

   while(!heap.empty())
   {
      std::cout << heap.top().first << '\n';
      heap.pop();
   }
}

我对您的原始代码做了几处修改；在正确性 (const) 和效率方面将它们作为改进建议。

Answer 2

您的比较函数在false 之前缺少return。这应该有效：

class compare{
public:
    bool operator()(data &a, data &b)
    {
        if (a.first < b.first) 
            return true;
        else 
            return false;
    }
};

VS 给出编译警告，但不是错误。正如巴里所指出的，这不是错误。有关更多详细信息，请参阅此 SO 答案 https://stackoverflow.com/a/1610454/3154588。 （我仍然认为这应该是一个编译错误，因为没有返回值的代码路径总是错误的——至少对于我编写的所有代码而言！）