PAT L2-002 链表去重

https://pintia.cn/problem-sets/994805046380707840/problems/994805072641245184

给定一个带整数键值的链表 L，你需要把其中绝对值重复的键值结点删掉。即对每个键值 K，只有第一个绝对值等于 K 的结点被保留。同时，所有被删除的结点须被保存在另一个链表上。例如给定 L 为 21→-15→-15→-7→15，你需要输出去重后的链表 21→-15→-7，还有被删除的链表 -15→15。

输入格式：

输入在第一行给出 L 的第一个结点的地址和一个正整数 N（≤，为结点总数）。一个结点的地址是非负的 5 位整数，空地址 NULL 用 − 来表示。

随后 N 行，每行按以下格式描述一个结点：

地址 键值 下一个结点

其中地址是该结点的地址，键值是绝对值不超过1的整数，下一个结点是下个结点的地址。

输出格式：

首先输出去重后的链表，然后输出被删除的链表。每个结点占一行，按输入的格式输出。

输入样例：

00100 5
99999 -7 87654
23854 -15 00000
87654 15 -1
00000 -15 99999
00100 21 23854

输出样例：

00100 21 23854
23854 -15 99999
99999 -7 -1
00000 -15 87654
87654 15 -1

代码：

#include <bits/stdc++.h>
using namespace std;

const int maxn = 2e7 + 10;
int st, N;

struct Node{
    int nx;
    int address;
    int val;
}node[maxn];

vector<Node> line;
map<int, int> mp;
vector<Node> ans, del;

int main() {
    scanf("%d%d", &st, &N);
    Node n;
    for(int i = 0; i < N; i ++) {
        scanf("%d%d%d", &n.address, &n.val, &n.nx);
        node[n.address] = n;
    }

    int last = st;
    while(last != -1) {
        line.push_back(node[last]);
        last = node[last].nx;
    }

    for(int i = 0; i < line.size(); i ++) {
        if(mp[abs(line[i].val)] == 0) {
            ans.push_back(line[i]);
            mp[abs(line[i].val)] = 1;
        } else del.push_back(line[i]);
    }

    if(ans.size() > 0){
        for(int i = 0; i < ans.size() - 1; i ++)
            printf("%05d %d %05d\n", ans[i].address, ans[i].val, ans[i + 1].address);

        printf("%05d %d -1\n", ans[ans.size() - 1].address, ans[ans.size() - 1].val);
    }

    if(del.size() == 0) printf("");
    else if(del.size() == 1) printf("%05d %d -1\n", del[0].address, del[0].val);
    else {
        printf("%05d %d %05d\n", del[0].address, del[0].val, del[1].address);
        for(int i = 1; i < del.size() - 1; i ++)
            printf("%05d %d %05d\n", del[i].address, del[i].val, del[i+1].address);

        printf("%05d %d -1\n", del[del.size() - 1].address, del[del.size() - 1].val);
    }
    return 0;
}