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（<= 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

题解：定义两个结构体，存储原来的链表和输出的链表。

          定义一个初始化为零的数组，来标记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");
    }
}