求数组中连续k个数最大值

题目

前K大问题：给定一个长度为n数组，求每个连续K个数的最大值，复杂度要求O(n×log(K))。
解释：例如数组为[1,3,2,4,5]，K= 2，则连续K个数的区间有[1,3],[3,2],[2,4],[4,5]，每个区间的最大值分别为3,3,4,5，所以输出3,3,4,5

代码

#include<bits/stdc++.h>
using namespace std;
constintmaxn=1e5+5;
int n,k,a[maxn];
multiset<int> ms;
int main() {
		cin >> n >> k;
		if(k > n || k == 0) {
			cerr <<"InputError"<< endl;
		}
		for(int i = 0; i < n; i++) {
			cin >> a[i];
		}
		for(int i = 0; i < k; i++){
			ms.insert(a[i]);
		}
		cout << *(--ms.end()) <<'';
		for(int i = k; i < n; i++) {
			ms.erase(ms.lower_bound(a[i - k]));
			ms.insert(a[i]);
			cout << *(--ms.end()) <<'';
		}
		cout<<endl;
		return 0;
}

变式题

给定A、B两个数列，各包含n个数，分别从A和B中任意取一个数相加得到和，这样会有n2种结果(包括重复的)，求n2个结果中前n个最小的和。

样例：
input：
5
1 3 4 2 0
7 3 5 2 11

output：
2 3 3 4 4

思路：维护一个大小为n的优先队列

代码

#include <algorithm>
#include <iostream>
#include <queue>
#include <stack>
#include <vector>
using namespace std;
vector<long long> a, b;
int n;
long long temp;
int main() {
    cin >> n;
    priority_queue<long long, vector<long long>> res;
    stack<long long> rres;
    for (int i = 0; i < n; i++) {
        cin >> temp;
        a.push_back(temp);
    }
    for (int i = 0; i < n; i++) {
        cin >> temp;
        b.push_back(temp);
    }
    sort(a.begin(), a.end());
    sort(b.begin(), b.end());
    a.erase(a.begin() + n / 2 + 1, a.end());
    b.erase(b.begin() + n / 2 + 1, b.end());
    for (int i = 0; i < a.size(); i++) {
        for (int j = 0; j < b.size(); j++) {
            if (res.size() == n) {
                if (a[i] + b[j] >= res.top()) {
                    break;
                } else {
                    res.pop();
                }
            }
            res.push(a[i] + b[j]);
        }
    }
    while (res.size()) {
        rres.push(res.top());
        res.pop();
    }
    while (rres.size()) {
        cout << rres.top() << " ";
        rres.pop();
    }
    return 0;
}

求数列中第k大的数

维护一个大小为k的优先队列。