1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
|
#include <iostream> #include <queue>
using namespace std;
template <typename T> void (const vector<vector<T>>& V, vector<T>& S) { size_t L = 0; for (const auto& X : V) L += X.size(); S.clear(); S.reserve(L); using range = pair<decltype(V[0].begin()), decltype(V[0].end())>; auto cmp = [](range a, range b) { return *(b.first) < *(a.first); }; priority_queue<range, vector<range>, decltype(cmp)> PQ(cmp); for (const auto& X : V) if (X.begin() != X.end()) PQ.push({X.begin(), X.end()}); while (!PQ.empty()) { auto R = PQ.top(); PQ.pop(); S.push_back(*(R.first++)); if (R.first != R.second) PQ.push(R); } }
int main() { vector<vector<int>> A = {{1, 2, 4}, {}, {2, 3, 5}, {3, 4, 6, 8}}; vector<int> B; merge(A, B); for (const auto& x : B) cout << x << endl; return 0; }
|
近期评论