[PTA团体程序设计天梯赛-练习集] L2-002 链表去重

题目详情 (pintia.cn)https://pintia.cn/problem-sets/994805046380707840/problems/994805072641245184

给定一个带整数键值的链表 L，你需要把其中绝对值重复的键值结点删掉。即对每个键值 K，只有第一个绝对值等于 K 的结点被保留。同时，所有被删除的结点须被保存在另一个链表上。例如给定 L 为 21→-15→-15→-7→15，你需要输出去重后的链表 21→-15→-7，还有被删除的链表 -15→15。

输入格式：

输入在第一行给出 L 的第一个结点的地址和一个正整数 N（≤105，为结点总数）。一个结点的地址是非负的 5 位整数，空地址 NULL 用 −1 来表示。

随后 N 行，每行按以下格式描述一个结点：

地址 键值 下一个结点

其中地址是该结点的地址，键值是绝对值不超过104的整数，下一个结点是下个结点的地址。

输出格式：

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

输入样例：

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<iostream>
#include<cstdio> 
#include<algorithm>
using namespace std;

const int N = 1e5;
struct node{
	int value;
	int next;
};

node alls[N];
int result[N],del[N];
int ret[N];//标记是否存在
int addressList[N]; 
int main()
{
	int head,n;
	cin >> head >> n;
	
	int site,value,next;
	for(int i = 0; i < n; i++)
	{
		cin >> site >> value >> next;
		alls[site].value = value;
		alls[site].next = next;
	}
	
	//把链表理顺 
	int cnt = 0, ph = head;
	while(ph != -1){
		addressList[cnt++] = ph;
		ph = alls[ph].next;
	}
	
	int n1 = 0, n2 = 0;
	for(int i = 0; i < cnt; i++)
	{
		int t = alls[addressList[i]].value;
		if(ret[abs(t)]){
			del[n2++] = addressList[i];
		}else{
			result[n1++] = addressList[i];
			ret[abs(t)] = 1;
		}
	}
	
    for (int i = 0; i < n1; i++) 
	{
	    printf ("%05d %d", result[i], alls[result[i]].value);
	    if (i != n1-1) {
	        printf (" %05d\n",result[i + 1]);
	    }else {
	        printf (" -1\n");
	    }
    }
    for (int i = 0; i < n2; i++) {
        printf ("%05d %d", del[i], alls[del[i]].value);
	    if (i != n2-1) {
	        printf (" %05d\n",del[i + 1]);
	    }else {
	        printf (" -1\n");
	    }
    }
	return 0;
}