【PAT甲级】1133 Splitting A Linked List (25 分)【链表】

1133 Splitting A Linked List

---

题意说明

给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而 $[0,K]$区间内的元素都排在大于 K 的元素前面。

但每一类内部元素的顺序是不能改变的。

例如：给定链表为 18→7→-4→0→5→-6→10→11→-2，K 为 10，则输出应该为 -4→-6→-2→7→0→5→10→18→11。

---

思路

• 首先，用数组模拟链表，可以参考这篇博文
  https://blog.csdn.net/weixin_43864567/article/details/116328487
• e[i]数组用来表示地址为i的节点的值为多少（这个数组是用来存储节点值的，为char类型）
• ne[i]数组用来表示地址为i的节点指向的下一个节点的地址为多少（这个数组是用来存储节点的地址的，为int类型）
• 对链表进行遍历，开三个vector数组a，b，c，若遇到节点存储的值为负值就存到a中，处于[0,k]区间就存到b中，否则就存到c中，最后，把a进行遍历输出，注意格式控制
• 将b和c插入到a中的操作：a.insert(a.end(),b.begin(),b.end());

---

代码

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

const int N = 1e5 + 10;

int e[N], ne[N];

int main()
{
    int head, n , k;
    cin >> head >> n >> k;
    for (int i = 0;i < n;i++)
    {
        int address, data, next;
        cin >> address >> data >> next;
        e[address] = data;
        ne[address] = next;
    }
    vector<int> a, b, c;
    for (int i = head;i != -1;i = ne[i])
    {
        if (e[i] < 0) a.push_back(i);
        else if (e[i] <= k) b.push_back(i);
        else c.push_back(i);
    }
    
    a.insert(a.end(),b.begin(),b.end());
    a.insert(a.end(),c.begin(),c.end());
    
    for (int i = 0;i <a.size();i ++)
    {
        if (i == a.size() - 1) printf("%05d %d -1\n", a[i], e[a[i]]);
        else
        {
            printf("%05d %d %05d\n", a[i], e[a[i]], a[i + 1]);
            //cout << a[i] << " " << e[a[i]] << " "<<a[i+1]<<"\n";
        }
    }

    
    return 0;
}