Peter算法小课堂—二叉堆(优先队列)

课前小视频:(7 封私信 / 62 条消息) 看动画,学算法,C++实现建立二叉堆,优先队列和堆排序的基础 - 知乎 (zhihu.com)

二叉堆(优先队列)

大家想想,什么数据结构能做到插入(删除)一个数、询问最小(大)值、 删除最小(大)值,时间复杂度最小。答案是:二叉堆!

那么,我们介绍一下二叉堆。看下图。

堆的定义

int h[MAXN],n;//用数组模拟堆

堆的插入

首先将元素x放到堆中的最后一个位置(即最底层最右边的位置),然后不断地把x往上调整,直到x调不动为止(即大于现在的父亲,或者x处于根节点)。

void up(int k){
	while(k>1){
		int i=k/2;
		if(h[k]<h[i]){
			swap(h[k],h[i]);
			k=i;
		}else break;
	}
}

堆的删除

具体方法:

void down(int k){
	while(k+k<=n){
		int i=k+k;
		if(i+1<=n&&h[i+1]<h[i]) i++;
		if(h[i]<h[k]){
			swap(h[i],h[k]);
			k=i;
		}else break;
	}
}

堆的插入/删除

void add(int x){
	++n;
	h[n]=x;
	up(n);
}
void del(int pos){
	swap(h[pos],h[n]);
	--n;
	up(pos);
	down(pos);
}

STL

优先队列STL:

大根堆:priority_queue

小根堆:priority_queue(int ,vector(int),greater(int)) p;

用法:p.top();p.push();p.pop();p.size();p.empty();

题目

1.给定 n个整数,用要求的排序方法(堆排序、归并排序)进行从大到小的排序。

#include <bits/stdc++.h>
using namespace std;
int n,a[100009],b[100009],k=1;
priority_queue<int> p;
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		p.push(a[i]);
	}
	for(int i=1;i<=n;i++){
		int a=p.top();p.pop();
		b[k]=a;
		k++;
	}
	for(int i=1;i<=n;i++){
		cout<<b[i]<<endl;
	}
	return 0;
}

2.

小明在玩一个数字游戏。游戏规则是这样的,一开始小明那里一个数字也没有,然后他会接受 2 种指令,根据这 2 种指令执行操作,并且回答其中的 q 指令。

2 种指令如下:

a X Y,表示增加一个数字 X,数字 X 的价值是 Y

q,查询目前最小的数字,并且把这个数字删除。如果存在多个数字相同的,删除价值最大的那个。

#include <bits/stdc++.h>
using namespace std;
int n,x,y;
char op[10];
typedef pair<int,int> pii;
priority_queue<pii,vector<pii>,greater<pii> > pq;
int main()
{
    scanf("%d",&n);
    while (n--) {
        scanf("%s",op);
        if (op[0]=='a') {
            scanf("%d%d",&x,&y);
            pq.push(make_pair(x,-y));
        } else {
            pii p=pq.top(); pq.pop();
            printf("%d %d\n",p.first,-p.second);
        }
    }
    return 0;
}

3.有两个长度都是 N的序列 A 和 B,在 A 和 B 中各取一个数相加可以得到 N^2 个和,求这 N^2 个和中最小的 M个。

#include <bits/stdc++.h>
using namespace std;
struct A {
    int v,id;
    bool operator > (const A& a) const {
        return v>a.v;
    }
};
int n,m,a[100010],b[100010],c[100010];
priority_queue<A,vector<A>,greater<A> > pq;
int main() {
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i) scanf("%d",&a[i]);
    for (int i=1;i<=n;++i) scanf("%d",&b[i]);
    sort(b+1,b+1+n);
    for (int i=1;i<=n;++i) c[i]=1;
    for (int i=1;i<=n;i++) pq.push((A){a[i]+b[c[i]],i});
    for (int i=1,v,id;i<=m;i++) {
        v=pq.top().v; id=pq.top().id; pq.pop();
        printf("%d\n",v);
        if (++c[id]<=n)
            pq.push((A){a[id]+b[c[id]],id});
    }
    return 0;
}

希望这些对大家有用,三连必回

相关推荐

  1. Peter算法课堂—差分数组

    2024-01-29 15:04:02       61 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-29 15:04:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-29 15:04:02       100 阅读
  3. 在Django里面运行非项目文件

    2024-01-29 15:04:02       82 阅读
  4. Python语言-面向对象

    2024-01-29 15:04:02       91 阅读

热门阅读

  1. linux ping 某台服务的端口

    2024-01-29 15:04:02       47 阅读
  2. 【面试】MySQL的几种查询方式

    2024-01-29 15:04:02       62 阅读
  3. flink源码分析 - jar包中提取主类和第三方依赖

    2024-01-29 15:04:02       45 阅读
  4. iOS 闭包和Block的区别

    2024-01-29 15:04:02       60 阅读
  5. Nginx加固安全策略,简单实用

    2024-01-29 15:04:02       57 阅读