社区讨论

用对顶堆做的注意点

P1801黑匣子参与者 1已保存回复 0

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
0 条
当前快照
1 份
快照标识符
@m6lpmh81
此快照首次捕获于
2025/02/01 12:45
去年
此快照最后确认于
2025/02/01 14:06
去年
查看原帖
插入大量数字之后先把大根堆的大小调整为i,再调整堆的有序,最后再调整大根堆的数字保持为i,如果在插入大量数字之后先调整有序再调整大根堆的大小就会超时
CPP
#include<iostream>
#include<queue>
using namespace std;
#define MAX 200010
typedef long long ll;
int n,m,s;
int add[MAX];
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m;
	for(int i=1; i<=n; i++) {
		cin>>add[i];
	}
	priority_queue<int,vector<int> ,greater<int> > down;//小根堆
	priority_queue<int> up;//大根堆 
	int j=0;
	for(int i=1; i<=m; i++) {
		int query;
		cin>>query;
		while(j<query) {//插入到优先队列,
			if(up.size()==0||add[i]<=up.top())up.push(add[++j]);
			else down.push(add[++j]);
		}
		while(up.size()>i) {  //先调整顺序就会tle,先把一个堆变小 
			down.push(up.top());
			up.pop();
		}
		while(up.size()<i) { 
			up.push(down.top());
			down.pop();
		}
		while(up.size()!=0&&down.size()!=0&&up.top()>down.top()) {
			down.push(up.top());
			up.pop();
		}
		while(up.size()>i) { 
			down.push(up.top());
			up.pop();
		}
		while(up.size()<i) { 
			up.push(down.top());
			down.pop();
		}
		cout<<up.top()<<endl;
	}
}

回复

0 条回复,欢迎继续交流。

正在加载回复...