天梯赛L2——002 链表去重

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

code：

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
const int maxn=100000;
struct NODE{
	int address;
	int key;
	int next;
	int num;
};
int cnt1,cnt2;
set<int> visit; 
bool cmp1(NODE a,NODE b){
	return a.num<b.num;
}
int main(){
	int head;
	int n;
	cin>>head>>n;
	NODE nodes[maxn];
	for(int i=0;i<maxn;i++) nodes[i].num=maxn*2;
	for(int i=0;i<n;i++){
		int add,k,ne;
		cin>>add>>k>>ne;
		nodes[add].address=add;
		nodes[add].key=k;
		nodes[add].next=ne; 
	}
	for(int i=head;i!=-1;i=nodes[i].next){
		if(visit.find(abs(nodes[i].key))==visit.end()){//不存在
			nodes[i].num=cnt1;
			cnt1++; 
			visit.insert(abs(nodes[i].key));
		}else{
			nodes[i].num=cnt2+maxn;//若存在 从maxn开始编号 
			cnt2++;
		}
	}
	sort(nodes,nodes+maxn,cmp1);//编号从小到大排序
	for(int i=0;i<n;i++){
		if(i!=cnt1-1&&i!=cnt1+cnt2-1){	//删除与未删除的最后一个是-1 
			printf("%05d %d %05d\n",nodes[i].address,nodes[i].key,nodes[i+1].address); 
		}else{
			printf("%05d %d -1\n",nodes[i].address,nodes[i].key);
		}
	} 
} 

自己写不出来，借鉴了大佬的。大佬的想法很巧妙，通过编号将链表分成没有被删除与被删除两块，被删除的链表编号比所有没有被删除的编号都要大，这样在按编号从小到大排序后，就可以按照顺序先输出没有被删除的，再输出被删除的。利用cnt1统计没有被删除的个数，以便输出最后一个的下一个地址是-1。