7-4 链表去重 (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<iostream>
#include<set>
#include<cmath>
using namespace std;
const int N=1e5;
struct Node{
    int data,next,local;
}info[N];
int a[N], after[N],surplus[N];
int main(){
    int head,p,n;
    cin>>head>>n;
    for(int i=0;i<n;i++){
        cin>>p;
        cin>>info[p].data>>info[p].next;
        info[p].local=p;
    }int cnt1,cnt2,cnt3;
        cnt1=cnt2=cnt3=0;
    //a[]按顺序保存地址
    while(head!=-1){
        a[cnt1++]=head;
        head=info[head].next;
    }set<int>s;
    for(int i=0;i<cnt1;i++){
        if(s.find(abs(info[a[i]].data))==s.end()){
            //如果s中找不到当前索引i下的data
            after[cnt2++]=a[i];
            s.insert(abs(info[a[i]].data));
        }else surplus[cnt3++]=a[i];
    }for(int i=0;i<cnt2-1;i++){
        printf("%05d %d %05d\n",after[i],info[after[i]].data,after[i+1]);
    }printf("%05d %d -1\n",after[cnt2-1],info[after[cnt2-1]].data);
    if(cnt3>0){
        for(int i=0;i<cnt3-1;i++){
            printf("%05d %d %05d\n",surplus[i],info[surplus[i]].data,surplus[i+1]);
        }printf("%05d %d -1\n",surplus[cnt3-1],info[surplus[cnt3-1]].data);
    }return 0;
}