L2-002. 链表去重

题目描述
给定一个带整数键值的单链表L，本题要求你编写程序，删除那些键值的绝对值有重复的结点。即对任意键值K，只有键值或其绝对值等于K的第一个结点可以被保留。同时，所有被删除的结点必须被保存在另外一个链表中。例如：另L为21→-15→-15→-7→15，则你必须输出去重后的链表21→-15→-7、以及被删除的链表-15→15。

输入

输入第一行包含链表第一个结点的地址、以及结点个数N（<= 105 的正整数）。结点地址是一个非负的5位整数，NULL指针用-1表示。

随后N行，每行按下列格式给出一个结点的信息：

Address Key Next

其中Address是结点的地址，Key是绝对值不超过104的整数，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 <bits/stdc++.h>
using namespace std;

struct Node{
    int s;
    int val;
    int next;
} node[100010];

bool book[100010] = {0};

vector<Node> v1, v2;

int main(){
    int fir;
    int n;
    cin>>fir>>n;
    int idx;
    for(int i=0; i<n; i++){
        cin>>idx;
        cin>>node[idx].val>>node[idx].next;
        node[idx].s = idx;
    }
    int nx = fir;
    while(nx != -1){
        if(book[abs(node[nx].val)] == 0){
            book[abs(node[nx].val)] = 1;
            v1.push_back(node[nx]);
        }else {
            v2.push_back(node[nx]);
        }
        nx = node[nx].next;
    }

    for(vector<Node>::iterator it=v1.begin(); it!=v1.end(); it++){
        if(it != v1.begin()){
            printf("%05d\n", (*it).s);
        }
        printf("%05d", (*it).s);
        cout<<" "<<(*it).val<<" ";
    }
    cout<<-1<<endl;

    if(v2.size()){ //一定要加，因为v2有可能为空 不加的话第三个数据过不去 
        for(vector<Node>::iterator it=v2.begin(); it!=v2.end(); it++){
            if(it != v2.begin()){
                printf("%05d\n", (*it).s);
            }
            printf("%05d", (*it).s);
            cout<<" "<<(*it).val<<" ";
        }
        cout<<-1<<endl;
    }


    return 0;
}