题解 | 【模板】堆

【模板】堆

https://www.nowcoder.com/practice/13f61c8c92404f5ea5d6fa4c692869fb

#include <stdio.h>
#include <stdlib.h>
void swap(int* a, int m, int n) {
    int temp;
    temp = a[m];
    a[m] = a[n];
    a[n] = temp;
}
void push(int* a, int x, int* length) {
    if (*length == 0) {
        a[0] = x;
        (*length)++;
    } else {
        int flag = *length;
        a[flag] = x;
        (*length)++;
        while (flag > 0 && a[(flag - 1) / 2] < a[flag]) {
            swap(a, flag, (flag - 1) / 2);
            flag = (flag - 1) / 2;
        }
    }
}
void top(int* a, int* length) {
    if (*length == 0) {
        printf("empty\n");
    } else {
        printf("%d\n", a[0]);
    }
}
void pop(int* a, int* length) {
    if (*length == 0) {
        printf("empty\n");
    } else {
        printf("%d\n", a[0]);
        a[0] = a[(*length) - 1];
        (*length)--;
        int flag = 0;
        while (1) {
            int left = 2 * flag + 1;
            int right = 2 * flag + 2;
            int max = flag;
            if (left < *length && a[left] > a[max]) {
                max = left;
            }
            if (right < *length && a[right] > a[max]) {
                max = right;
            }
            if (max != flag) {
                swap(a, flag, max);
                flag = max;
            } else {
                break;
            }
        }
    }
}
int main() {
    int n;
    int x;
    int a[100000];
    int length = 0;
    char s[5];
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
        scanf("%s", s);
        if (s[0] == 'p' && s[1] == 'u') {
            scanf("%d", &x);
            push(a, x, &length);
        } else if (s[0] == 't') {
            top(a, &length);
        } else if (s[0] == 'p' && s[1] == 'o') {
            pop(a, &length);
        }
    }
    return 0;
}

全部评论

相关推荐

昨天 14:55
门头沟学院 Java
点赞 评论 收藏
分享
Gaynes:查看图片
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务