L2-002 链表去重 (25 分)

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

输入格式：

输入在第一行给出 L 的第一个结点的地址和一个正整数 N（≤10^5，为结点总数）。一个结点的地址是非负的 5 位整数，空地址 NULL 用 −1 来表示。随后 N 行，每行按以下格式描述一个结点：

地址 键值 下一个结点

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

输出格式：

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

输入样例：

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 = 1e5+10;
struct Dot{
    int self;
    int next;
};
Dot dot[MAXN];
int main(){
    int root;
    int n;
    cin>>root>>n;
    for(int i=0; i<n; i++) {
        int head;
        cin>>head;
        cin>>dot[head].self>>dot[head].next;
    }
    int cnt1 = 0;
    int res[MAXN];
    int cnt2 = 0;
    int del[MAXN];
    int vis[MAXN];
    fill(vis, vis + MAXN, 0);
    int r1 = root;
    while (r1 != -1) {
        if(vis[abs(dot[r1].self)] == 0) {
            vis[abs(dot[r1].self)] = 1;
            res[cnt1++] = r1;
        } else {
            del[cnt2++] = r1;
        }
        r1 = dot[r1].next;
    }
    for(int i=0; i<cnt1 - 1; i++) {
        printf("%05d %d %05d\n",res[i], dot[res[i]].self, res[i+1]);
    }
    printf("%05d %d -1\n",res[cnt1 - 1], dot[res[cnt1 - 1]].self);

    if(cnt2 > 0) {
        for(int i=0; i<cnt2 - 1; i++) {
            printf("%05d %d %05d\n",del[i], dot[del[i]].self, del[i+1]);
        }
        printf("%05d %d -1\n",del[cnt2 - 1], dot[del[cnt2 - 1]].self);
    }
    return 0;
}