小根堆(模板)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/wwwlps/article/details/88734975
#include<bits/stdc++.h>
using namespace std;
const int maxn=55;
int n,a[maxn],heap[maxn];
void siftup(int i)//建立小根堆
{
    int flag=0;
    if(i==1)
        return ;
    while(i!=1&&flag==0)
    {
        if(heap[i]<heap[i/2])
            swap(heap[i],heap[i/2]);
        else
            flag=1;
        i=i/2;
    }
}

void siftdown(int i)//删除最小元素并插入新元素,i为1
{
    int flag=0;
    int t;
    while(2*i<=n&&flag==0)
    {
        if(heap[i]>heap[2*i])
          t=2*i;
        else
          t=i;
        if(i*2+1<=n)
        {
            if(heap[t]>heap[i*2+1])
                t=i*2+1;
        }
        if(t!=i)
        {
            swap(heap[t],heap[i]);
            i=t;
        }
        else
            flag=1;
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>heap[i];
        siftup(i);
    }
    for(int i=1;i<=n;i++)
    {
        cout << heap[i] << ' ';;
    }
    cout << endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/wwwlps/article/details/88734975