【PTA L2-002】链表去重

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Xylon_/article/details/87889761

                              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;
#define ll long long
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))

struct ac
{
    int num,next;
}a[100005];

int vis[10005];        //标记绝对值是否重复
int b[100005];        //记录键值重复的节点

int main()
{
    int head,n,i,x,y,z;
    mem(vis,0);
    cin>>head>>n;
    while(n--)
    {
        cin>>x>>y>>z;
        a[x].num=y;
        a[x].next=z;
    }
    printf("%05d %d ",head,a[head].num);
    int tx=abs(a[head].num);
    vis[tx]=1;
    int t=head;
    int j=0;
    while(1)
    {
        t=a[t].next;
        if(t==-1)
        { 
            printf("-1\n");
            break;
        }
        tx=abs(a[t].num);
        if(!vis[tx])
        {
            vis[tx]=1;
            printf("%05d\n%05d %d ",t,t,a[t].num);
        }
        else
        {
            b[j++]=t;
        }
    }
    if(j>0)
    {
        printf("%05d %d ",b[0],a[b[0]].num);
        for(i=1;i<j;i++)
            printf("%05d\n%05d %d ",b[i],b[i],a[b[i]].num);
        printf("-1\n");
    }
    return 0;
}