PTA程序设计天梯赛L2-002链表去重

前言

题目

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

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

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

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

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

输入样例：
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

题目大意

给定一个链表，已知每个节点的值和地址，求最终去重后的链表

预备知识（可选）

结构体模拟链表

思路和算法

本题的地址处理比较麻烦，但已知题目信息地址是非负的5位整数，于是可以用结构体数组的下标来表示地址。
对于去重操作，可以使用STL中的集合，set、unordered_set

代码

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

const int N = 1e5+10;
struct node{
    int val;
    int next;
}node[N];

int main() {
    int h, n;
    cin >> h >> n;
    for(int i = 0; i < n; i++) {
        int p, v, ne;
        cin >> p >> v >> ne;
        node[p].val = v;
        node[p].next = ne;
    }
    vector<int> res1, res2;
    unordered_set<int> se;
    for(int i = h; i != -1; i = node[i].next) {
        int m = abs(node[i].val);
        if(!se.count(m)) {//如果之前没有这个数，则插入，更新set
            se.insert(m);
            res1.push_back(i);
        }else {//如果有这个树，则删除
            res2.push_back(i);
        }
    }
    for(int i = 0; i < res1.size(); i++) {
        int t = res1[i];
        if(i == res1.size() - 1) printf("%05d %d -1\n", t, node[t].val);//删除的数字组成链表的最终地址为-1
        else printf("%05d %d %05d\n", t, node[t].val, res1[i+1]);
    }
    for(int i = 0; i < res2.size(); i++) {
        //cout << 1 << endl;
        int t = res2[i];
        if(i == res2.size() - 1) printf("%05d %d -1\n", t, node[t].val);
        else printf("%05d %d %05d\n", t, node[t].val, res2[i+1]);
    }
    
    return 0;
}