HDU 幸福列车优先队列

smile6a6a

于 2022-12-24 11:16:44 发布

阅读量97

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/m0_73832482/article/details/128426921

版权

这篇博客介绍了一种使用优先队列优化解决火车乘客管理问题的方法。列车长linle根据人品值和名字顺序对乘客进行排序并执行各种操作，如上车、合并列车和踢出人品最差的乘客。博主分享了如何利用C++的结构体和友元函数实现优先队列，以高效地执行这些操作，从而避免超时错误。博客中还包含了完整的代码示例，演示了如何处理大规模输入数据。

摘要由CSDN通过智能技术生成

幸福列车

一批幸福的列车即将从杭州驶向幸福的终点站――温州，身为总列车长的linle有一些奇怪的癖好。

他会记录下全部乘客的名字（name）和他们的人品值（RP），根据这些将他们排序，并不时地从某辆列车里踢出人品最不好（RP值最低）的一个人，当两个人人品一样不好时，他就会踢出名字难听的人（linle认为按字典顺序，排在越在后面的人名字越难听）。

当然出于列车行驶需要，他还会不时的发布一些命令，比如让某个乘客上车，合并某两辆列车等。

linle的上一任秘书***因为不能高效地执行他的这些命令而被炒鱿鱼，他现在正在寻觅新的秘书人选，你能不能胜任呢？（谢绝男士，待遇丰厚~）

Input

本题包含多组测试，请处理到文件结束。对于每一组测试，第一行包含两个整数 N ，M ，表示一共有N( N<=10000 ) 辆列车，执行M( M<=10000 )次操作。接下来有 N （从1开始记数）辆列车的信息，每辆列车先有一个数字 Xi(1 <= Xi <= 100 )，表示该列车有Xi个乘客，接下来Xi行乘客信息，每个乘客包含名字（20个字符以内，不包含空白符）和人品（0<= RP <=30000）。再接下来有 M 行操作信息，一共有3种操作，分别为

GETON Xi name RP 表示有一个叫name的人品为RP的人登上第Xi列车

JOIN Xi Xj 表示有将第Xj辆列车合并到Xi辆列车

GETOUT Xi 表示从第Xi辆列车踢出一个人品最差的人

测试数据保证每个操作均合法，即不会将已经被合并到其他列车的列车再进行合并，也不会从一辆空列车里踢出乘客

Output

对于每个 GETOUT 命令，输出被踢出的那个人的名字

Huge input, scanf is recommended.

题解

我的第一思路是用vector去模拟链表做，结果发现排序要花太多的时间，超时了，然后就想到了教练和我讲过的优先队列，可以大大的减小时间复杂度。

因为人品和名字是在一起的，所以这里要用到结构体，并且写一个重载函数来规定优先顺序。

struct comp
{
    char* s;//存名字
    int r;//存人品
    friend bool operator<(comp x,comp y)//友元函数重载
    {
        string x0 = x.s; string y0 = y.s;//用string容器以方便比较大小
        if (x.r == y.r)return x0 < y0;//当人品相同时比较哪个名字丑
        return x.r > y.r;//比较人品
    }
};

然后就是创建优先队列，这里注意要在每一个新测试数据时定义一个新的优先队列，或者清空之前的队列，否则会被上一个数据干扰。

priority_queue <comp> x[10005];


然后一个要注意的点就是这题的数据量很大，所以最好用scanf输入，并且变量的定义尽量靠外。

#include<iostream>
#include<string>
#include<cstring>
#include<queue>
using namespace std;
struct comp
{
    char* s;//存名字
    int r;//存人品
    friend bool operator<(comp x,comp y)//友元函数重载
    {
        string x0 = x.s; string y0 = y.s;//用string容器以方便比较大小
        if (x.r == y.r)return x0 < y0;//当人品相同时比较哪个名字丑
        return x.r > y.r;//比较人品
    }
};
bool com(const char* a,const char* b)//比较函数，来确定操作种类
{
    int len = strlen(b);
    for (int i = 0; i < len; i++)
    {
        if (*a != *b)return 0;//若不同返回0。
        a++;
        b++;
    }
    return 1;
}
int main()
{
    int n, m;
    while (scanf("%d%d", &n, &m) != EOF)//多组数据
    {   
        int a;
        priority_queue <comp> x[10005];//创建优先队列
        //开始输入操作
        int c1;
        for (int i = 1; i <= n; i++)
        {           
            scanf("%d",&c1);
            for (int j = 1; j <= c1; j++)
            {
                char*s1 = (char*)malloc(sizeof(char) * 10);//预存空间，未初始化的字符型指针不能访问。
                scanf("%s", s1);
                scanf("%d", &a);
                x[i].push({s1,a});//压入队列
            }
        }
        for (int i = 1; i <= m; i++)
        {
            char* s;
            s = (char*)malloc(sizeof(char) * 10);//预存空间
            scanf("%s", s);
            int a, b;//在这定义变量以减少变量定义次数
            if (com(s ,"GETOUT"))
            {
                scanf("%d", &a);
                printf("%s\n", x[a].top().s);
                x[a].pop();//踢出，弹出队列
            }
            if (com(s,"JOIN"))
            {
                scanf("%d%d", &a, &b);
                while (x[b].size())//让x[b]队列加入队列x[a]
                {
                    x[a].push(x[b].top());
                    x[b].pop();
                }
            }
            if (com(s,"GETON"))
            {
                scanf("%d%s%d", &a, s, &b);//上车
                x[a].push({s,b});
            }
        }
    }
    return 0;
}