团体程序设计天梯赛-练习集 L2-2. 链表去重 数组模拟链表

L2-2. 链表去重

时间限制

300 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

给定一个带整数键值的单链表L，本题要求你编写程序，删除那些键值的绝对值有重复的结点。即对任意键值K，只有键值或其绝对值等于K的第一个结点可以被保留。同时，所有被删除的结点必须被保存在另外一个链表中。例如：另L为21→-15→-15→-7→15，则你必须输出去重后的链表21→-15→-7、以及被删除的链表-15→15。

输入格式：

输入第一行包含链表第一个结点的地址、以及结点个数N（<= 10⁵ 的正整数）。结点地址是一个非负的5位整数，NULL指针用-1表示。

随后N行，每行按下列格式给出一个结点的信息：

Address Key Next

其中Address是结点的地址，Key是绝对值不超过10⁴的整数，Next是下一个结点的地址。

输出格式：

首先输出去重后的链表，然后输出被删除结点组成的链表。每个结点占一行，按输入的格式输出。

输入样例：

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 <cstdio>
#include <cstring>
#include <iostream>
#include <map>

using namespace std;

//p为原始链表，bin为删除的节点放的链表
struct node {
	int n, Nex;
}p[1000000 + 10], bin[1000000 + 10];

int abs(int x) {
	return x < 0 ? -x : x;
}

int main()
{
	int Address, Key, Next, H, N;
	scanf("%d%d", &H, &N);
	for (int i = 0; i < N; i++) {
		scanf("%d%d%d", &Address, &Key, &Next);
		p[Address].n = Key;
		p[Address].Nex = Next;
	}
	map<int, int> m;
	//第一个肯定不会被删，可以大胆用一个当头结点
	int len = 0, hb = -1, rearb = -1;
	for (int pre = -1, i = H; i != -1;) {
		if (m[abs(p[i].n)] == 0) {
			m[abs(p[i].n)]++;
			pre = i;
			i = p[i].Nex;
		}
		else {
			p[pre].Nex = p[i].Nex;
			if (hb == -1) {
				hb = rearb = i;
				bin[hb] = p[i];
				bin[hb].Nex = -1;
			}
			else {
				bin[rearb].Nex = i;
				rearb = i;
				bin[rearb].n = p[i].n;
				bin[rearb].Nex = -1;
			}
			i = p[pre].Nex;
		}
	}
	for (int i = H; i != -1; i = p[i].Nex) {
		printf(p[i].Nex == -1 ? "%05d %d %d\n" : "%05d %d %05d\n", i, p[i].n, p[i].Nex);
	}
	for (int i = hb; i != -1; i = bin[i].Nex) {
		printf(bin[i].Nex == -1 ? "%05d %d %d\n" : "%05d %d %05d\n", i, bin[i].n, bin[i].Nex);
	}
	return 0;
}