L2-002 链表去重

L2-002 链表去重

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

输入格式：

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

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

地址 键值 下一个结点

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

输出格式：

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

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

int main(){
	cout.sync_with_stdio(false);
	cout.tie(nullptr);
	int r,n;
	cin>>r>>n;
	vector<pair<int,int>>s(500005),deleted(500005);
	for(int i=0;i<n;i++){
		int a,b,c;cin>>a>>b>>c;
		s[a]={b,c};
	}
	int now = r,next=s[now].second,pre=-1,deleted_r=-1;
	set<int>st;
	while(now!=-1){
		st.insert(abs(s[now].first));
		while(st.count(abs(s[next].first)) &&next !=-1) {
			deleted[next]=s[next];
			deleted[next].second=-1;
			if(pre!=-1){
				deleted[pre].second=next;
			}else{
				deleted_r=next;
			}
			pre=next;
			next=s[next].second;
		}
		s[now].second=next;
		now=next;
		next=s[now].second;
	}
	now = r,next=s[now].second;
	while(now!=-1){
		if(s[now].second==-1) printf("%05d %d %d\n",now,s[now].first,s[now].second);
		else printf("%05d %d %05d\n",now,s[now].first,s[now].second);
		now=next;
		next=s[now].second;
	}
	now = deleted_r,next=deleted[now].second;
	while(now!=-1){
		if(deleted[now].second==-1) printf("%05d %d %d\n",now,deleted[now].first,deleted[now].second);
		else printf("%05d %d %05d\n",now,deleted[now].first,deleted[now].second);
		now=next;
		next=deleted[now].second;
	}
}