PTA 天梯赛 L2-002 链表去重 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

解题思路：

模拟过程即可，绝对值重复的地址放入一个数组中，绝对值不重复的地址放入另一数组中，为了判断绝对值是否重复，我们需要一个 bool 型的判重数组。

代码示例：

#include <iostream>
#include <vector>

using namespace std;

struct Node
{
    int data;
    int next;
}res[100005];

vector<int> stop; //留下
vector<int> del; //移除
bool isRepet[100005] = { 0 }; //判重数组

int main()
{
    int first_ip, n;
    cin >> first_ip >> n;

    while(n--)
    {
        int ip;
        cin >> ip;
        cin >> res[ip].data >> res[ip].next;
    }

    int p = first_ip;

    while (p != -1)
    {
        int x = abs(res[p].data); //获取节点绝对值

        if (isRepet[x]) //节点绝对值重复就要将地址存入移除数组del中
        {
            del.emplace_back(p);
        }
        else //节点绝对值没重复就将地址存入留下数组stop中
        {
            stop.emplace_back(p);
            isRepet[x] = true; //标记此节点绝对值已存在
        }

        p = res[p].next;
    }


    //先输出留下的地址
    int i = 0;
    for (; i < stop.size() - 1; i++)
    {
        printf("%05d %d %05d\n", stop[i], res[stop[i]].data, stop[i + 1]);
    }
    printf("%05d %d -1\n", stop[i], res[stop[i]].data);

    //再输出已被移除的地址
    if (del.size() > 0) //如果没有被移除的地址 就不输出任何东西
    {
        int j = 0;
        for (; j < del.size() - 1; j++)
        {
            printf("%05d %d %05d\n", del[j], res[del[j]].data, del[j + 1]);
        }
        printf("%05d %d -1\n", del[j], res[del[j]].data);
    }

    return 0;
}

运行结果：