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

L2-002 链表去重 （25 分)

 

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

输入格式：

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

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

地址 键值 下一个结点

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

输出格式：

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

输入样例：

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-1
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;

struct node{
    int val,nex;
}a[maxn];

int vis[maxn];
int ans[maxn],ret[maxn];

int main()
{
    int st,n;
    cin>>st>>n;
    for(int i=1;i<=n;i++){
        int head;
        cin>>head;
        cin>>a[head].val>>a[head].nex;
    }
    if(n==1) printf("%05d %d %d
",st,a[st].val,a[st].nex);
    else{
        int i=st,l1=0,l2=0;
        while(i!=-1){
            if(!vis[abs(a[i].val)]){
                vis[abs(a[i].val)]=1;
                ans[++l1]=i;
            }
            else{
                ret[++l2]=i;
            }
            i=a[i].nex;
        }
        for(int i=1;i<l1;i++){
            printf("%05d %d %05d
",ans[i],a[ans[i]].val,ans[i+1]);
        }
        printf("%05d %d -1
",ans[l1],a[ans[l1]].val);
        for(int i=1;i<l2;i++){
            printf("%05d %d %05d
",ret[i],a[ret[i]].val,ret[i+1]);
        }
        printf("%05d %d -1
",ret[l2],a[ret[l2]].val);
    }
}