L2-002 链表去重

给定一个带整数键值的链表 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 <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
 
using namespace std;
const int N = 1e5 + 3;
 
struct node{
    int data;
    int next;
}list[N];
 
int main()
{
    int head, n;
    scanf("%d%d", &head, &n);
    while(n--)
    {
        int id;
        scanf("%d", &id);
        scanf("%d%d", &list[id].data, &list[id].next);
    }
    int ans[N], k1 = 0;
    int res[N], k2 = 0;
    bool vis[N];
    memset(vis, 0, sizeof(vis));
    int p = head;
    while(p != -1)
    {
        int m = abs(list[p].data);
        if(!vis[m])
        {
            ans[k1++] = p;
            vis[m] = 1;
        }
        else
        {
            res[k2++] = p;
        }
        p = list[p].next;
    }
    printf("%05d", head);
    for(int i = 1; i < k1; i++)
    {
        printf(" %d %05d\n%05d", list[ans[i-1]].data, ans[i], ans[i]);
    }
    printf(" %d -1\n", list[ans[k1-1]].data);
    if(k2 > 0)
    {
        printf("%05d", res[0]);
        for(int i = 1; i < k2; i++)
        {
            printf(" %d %05d\n%05d", list[res[i-1]].data, res[i], res[i]);
        }
        printf(" %d -1", list[res[k2-1]].data);
    }
    return 0;
}