求助手写二叉堆
  • 板块灌水区
  • 楼主qwq___qaq
  • 当前回复8
  • 已保存回复8
  • 发布时间2021/10/2 12:56
  • 上次更新2023/11/4 05:10:07
查看原帖
求助手写二叉堆
556362
qwq___qaq楼主2021/10/2 12:56

题目 在题目讨论版里面发了没人回啊。

#include<bits/stdc++.h>
const int maxn=1e6+5;
using namespace std;
int n,heap[maxn],t,a[maxn],s,ans;
void down(int i,int n){
    while(i*2<=n){
        int T=i*2;
        if(T<n&&heap[T]>heap[T+1])
            T++;
        if(heap[i]>heap[T]){
            swap(heap[i],heap[T]);
            i=T;
        } else
            break;
    }
}
int main(){
    scanf("%d%d",&t,&n);
    ans=n;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        heap[i]=a[i];
    }
    t--;
    while(1){
        if(ans==1)
            break;
        if(heap[ans]-heap[1]<=heap[2])
            break;
        heap[1]=heap[ans--]-heap[1];
        down(1,ans);
    }
    printf("%d\n",ans);
    while(t--){
        ans=n;
        scanf("%d",&s);
        for(int i=1,k,t;i<=s;i++){
            scanf("%d%d",&k,&t);
            a[k]=t;
        }
        for(int i=1;i<=n;i++)
            heap[i]=a[i];
        while(1){
            if(ans==2)
                ans--;
            if(ans==1)
                break;
            if(heap[ans]-heap[1]<=heap[2])
                break;
            heap[1]=heap[ans--]-heap[1];
            down(1,ans);
        }
        printf("%d\n",ans);
    }
    return 0;
}
2021/10/2 12:56
加载中...