AcWing 839. 模拟堆

维护一个集合,初始时集合为空,支持如下几种操作:

  1. I x,插入一个数 x;
  2. PM,输出当前集合中的最小值;
  3. DM,删除当前集合中的最小值(数据保证此时的最小值唯一);
  4. D k,删除第 k 个插入的数;
  5. C k x,修改第 k个插入的数,将其变为 x;

现在要进行 N 次操作,对于所有第 2 个操作,输出当前集合的最小值。

输入样例:
8
I -10
PM
I -10
D 1
C 2 8
I 6
PM
DM
输出样例:
-10
6

 代码如下:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N=100010;
int n,m,Size;
int heap[N],ph[N],hp[N];     //ph指p->h,hp指h->p的指针

void heap_swap(int a,int b){
    swap(ph[hp[a]],ph[hp[b]]);
    swap(hp[a],hp[b]);  
    swap(heap[a],heap[b]);
}

void up(int t){
    while(t/2 && heap[t/2]>heap[t]){
        heap_swap(t/2,t);
        t/=2;
    }
}
    //利用int类除法的性质,可以不用分奇偶实现up
    // if(t==1) return ;
    // int idx;
    // if(t%2)    idx=(t-1)/2;
    // else    idx=t/2;
    // if(heap[idx]>heap[t]){
    //     swap(heap[idx],heap[t]);     //加入映射关系后不只要交换heap中的值
    //     up(idx);     //up实现较简单,无需递归实现,可以优化掉
    // }

void down(int t){
    int idx=t;
    if( t*2 <= Size && heap[t*2] < heap[idx] ) idx=2*t;
    if( t*2+1 <= Size && heap[t*2+1] < heap[idx] ) idx=2*t+1;   
    //如果左右子叶都大于该节点,idx更新为其中任一的下标都行,第二句判断之前用else if在数据较多时会wrong
    if(idx!=t){
        heap_swap(idx,t);
        down(idx);
    }
}

int main()
{
    m=0,Size=0;
    int k,x;
    cin>>n;
    char op[5];
    while(n--){
        scanf("%s",op);
        if(!strcmp(op,"I")){
            cin>>x;
            m++;
            heap[++Size]=x;
            ph[m]=Size,hp[Size]=m;
            up(Size);
        }
        else if(!strcmp(op,"PM"))
            cout<<heap[1]<<'\n';
        else if(!strcmp(op,"DM")){
            heap_swap(1,Size);
            Size--;
            down(1);
        }
        else if(!strcmp(op,"D")){
            cin>>k;
            k=ph[k];        //通过统计次序的数组找到,转换成在heap中对应的位置
            heap_swap(k,Size);
            Size--;
            down(k),up(k);
        }
        else{       //'C'
            cin>>k>>x;
            k=ph[k];
            heap[k]=x;
            down(k),up(k);
        }

    }
    return 0;
}

难点:随机的修改和删除,需要双向映射

排序是一种基于数据结构的排序算法。它通过将待排序的序列构建成一个大顶(或小顶),然后依次将顶元素与最后一个元素交换,并重新调整,直到整个序列有序。 具体的排序算法如下: 1. 构建初始:将待排序序列构建成一个大顶。从最后一个非叶子节点开始,依次向前调整每个节点,使其满足大顶的性质。这一过程称为“下沉”操作。 2. 将顶元素与最后一个元素交换:将顶元素与待排序序列的最后一个元素进行交换,此时最后一个元素即为当前序列的最大值。 3. 重新调整:将交换后的顶元素进行“下沉”操作,使其满足大顶的性质。 4. 重复步骤2和3,直到整个序列有序。 下面是C++实现排序的代码: ```cpp #include <iostream> #include <vector> using namespace std; // 下沉操作 void heapify(vector<int>& nums, int n, int i) { int largest = i; // 初始化最大值为当前节点 int left = 2 * i + 1; // 左子节点 int right = 2 * i + 2; // 右子节点 // 如果左子节点大于根节点,则更新最大值 if (left < n && nums[left] > nums[largest]) { largest = left; } // 如果右子节点大于最大值,则更新最大值 if (right < n && nums[right] > nums[largest]) { largest = right; } // 如果最大值不是当前节点,则交换节点,并继续调整 if (largest != i) { swap(nums[i], nums[largest]); heapify(nums, n, largest); } } // 排序 void heapSort(vector<int>& nums) { int n = nums.size(); // 构建初始,从最后一个非叶子节点开始向前调整 for (int i = n / 2 - 1; i >= 0; i--) { heapify(nums, n, i); } // 依次将顶元素与最后一个元素交换,并重新调整 for (int i = n - 1; i > 0; i--) { swap(nums[0], nums[i]); heapify(nums, i, 0); } } int main() { vector<int> nums = {4, 2, 6, 8, 5, 7}; heapSort(nums); cout << "Sorted array: "; for (int num : nums) { cout << num << " "; } return 0; } ``` 以上是排序的C++实现代码,通过构建初始和重复交换顶元素的过程,最终实现了对序列的排序。时间复杂度为O(nlogn),其中n是序列的长度。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值