【优先队列】【链表】【priority_queue】【list】hdu 1434

最新推荐文章于 2024-10-28 14:12:21 发布

duyiwuer2009

最新推荐文章于 2024-10-28 14:12:21 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： C/C++ STL OJ题目文章标签： STL

本文链接：https://blog.youkuaiyun.com/duyiwuer2009/article/details/23709939

C/C++ 同时被 3 个专栏收录

49 篇文章

订阅专栏

STL

16 篇文章

订阅专栏

OJ题目

6 篇文章

订阅专栏

本文对比分析了两种解决方法：使用优先队列（小根堆）和链表。优先队列方法在内存占用和运行速度上表现更优，而链表方法中，list.merge() 保证了有序。在优先队列实现中，采用了greater functor并避免多次建堆，链表实现则需要注意sort的less functor和重载操作。

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

方法一：优先队列（9203MS 21048K）

方法二：链表（9640MS 33620K）

性能分析：优先队列的方法内存占用较少，速度也更快。

注意：方法一中使用的是 greater functor, 需要重载 ">", 方法二 sort 采用默认的 less functor, 需要重载 "<".

方法一：优先队列

1. 这里的优先队列应该是使用小根堆，因此用 greater

2. 这里用构造函数初始化优先队列，而不是先建立空队列，然后不停 push. 这样做可以省去多次建堆操作，不初始化用的 vector 浪费了空间。

/*
 * hdu 1434
 * http://acm.hdu.edu.cn/showproblem.php?pid=1434
 * 【优先队列】【priority_queue】
 * 9203MS	21048K
 * 1. 这里的优先队列应该是使用小根堆，因此用 greater
 * 2. 这里用构造函数初始化优先队列，而不是先建立空队列，然后不停 push.
 *    这样做可以省去多次建堆操作，不初始化用的 vector 浪费了空间。
 */
#include <iostream>
#include <queue>
#include <vector>
#include <functional>
#include <string>
using namespace std;

struct SPassenger
{
    int     iRp;
    string  strName;
};

bool operator>(const SPassenger &oPl, const SPassenger &oPr)
{
    if (oPl.iRp == oPr.iRp)
    {
        return (oPl.strName < oPr.strName);
    }
    return (oPl.iRp > oPr.iRp);
}

typedef priority_queue<SPassenger, vector<SPassenger>,
                        greater<SPassenger> > Train;

int main()
{
    vector<Train>       vecTrains;
    int                 iTrainNum, iCmdNum, iPsgrNum;
    SPassenger          oTmpPsgr;
    string              strTmp;
    int                 i, j, iRp;
    greater<SPassenger> oTmpComp;

    while (cin >> iTrainNum >> iCmdNum)
    {
        while (iTrainNum--)
        {
            vector<SPassenger>  vecTmp;

            cin >> iPsgrNum;
            while (iPsgrNum--)
            {
                cin >> oTmpPsgr.strName >> oTmpPsgr.iRp;
                vecTmp.push_back(oTmpPsgr);
            }
            Train oTmpTrain(oTmpComp, vecTmp);
            vecTrains.push_back(oTmpTrain);
        }

        while (iCmdNum--)
        {
            cin >> strTmp;
            if ("GETON" == strTmp)
            {
                cin >> i >> oTmpPsgr.strName >> oTmpPsgr.iRp;
                vecTrains[i-1].push(oTmpPsgr);
            }
            else if ("JOIN" == strTmp)
            {
                cin >> i >> j;
                while (!vecTrains[j-1].empty())
                {
                    vecTrains[i-1].push(vecTrains[j-1].top());
                    vecTrains[j-1].pop();
                }
            }
            else    // GETOUT
            {
                cin >> i;
                cout << vecTrains[i-1].top().strName << endl;
                vecTrains[i-1].pop();
            }
        }

        vecTrains.clear();
    }

    return 0;
}

方法二：链表

之所以用 list, 主要是考虑到合并操作时 list.merge() 能保证有序（前提是参数 list 有序）

/*
 * hdu 1434
 * http://acm.hdu.edu.cn/showproblem.php?pid=1434
 * 9640MS	33620K
 * 【链表】【list】
 * 之所以用 list, 主要是考虑到合并操作时 list.merge() 能保证有序（前提是参数 list 有序）
 */
#include <iostream>
#include <vector>
#include <list>
#include <string>
using namespace std;

struct SPassenger
{
    int     iRp;
    string  strName;
};

bool operator<(const SPassenger &oPl, const SPassenger &oPr)
{
    if (oPl.iRp == oPr.iRp)
    {
        return (oPl.strName > oPr.strName);
    }
    return (oPl.iRp < oPr.iRp);
}

typedef list<SPassenger> Train;
typedef list<SPassenger>::iterator ItTrain;

int main()
{
    vector<Train>       vecTrains;
    int                 iTrainNum, iCmdNum, iPsgrNum;
    SPassenger          oTmpPsgr;
    string              strTmp;
    int                 i, j, iRp;
    ItTrain             itTrain;

    while (cin >> iTrainNum >> iCmdNum)
    {
        while (iTrainNum--)
        {
            Train oTmpTrain;

            cin >> iPsgrNum;
            while (iPsgrNum--)
            {
                cin >> oTmpPsgr.strName >> oTmpPsgr.iRp;
                oTmpTrain.push_back(oTmpPsgr);
            }
            oTmpTrain.sort();
            vecTrains.push_back(oTmpTrain);
        }

        while (iCmdNum--)
        {
            cin >> strTmp;
            if ("GETON" == strTmp)
            {
                cin >> i >> oTmpPsgr.strName >> oTmpPsgr.iRp;
                itTrain = vecTrains[i-1].begin();
                while ( !(oTmpPsgr < *itTrain) &&
                       (itTrain != vecTrains[i-1].end()) )
                {
                    ++itTrain;
                }
                vecTrains[i-1].insert(itTrain, oTmpPsgr);
            }
            else if ("JOIN" == strTmp)
            {
                cin >> i >> j;
                vecTrains[i-1].merge(vecTrains[j-1]);
            }
            else    // GETOUT
            {
                cin >> i;
                cout << vecTrains[i-1].front().strName << endl;
                vecTrains[i-1].pop_front();
            }
        }

        vecTrains.clear();
    }

    return 0;
}