蓝桥杯(C++ 整数删除 优先队列 )

发布时间:2024年01月18日

优先队列:

优先队列具有队列的所有特性,包括队列的基本操作,只是在这基础上添加了内部的一个排序,它本质是一个堆实现的。

1.头文件&定义

#include <queue>
#include <functional> //greater<>

// 定义
priority_queue<int> pq;

2.默认优先输出大数据

priority_queue<Type, Container, Functional>

其中, Type 为数据类型. Container 为保存数据的容器. Functional 为元素比较的方式.

若不写后面两个参数.

容器 默认使用 vector

比较方式 默认使用 operator < 即优先队列是大顶堆. 队头元素最大

srand(time(NULL));
priority_queue<int> pq1; // 默认是最大堆...
std::cout << "start..." << endl;
for (int i = 0; i < 10; i++) {
    int t = rand() % 100;
    cout << t << ends;
    pq1.push(t);
}
std::cout << endl;
while (!pq1.empty())
{
    cout << pq1.top() << ends;
    pq1.pop();
}
cout << endl;

3.优先输出小数据 即小顶堆

priority_queue<int, vector<int>, greater<int> > p;

使用?greater<int>?. 即改用?operator >
小根堆声明方式
大根堆是把大的元素放堆顶,小根堆就是把小的元素放到堆顶。

priority_queue<int, vector<int>, greater<int>> pq2; // 最小堆

std::cout << "start..." << endl;
for (int i = 0; i < 10; i++) {
    int t = rand() % 100;
    std::cout << t << ends;
    pq2.push(t);
}
std::cout << endl;

while (!pq2.empty())
{
    cout << pq2.top() << ends;
    pq2.pop();
}
cout << endl;

#include<iostream>
#include<queue>
#include<functional>
using namespace std;
using ll = long long;
ll v[10001000], l[1000010], r[1000010];//l记录左边坐标,r记录右边坐标
void Delet(int i)//删除该点
{
	r[l[i]] = r[i], l[r[i]] = l[i];//该点的右边的点的下标赋给该点左边点记录右边点下标的值,同理左边
	v[l[i]] += v[i], v[r[i]] += v[i];//该点加给左右两边的点
}
int main()
{
	int n, k;
	cin >> n >> k;
	priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> h;//最小堆
	r[0] = 1, l[n + 1] = n;
	for (int i = 1; i <= n; i++)
	{
		cin >> v[i];
		l[i] = i - 1, r[i] = i + 1;//记录该点的左右坐标
		h.push({ v[i],i });//权值和下标
	}
	while (k--)
	{
		pair<ll, int> p = h.top();
		h.pop();//取队首
		if (p.first != v[p.second])//v会更新,需要重新放入堆
		{
			h.push({ v[p.second],p.second });
			k++;
		}
		else//删除该点
			Delet(p.second);
	}
	int head = r[0];//从右到左输出
	while (head != n + 1)
	{
		cout << v[head] << " ";
		head = r[head];
	}
}

文章来源:https://blog.csdn.net/qq_74156152/article/details/135679925
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。