团体程序设计天梯赛-练习集 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

维护一下删除时候的操作就行了。

#include <cstdio>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <map>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N=1e5+10;
struct node
{
    int data;
    int next;
} a[N],b[N];
int main()
{
    int head,n;
    while(cin>>head>>n)
    {
        int i;
        for(i=0; i<n; i++)
        {
            int address;
            cin>>address;
            int key,Next;
            cin>>key>>Next;
            a[address].data=key;
            a[address].next=Next;
        }
        map<int,int>m;
        m.clear();
        int pre,ha,hb;
        pre=ha=hb=-1;
        i=head;
        for(; ~i;)
        {
            if(!m[abs(a[i].data)])
            {
                m[abs(a[i].data)]=1;
                pre=i;
                i=a[i].next;
            }
            else
            {
                a[pre].next=a[i].next;
                if(ha==-1)
                {
                    ha=hb=i;
                    b[ha].data=a[i].data;
                    b[ha].next=-1;
                }
                else
                {
                    b[hb].next=i;
                    hb=i;
                    b[hb].next=-1;
                    b[hb].data=a[i].data;
                }
                i=a[i].next;
            }
        }
        for(i=head; ~i; i=a[i].next)
        {
            if(a[i].next==-1)
                printf("%05d %d %d\n",i,a[i].data,a[i].next);
            else
                printf("%05d %d %05d\n",i,a[i].data,a[i].next);
        }
        for(i=ha; ~i; i=b[i].next)
        {
            if(b[i].next==-1)
                printf("%05d %d %d\n",i,b[i].data,b[i].next);
            else
                printf("%05d %d %05d\n",i,b[i].data,b[i].next);
        }
    }
    return 0;
}

---