PTA团体程序设计天梯赛-练习集 L2-002 链表去重（带思路和详细注释）

给定一个带整数键值的链表 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

思路：用一个结构体记录链表的地址键值和下一个节点，用结构体的num值处理先输出去重后的链表，然后输出被删除的链表，我们将没有删除的结点的num标记为cnt1，将需要删除的结点的num标记为inf + cnt2，因为一开始初始化为了2 * inf，所以我们通过对num排序可以达到：num = [0,cnt1)为不删除结点，num = [cnt1,cnt1+cnt2)为删除结点，num = [cnt1+cnt2,2 * inf)为无效结点

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
const int inf = 100010;
struct Node{
	int address;
	int key;
	int end;
	int num;
};	
int vis[inf];
Node node[inf];
//依据节点num值排序，保证先输出去重后的链表，然后输出被删除的链表
int cmp(Node a,Node b){
	return a.num < b.num;
}

int main(){
	int begin,n;
	scanf("%d %d", &begin, &n);
	int a;
	//初始化为2*inf
	for(int i = 0; i < inf; i++) {
        node[i].num = 2 * inf;
    }
    //输入节点
	for(int i = 0;i < n;i++){
		scanf("%d", &a);
        scanf("%d%d", &node[a].key, &node[a].end);
        node[a].address = a;
	}	
	//cnt1用来记录无重复键值节点有多少个
	///cnt2用来记录重复键值节点有多少个
	int cnt1 = 0,cnt2 = 0;
	//遍历链表
	for(int i = begin;i != -1;i = node[i].end){
		//无重复键值则节点的num值等于cnt1
		if(vis[abs(node[i].key)] == 0){
			vis[abs(node[i].key)] = 1;
			node[i].num = cnt1;
			cnt1 += 1;
		}
		//有重复键值节点的num值等于cnt2
		else{
			node[i].num = inf + cnt2;
			cnt2 += 1;
		} 
	}
	/*依据节点num值进行排序，可以保证先输出去重后的链表，然后输出
	被删除的链表*/
	sort(node, node + inf, cmp);
	for(int i = 0;i < cnt1+cnt2;i++){
		/*去重后的链表的最后一个节点和被删除的链表的最后一个节点的next
		是为空的，需要特殊处理*/
		if(i != cnt1 -1 && i != cnt1 + cnt2 - 1){
			printf("%05d %d %05d\n", node[i].address, node[i].key, node[i+1].address);
		}
		else{
			printf("%05d %d -1\n", node[i].address, node[i].key);
		}
	}
	return 0;
}