题解 | 【模板】堆
【模板】堆
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; }