PTA 链表去重

C++给定一个带整数键值的链表 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;
struct ListNode
{
    int data;
    int next;
} a[100100];
int main()
{
    int b[100100]; 
    int flag[100100] = {0}; 
    int first, n;
    cin >> first >> n;

    for (int i = 0; i < n; i++)
    {
        int add;
        cin >> add;
        cin >> a[add].data >> a[add].next;
    }
    int m = first;
    int value = abs(a[m].data); 
    printf("%05d %d", first, a[m].data);
    flag[value] = 1;
    int cnt = 0;
    while (1)
    {
        m = a[m].next;
        if (m == -1)
        {
            cout << " -1\n";
            break;
        }
        value = abs(a[m].data);
        if (flag[value] == 0)
        {
            printf(" %05d\n%05d %d", m, m, a[m].data);
            flag[value] = 1;
        }
        else
        {
            b[cnt] = m; 
            cnt++;
        }
    }
    if (cnt) 
    {
        for (int i = 0; i <= cnt - 2; i++)
        {
            printf("%05d %d %05d\n", b[i], a[b[i]].data, b[i + 1]);
        }
        printf("%05d %d -1", b[cnt - 1], a[b[cnt - 1]].data); 
    }
}

提交结果：