PAT L2-002. 链表去重

题目链接<https://www.patest.cn/contests/gplt/L2-002>

L2-002. 链表去重

时间限制300 ms
内存限制65536 kB
代码长度限制8000 B
判题程序Standard作者陈越
给定一个带整数键值的单链表L，本题要求你编写程序，删除那些键值的绝对值有重复的结点。即对任意键值K，只有键值或其绝对值等于K的第一个结点可以被保留。同时，所有被删除的结点必须被保存在另外一个链表中。例如：另L为21→-15→-15→-7→15，则你必须输出去重后的链表21→-15→-7、以及被删除的链表-15→15。输入格式：输入第一行包含链表第一个结点的地址、以及结点个数N（<= 105 的正整数）。结点地址是一个非负的5位整数，NULL指针用-1表示。随后N行，每行按下列格式给出一个结点的信息：Address Key Next其中Address是结点的地址，Key是绝对值不超过104的整数，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

题解：定义两个结构体，存储原来的链表和输出的链表。
          定义一个初始化为零的数组，来标记key的出现次数，如果重复则将其连入输出的链表。#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <math.h>
#include <string.h>
#include <string>
#include <map>
using namespace std;
typedef long long LL;
struct Node{
int key,next;
}a[100005];
int b[100005];
Node c[100005];
int main(){
int s,n;
cin>>s>>n;
int ad;
for(int i=1;i<=n;i++){
cin>>ad;
cin>>a[ad].key>>a[ad].next;
}
int now=-1,flag=0,st;
c[now].next=-1;
for(int i=s;~i;){
printf("%05d %d ",i,a[i].key);
b[abs(a[i].key)]=1;
i=a[i].next;
while(i>=0&&b[abs(a[i].key)]){
if(!flag){
now=st=i;
c[now].key=a[i].key;
}
else{
c[now].next=i;
c[i].key=a[i].key;
c[i].next=-1;
now=i;
}
i=a[i].next;
flag=1;
}
if(i>=0)
printf("%05d\n",i);
else printf("-1\n");
}
if(flag)
for(int i=st;~i;i=c[i].next){
printf("%05d %d ",i,c[i].key);
if(c[i].next>=0)
printf("%05d\n",c[i].next);
else printf("-1\n");
}
}