L2-002. 链表去重

给定一个带整数键值的单链表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 <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

int a[100000][2];
bool vis [10000+5];   //  标记是否重复

int main()
{
    int s,N;
    memset(vis,0,sizeof(vis));
    cin>>s>>N;
    for(int i =0; i < N; i++)
    {
        int temp;
        cin>>temp;
        cin>>a[temp][0];
        cin>>a[temp][1];
    }
    int i = a[s][1], mark = -1, m = s, n = 0;   //mark用来标记重复那组链表的第一个地址


    if(a[s][0] < 0)
        vis[-a[s][0]] = 1;
    else
        vis[a[s][0]] = 1;            //标记第一个

    while(i != -1)              //不造为啥i赋为第一个地址，再在while循环外在判断一次就超时
    {
        int temp = a[i][0];
        if(temp < 0)
            temp = -temp;
        if(!vis[temp])
        {
            a[m][1] = i;
            m = i;
            vis[temp] = 1;
        }
        else if(vis[temp] && mark == -1)             //mark存二组首地址
        {
            n = i;
            mark = i;
        }
        else if(vis[temp] && mark != -1)
        {
            a[n][1] = i;
            n = i;
        }
        i = a[i][1];
    }
    a[m][1] = -1;
    a[n][1] = -1;
    i = s;
    while(a[i][1] != -1)
    {
        printf("%05d %d %05d\n",i,a[i][0],a[i][1]);
        i = a[i][1];
    }
    printf("%05d %d -1\n",i,a[i][0]);


    if(mark != -1)
    {
        i = mark;
        while(a[i][1] != -1)
        {
            printf("%05d %d %05d\n",i,a[i][0],a[i][1]);
            i = a[i][1];
        }
        printf("%05d %d -1\n",i,a[i][0]);
    }
}