【PTA】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

实现思路

主要是处理绝对值重复的点，设置 delPre 来处理被删除节点所在的链表。
我用了map，，，但感觉set更适合一点。
之前在输出被删除的链表时，第三个 %d 少了 05，结果硬是没有看出来 T-T

实现代码（C++）

#include <iostream>
#include <iomanip>
#include <map>

using namespace std;

int main() {
	map<int,pair<int,int>> list, del; // curr, <value, next>
	int head, n;
	scanf("%d %d", &head, &n);
	// head node
	list[head] = make_pair(-1, -1);
	
	for (int i = 0; i < n; i++) {
		int curr, val, next;
		scanf("%d %d %d", &curr, &val, &next);
		list[curr] = make_pair(val, next);
	}
	
	map<int,int> rVal; // map for reserve val
	int temp = head, pre, delPre = -1;
	int delHead = -1;
	while (temp != -1) {
		rVal[abs(list[temp].first)]++;
		if (rVal[abs(list[temp].first)] <= 1) {
			pre = temp;
		} else {
			list[pre].second = list[temp].second;
			
			if (delHead == -1) {
				delHead = temp;
			} else {
				del[delPre].second = temp;
			}
			
			del[temp] = make_pair(list[temp].first, -1);
			delPre = temp;
		}
		temp = list[temp].second;
	}
	
	temp = head;
	while (temp != -1) {
		int next = list[temp].second;
		if (next != -1) {
			printf("%05d %d %05d\n", temp, list[temp].first, list[temp].second);
		} else {
			printf("%05d %d -1\n", temp, list[temp].first);
		}
		
		temp = list[temp].second;
	}
	
	temp = delHead;
	while (temp != -1) {
		int next = del[temp].second;
		if (next != -1) {
			printf("%05d %d %05d\n", temp, del[temp].first, del[temp].second); // 就是在这里
		} else {
			printf("%05d %d -1\n", temp, del[temp].first);
		}
		
		temp = del[temp].second;
	}
	
	return 0;
}