pta L2-002 链表去重

1.题目

L2-002 链表去重 - 团体程序设计天梯赛-练习集 (pintia.cn)

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

2.题解代码

#include<bits/stdc++.h>
using namespace std;
struct Node{
	int address;
	int data;
	int next;
};
int main(){
	int jian[10001];
	memset(jian,0,sizeof(jian));
	Node a[100000];
	int first,n;
	cin>>first>>n;
	for(int i=0;i<n;i++){
		int address,data,next;
		cin>>address>>data>>next;
		a[address].address=address;
		a[address].data=data;
		a[address].next=next;
	}
	Node b[n],c[n];
	int i=first,t1=0,t2=0;
	while(a[i].next!=-1){
		if(!jian[abs(a[i].data)]){
			jian[abs(a[i].data)]=1;
			b[t1].address=a[i].address;
			b[t1].data=a[i].data;
			b[t1].next=a[i].next;
			t1++;
		}else {
			c[t2].address=a[i].address;
			c[t2].data=a[i].data;
			c[t2].next=a[i].next;
			t2++;
		}
		i=a[i].next;
	}
	if(!jian[abs(a[i].data)]){
		jian[abs(a[i].data)]=1;
		b[t1].address=a[i].address;
		b[t1].data=a[i].data;
		b[t1].next=a[i].next;
		t1++;
	}else {
		c[t2].address=a[i].address;
		c[t2].data=a[i].data;
		c[t2].next=a[i].next;
		t2++;
	}
		printf("%05d %d",b[0].address,b[0].data);
	for(int i = 1; i < t1; i++){
		printf(" %05d\n", b[i].address); 
		printf("%05d %d", b[i].address, b[i].data);
	}
	cout << " -1" << endl;
	if(t2){
		printf("%05d %d", c[0].address, c[0].data);
		for(int i = 1; i < t2; i++){
			printf(" %05d\n", c[i].address);
			printf("%05d %d", c[i].address, c[i].data);
		}
		cout << " -1" << endl;
	}
	return 0;
}