堆,优先队列,堆排序, 大顶堆

#include <stdio.h>
#include <time.h>
#include <stdlib.h>

#define swape(a, b) ({\
                __typeof(a) temp = a;\
                a = b; b = temp;\
                })

typedef struct priority_queue {
        int *data, cnt, size;
} Priority_queue;

Priority_queue *init(int n) {
        Priority_queue *p = (Priority_queue *)malloc(sizeof(Priority_queue));
        p->data = (int *)malloc(sizeof(int) * (n + 1));
        p->cnt = 0; p->size = n;
        return p;
}

void clear(Priority_queue *p) {
        if (!p) return ;
        free(p->data);
        free(p);
        return ;
}

int empty(Priority_queue *p) {
        return !q->cnt;
}

int top(Priority_queue *p) {
        return p->data[1];
}

int push(Priority_queue *p, int n) {
        if(!p || p->cnt >= p->size) return 0;
        p->data[++p->cnt] = n;
        int ind = p->cnt;
        while(ind >> 1 && n > p->data[ind >> 1]) {
                swape(p->data[ind >> 1], p->data[ind]);
                ind >>= 1;
        }
        return 1;
}

int pop(Priority_queue *p) {
        if(!p || empty(p)) return 0;
        p->data[1] = p->data[p->cnt--];
        int ind = 1, key = 1;
        while(ind << 1 < p->cnt) {
                p->data[ind << 1] > p->data[ind] && (key = ind << 1);
                p->cnt > key && p->data[key] < p->data[ind << 1 | 1] && (key = ind << 1 | 1);
                if (ind == key) break;
                swape(p->data[ind], p->data[key]);
                ind = key;
        }
        return 1;
}

int main() {
        int k;
        srand(time(0));
#define MAX_N 20
        Priority_queue *p = init(MAX_N);
        for (int i = 1; i <= MAX_N; i++) k = rand() % 100, printf(" %d,", k), push(p, k);
        printf("\n"); for (int i = 1; i <= MAX_N; i++) printf(" %d ", p->data[i]);printf("\n");
        for (int i = 1; i <= MAX_N; i++) printf(" %d,", top(p)), pop(p);
#undef MAX_N
        clear(p);
        printf("\n");
        return 0;
}

posted @ 2021-08-07 00:28  代码附体  阅读(34)  评论(0编辑  收藏  举报