天梯赛 L2-002 链表去重 (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 <bits/stdc++.h>
using namespace std;
const int N=1000005;
int n,s,s2,d,d2;
int a[N],ne[N],nex; 
bool st[N];
int main()
{
	cin>>s>>n;
	memset(ne,-1,sizeof ne);
	while(n--)
	{
		int x,y,z;
		cin>>x>>y>>z;
		a[x]=y;
		ne[x]=z;
	}
	d=s;s2=-1;d2=-1;
	for(int i=s;i!=-1;i=nex)
	{
		int ab=abs(a[i]);
		nex=ne[i];
		if(st[ab]){
			if(s2==-1)s2=i;
			if(d2!=-1)ne[d2]=i;
			d2=i;
			ne[i]=-1;
		}else {
			ne[d]=i;
			d=i;
			ne[i]=-1;
		}
		st[ab]=true;
	}
	for(int i=s;i!=-1;i=ne[i])
	{
		if(ne[i]!=-1)
		printf("%05d %d %05d\n",i,a[i],ne[i]);
		else printf("%05d %d %d\n",i,a[i],ne[i]);
	}
	for(int i=s2;i!=-1;i=ne[i])
	{
		if(ne[i]!=-1)
		printf("%05d %d %05d\n",i,a[i],ne[i]);
		else printf("%05d %d %d\n",i,a[i],ne[i]);
	}
	return 0;
}

二刷：

#include <bits/stdc++.h>
using namespace std;
const int N=1000004;
int w[N],ne[N],d,s2=-1,d2,nex;
map<int,bool> st;

int main(){
	int s,n;cin>>s>>n;
	for(int i=1;i<=n;++i){
		int x,y,z;cin>>x>>y>>z;
		w[x]=y;
		ne[x]=z;
	}
	d=s;
	for(int i=s;i!=-1;i=nex){
		nex=ne[i];
		int x=w[i];
		if(st[abs(x)]){
			if(s2==-1)s2=d2=i;
			else {
				ne[d2]=i;
				d2=i;
			}
		}else {
			ne[d]=i;
			d=i;
		}
		st[abs(x)]=true;
	}
	ne[d]=ne[d2]=-1;
	for(int i=s;i!=-1;i=ne[i])
	{
		if(ne[i]!=-1)
		printf("%05d %d %05d\n",i,w[i],ne[i]);
		else printf("%05d %d %d\n",i,w[i],ne[i]);
	}
	for(int i=s2;i!=-1;i=ne[i])
	{
		if(ne[i]!=-1)
		printf("%05d %d %05d\n",i,w[i],ne[i]);
		else printf("%05d %d %d\n",i,w[i],ne[i]);
	}
	return 0;
}