【发布时间】:2020-02-04 12:19:47
【问题描述】:
让我们看一个从两个排序数组中找到最小的m 数字对的玩具示例。
抛开算法效率问题不谈,我想为优先级队列提供一个比较器
在初始化过程中需要某些参数:
class Foo {
struct natural_order {
const std::vector<int> &x,&y;
natural_order( const std::vector<int> &x, const std::vector<int> &y ): x(x), y(y) {};
bool operator () ( const std::pair<int,int> &a, const pair<int,int> &b ) const {
return x[a.first]+y[a.second] < x[b.first]+y[b.second];
}
};
struct cmp {
std::unique_ptr<natural_order> o;
cmp( const std::vector<int> &x, const std::vector<int> &y ) {
o= std::make_unique<natural_order>(x,y);
}
bool operator () ( const std::pair<int,int> &a, const pair<int,int> &b ) const {
return (*o)(b,a);
}
};
public:
std::vector<std::vector<int>> m_smalles_pairs( std::vector<int> &x, std::vector<int> &y, int m ) {
std::vector<std::vector<int>> res(m);
std::priority_queue<int,std::vector<int>,cmp(x,y)> pq; //<-- the problem is here. Does not compile
for ( int i= 0; i < m; ++i ) {
auto pr= pq.top(); pq.pop();
res[i]= std::vector<int>{x[pr.first],y[pr.second]};
if ( pr.first+1 < x.size() )
pq.push({pr.first+1,pr.second});
if ( pr.second+1 < y.size() )
pq.push({pr.first,pr.second+1});
}
return res;
}
};
本质上,我希望使用一个参数来初始化比较器,但是如何将它作为priority_queue 的第三个参数提供呢?如果不需要x 和y 参数,我会在上面简单地写cmp。
编辑:它似乎使用 lambda 以类似的方式进行,如此处所述C++ priority_queue with lambda comparator error
我能够解决我的问题,但只是好奇struct-comparator 是否允许这种事情。
【问题讨论】:
标签: c++ priority-queue