PAT乙级真题 1075 链表元素分类 C++实现（测试点5：用map会超时）

题目

给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而 [0, K] 区间内的元素都排在大于 K 的元素前面。但每一类内部元素的顺序是不能改变的。例如：给定链表为 18→7→-4→0→5→-6→10→11→-2，K 为 10，则输出应该为 -4→-6→-2→7→0→5→10→18→11。
输入格式：
每个输入包含一个测试用例。每个测试用例第 1 行给出：第 1 个结点的地址；结点总个数，即正整数N (≤10^ 5)；以及正整数K (≤10^3​​ )。结点的地址是 5 位非负整数，NULL 地址用 −1 表示。
接下来有 N 行，每行格式为：
Address Data Next
其中 Address 是结点地址；Data 是该结点保存的数据，为 [−10^ 5,10 ^ 5] 区间内的整数；Next 是下一结点的地址。题目保证给出的链表不为空。
输出格式：
对每个测试用例，按链表从头到尾的顺序输出重排后的结果链表，其上每个结点占一行，格式与输入相同。
输入样例：
00100 9 10
23333 10 27777
00000 0 99999
00100 18 12309
68237 -6 23333
33218 -4 00000
48652 -2 -1
99999 5 68237
27777 11 48652
12309 7 33218
输出样例：
33218 -4 68237
68237 -6 48652
48652 -2 12309
12309 7 00000
00000 0 99999
99999 5 23333
23333 10 00100
00100 18 27777
27777 11 -1

思路

用空间换时间，遍历链表节点，将小于0的、[0，k]内的、大于k的分别取出来存到新的链表中，最后将这三个链表拼到一起输出即可。

一开始用string存储地址（不用再管输出格式了），用map模拟存储链表，结果测试点5超时。

后换成int存储地址，用vector模拟存储链表，终于压线通过。

代码

用map存储链表，拼接链表

实实在在拼接链表的方法，用map模拟存储链表，测试点5超时。

代码如下：

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

struct Node{
   
   
    int data;
    string next;
};

int main()
{
   
   
    string head;
    int n, k;
    cin >> head >> n >> k;

    map<string, Node*> nodeMap;
    for (int i=0; i<n; i++){
   
   
        Node *node = new Node();
        string addr;
        cin >> addr >> node->data >> node->next;
        nodeMap[addr] = node;
    }

    map<string, Node*> sortMap;
    string h1 = "h1";
    string h2 = "h2";
    string h3 = "h3";
    Node *node = new Node();
    sortMap[h1] = node;
    node = new Node();
    sortMap[h2] = node;
    node = new Node();
    sortMap[h3] = node;
    string r1 = h1;
    string r2 = h2;
    string r3 = h3;

    for (string h=head; h!="-1"; h=nodeMap[h]->next){
   
   
        if (nodeMap[h]->data<0){
   
   
            node = new Node();