问题描述
过去,我编写了几乎类似的代码,并且它起作用了(我记得有些模糊)。似乎比较器在这里不能正常工作??有任何线索吗?
#include<iostream>
#include<vector>
#include<queue>
#include<iterator>
#include<algorithm>
using namespace std;
typedef pair<vector<int>::iterator,vector<int>::iterator> PR;
struct CompareFn{
bool operator()(const PR& a,const PR& b){
//cout<<"a and b first: "<<*(a.first)<<" "<< *(b.first)<<endl;
return *a.first > *b.first;
}
};
vector<int> mergeKSortedArrays(vector<vector<int>> &A) {
vector<int> result;
priority_queue<PR,vector<PR>,CompareFn> PQ;
for(auto e:A){
if(e.size()>0) PQ.push({e.begin(),e.end()});
}
while(PQ.size()>0) {
PR tmp = PQ.top(); PQ.pop();
auto cur=tmp.first;
auto lst=tmp.second;
result.emplace_back (*cur);
if((++cur)!=lst) PQ.push({cur,lst});
}
return result;
}
int main() {
vector<vector<int>> v= {{2,3,8,10},{1,4,12},{4,5,8}};
vector<int> result = mergeKSortedArrays(v);
copy(result.begin(),result.end(),ostream_iterator<int>(cout," "));
return 0;
}
我期望它几乎可以像整数一样工作于一对迭代器。但事实并非如此。
解决方法
从begin()
中的end()
的副本获得的vector
和for(auto e : A)
迭代器在迭代结束后无效,并且临时vector
e
被销毁了。
使用对内部vector
的引用:
for(auto& e : A) { // "auto& e" makes "e" a reference to the existing vector
if(e.size()>0) PQ.emplace(e.begin(),e.end());
}
这里是另外一个demo,我在其中应用了适当的const
限定词。