？（2015-3）优先队列的实现

最新推荐文章于 2021-09-15 22:27:59 发布

julia7_

最新推荐文章于 2021-09-15 22:27:59 发布

阅读量448

点赞数

CC 4.0 BY-SA版权

分类专栏：机试

本文链接：https://blog.youkuaiyun.com/weixin_35093872/article/details/88065887

机试专栏收录该内容

33 篇文章

订阅专栏

本文探讨了使用优先队列和Set数据结构实现实时任务调度的两种方法。通过具体代码示例，详细讲解了如何添加、删除任务以及获取最高优先级任务的过程。对比了两种实现方式的优缺点，帮助读者理解在不同场景下选择合适的数据结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

ADD N P：往队列里加入id为N的优先级为P的任务

NEXT:输出下一个最高优先级的任务的id，如果优先级相同输出id小的任务，若队列中没有任务输出-1

REMOVE N：移除id为N的任务

COUNT：输出队列中的任务数量

思路1：

采用set实现，但是不行...不会

#include <cstdio>
#include <string>
#include <set>
#include <iostream>
using namespace std;

struct node{
    int id;
    int p;
};
struct cmp{
    bool operator () (node a, node b){
        if(a.p == b.p){
            return a.id > b.id;         //优先级相同则id小的在前
        }
        return a.p < b.p;               //优先级大的在前
    }
};

set<node> s;

void Add(int id, int p){
    node a;
    a.id = id;
    a.p = p;
    s.insert(a);
    printf("添加成功\n");
}

void Next(){
    if(!s.empty()){
        set<node>::iterator it = s.begin();
        printf("下一个元素id为：%d\n", *it);
    }
    else{
        printf("-1\n");
    }
}

void Remove(int id){

}

void Count(){
    printf("%d", s.size());
}

int main(){
    string str;
    int id, p;
    cin >> str;
    while(1){
        if(str == "ADD"){
            scanf("%d%d", &id, &p);
            Add(id, p);
        }
        else if(str == "NEXT"){
            Next();
        }
        else if(str == "REMOVE"){
            scanf("%d", &id);
            Remove(id);
        }
        else if(str == "COUNT"){
            Count();
        }
        else{
            break;
        }
    }

}

思路2：

采用优先队列，发现remove难以实现

#include <queue>
#include <iostream>
using namespace std;

struct node{
    int id;
    int p;
};
struct cmp{
    bool operator () (node a, node b){
        if(a.p == b.p){
            return a.id > b.id;         //优先级相同则id小的在前
        }
        return a.p < b.p;               //优先级大的在前
    }
};

priority_queue<node, vector<node>, cmp> q;
priority_queue<node, vector<node>, cmp> temp;

void add(int id, int p){
    node a;
    a.id = id;
    a.p = p;
    q.push(a);
}

void next(){
    node a = q.top();
    printf("%d %d", a.id, a.p);
}

void Remove(int id){
    node a = q.top()；
    while(a.id != id){
        node b = a;
        q.pop();
        temp.push(b);
    }
}

int main(){
    string str;
    int id, p;
    cin >> str;
    while(1){
        if(str == 'ADD'){
            scanf("%d%d", &id, &p);
            add(id, p);
        }
        else if(str == 'NEXT'){
            next();
        }
        else if(str == 'REMOVE'){
            scanf("%d", &id);
            Remove(id);
        }
        else{
            break;
        }
    }
    return 0;
}