数据结构--最小堆、最大堆(小根堆、大根堆)

1.最小堆的实现参考:https://blog.csdn.net/genios/article/details/8157031


int a[N];
void Build(){
    for(int i=n/2;i>=1;i--){
        int tem=a[i];
        int j=i*2;
        while(j<=n){
            if((j+1)<=n && (a[j+1]<a[j]))
                j++;
            if(tem>a[j]){
                a[j/2]=a[j];
                j*=2;
            }else break;
        }
        a[j/2]=tem;
//        for(int i=1;i<=n;i++)
//            cout<<a[i]<<" ";
//        cout<<endl;
    }
}
int main()
{
    int m;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    Build();
    for(int i=1;i<=n;i++)
        printf("%d ",a[i]);
    return 0;
}

最大堆的实现差不多

2.以最小堆为例,说明加入一个元素

void Insert(int x){
    int i=++n;
    while(i!=1 && x < a[i/2]){
        a[i]=a[i/2];
        i/=2;
    }
    a[i]=x;
}
void Erase(int x){
    //对于最小堆和最大堆而言,删除是针对于根节点而言
    int y=a[n--];//取最后一个元素,放到根上,重构
    int son=2,root=1;
    while(son<=n){
        if(son<n && a[son] > a[son+1])
            son++;
        if(y > a[son]){
            a[root]=a[son];
            root=son;
            son*=2;
        }else break;
    }
    a[root]=y;
}

9 804 9059 9155 8849
9 804 9155 9059 8849
建堆有两种方法,一种是上述的已知序列得到的,另一种则是边插入边建立;

边插入边建立:

int insert(int i){
	int temp;
	while(a[i/2]>a[i] && i!=1){
		temp=a[i];
		a[i]=a[i/2];
		a[i/2]=temp;
		i=i/2;
	}
	return 0;
}
int main()
{

    cin>>a[1];
    for(int i=2;i<=n;i++){
       cin>>a[i];
        insert(i);
    }
    return 0;
}

但是两者有些情况下是不一样的,下面的附赠上我的对比代码,有兴趣的可以执行程序看一下;

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>
#include <set>
#include <utility>
#include <sstream>
#include <iomanip>
using namespace std;

const int N = 1e5 + 10;
const int mod = 1e9+7;
int n;
int a[N*10],b[N];
void Build(){
    for(int i=n/2;i>=1;i--){
        int tem=b[i];
        int j=i*2;
        while(j<=n){
            if((j+1)<=n && (b[j+1]<b[j]))
                j++;
            if(tem>b[j]){
                b[j/2]=b[j];
                j*=2;
            }else break;
        }
        b[j/2]=tem;
    }
    int y=0;
    for(int i=1;i<=n;i++){
        if(a[i]!=b[i])
        {
            y=1;
            break;
        }
    }
    if(y){
        for(int i=1;i<=n;i++)
            cout<<a[i]<<" ";
        cout<<endl;
        for(int i=1;i<=n;i++)
            cout<<b[i]<<" ";
        cout<<endl;
        cout<<endl;
    }
}

int insert(int i){
	int temp;
	while(a[i/2]>a[i] && i!=1){
		temp=a[i];
		a[i]=a[i/2];
		a[i/2]=temp;
		i=i/2;
	}
	return 0;
}
int main()
{
    int flag=0;
    srand((unsigned)time(NULL));
    flag=1000;
    while(flag--){
    cout<<flag<<endl;
    n=rand()%10+1;
    a[1]=rand()%10000;
    b[1]=a[1];
    for(int i=2;i<=n;i++){
       a[i]=rand()%10000;
       b[i]=a[i];
       insert(i);
    }
    Build();

    }
    return 0;
}

注:PTA7-12 关于堆的判断 (25分)这一题,只能用边插入边建堆来实现,别问我怎么知道的,QWQ

发布了230 篇原创文章 · 获赞 17 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/c___c18/article/details/104136344