个人练习-PAT甲级-1133 Splitting A Linked List

题目链接https://pintia.cn/problem-sets/994805342720868352/problems/994805346776760320

题目大意：给一个链表的头地址和数字K，要求将链表重排，按顺序为元素【小于0】【在[0, K]之间】【大于K】三部分

思路：简单题，从头遍历链表即可。开始时想用arr[]这个vector的数组直接按顺序输出，结果发现要处理某一个vector为空的情况（对应上面三部分中有一部分缺失的情况），有点麻烦，于是新开一个vector再存一遍就没那么多事了。

完整代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <algorithm>
#include <queue>
#include <cstring>
#include <string>
#include <map>
#include <set>


using namespace std;


class Node{
public:
    int add;
    int data;

    Node(){}
    Node(int add, int data) {
        this->add = add;
        this->data = data;
    }
};


int get_data[100000];
int get_next[100000];
vector<Node> arr[3];
vector<Node> ret;


int main() {
    int head, N, K;
    scanf("%d %d %d", &head, &N, &K);

    memset(get_next, -1, sizeof(get_next));

    for (int i = 0; i < N; i++) {
        int add, data, next;
        scanf("%d %d %d", &add, &data, &next);
        get_next[add] = next;
        get_data[add] = data;
    }

    int add = head;
    while (add != -1) {
        int data = get_data[add];
        if (data < 0)
            arr[0].push_back(Node(add, data));
        else if (data >= 0 && data <= K)
            arr[1].push_back(Node(add, data));
        else
            arr[2].push_back(Node(add, data));
        add = get_next[add];
    }




    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < arr[i].size(); j++) {
            ret.push_back(arr[i][j]);
        }
    }

    for (int i = 0; i < ret.size(); i++) {
        if (i != ret.size()-1)
            printf("%05d %d %05d\n", ret[i].add, ret[i].data, ret[i+1].add);
        else
            printf("%05d %d -1\n", ret[i].add, ret[i].data);
    }


    return 0;
}