PTA L2链表去重 (25分)

                                           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 maxn = 1e6;
struct Node{
	int nxt;
	int num;
}node[maxn],bin[maxn];
int main() {
	int n,nn,m,pre,bpre,bhead,cnt=0;
	cin>>n>>m;
	nn=n;
	for(int i =1;i<=m;i++) {
		int u,v,w;
		cin>>u>>w>>v;
		node[u].nxt = v;
		node[u].num = w;
	}
	map<int,int> mp;
	for(int i=n;i!=-1;i=node[i].nxt) {
		if(mp[abs(node[i].num)]==0) {
			pre = i;
			mp[abs(node[i].num)]++;
		}
		else {
			nn--;
			bin[i].num = node[i].num;
			bin[i].nxt = -1;
			cnt ++;
			if(cnt==1)
				bhead=i;
			else{
				bin[bpre].nxt = i;
			}
			bpre = i;
			node[pre].nxt = node[i].nxt;
		}
	}
	if(nn==n) {
		for(int i=n;i!=-1;i=node[i].nxt) {
			printf(node[i].nxt==-1?"%05d %d %d\n":"%05d %d %05d\n",i,node[i].num,node[i].nxt);	
		}	
	}
	else {
		//链表 删除重复的元素的结点 
		for(int i=n;i!=-1;i=node[i].nxt) {
			printf(node[i].nxt==-1?"%05d %d %d\n":"%05d %d %05d\n",i,node[i].num,node[i].nxt);
		}
		//保存被删除的结点	
		for(int i=bhead;i!=-1;i=bin[i].nxt) {
			printf(bin[i].nxt==-1?"%05d %d %d\n":"%05d %d %05d\n",i,bin[i].num,bin[i].nxt);
		}	
	}
	
	return 0;
}