PTA L2链表去重 (25分)

                                           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

思路:用链表进行实现

注意:第三个测试点一直段错误,后来查了下发现有种特殊情况,即全部都不重复,就不用再输出保存删除结点的结构体了。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6;
struct Node{
	int nxt;
	int num;
}node[maxn],bin[maxn];
int main() {
	int n,nn,m,pre,bpre,bhead,cnt=0;
	cin>>n>>m;
	nn=n;
	for(int i =1;i<=m;i++) {
		int u,v,w;
		cin>>u>>w>>v;
		node[u].nxt = v;
		node[u].num = w;
	}
	map<int,int> mp;
	for(int i=n;i!=-1;i=node[i].nxt) {
		if(mp[abs(node[i].num)]==0) {
			pre = i;
			mp[abs(node[i].num)]++;
		}
		else {
			nn--;
			bin[i].num = node[i].num;
			bin[i].nxt = -1;
			cnt ++;
			if(cnt==1)
				bhead=i;
			else{
				bin[bpre].nxt = i;
			}
			bpre = i;
			node[pre].nxt = node[i].nxt;
		}
	}
	if(nn==n) {
		for(int i=n;i!=-1;i=node[i].nxt) {
			printf(node[i].nxt==-1?"%05d %d %d\n":"%05d %d %05d\n",i,node[i].num,node[i].nxt);	
		}	
	}
	else {
		//链表 删除重复的元素的结点 
		for(int i=n;i!=-1;i=node[i].nxt) {
			printf(node[i].nxt==-1?"%05d %d %d\n":"%05d %d %05d\n",i,node[i].num,node[i].nxt);
		}
		//保存被删除的结点	
		for(int i=bhead;i!=-1;i=bin[i].nxt) {
			printf(bin[i].nxt==-1?"%05d %d %d\n":"%05d %d %05d\n",i,bin[i].num,bin[i].nxt);
		}	
	}
	
	return 0;
}

发布了87 篇原创文章 · 获赞 56 · 访问量 9133

猜你喜欢

转载自blog.csdn.net/Ven21959/article/details/104028004