PTA L2-002 链表去重

PTA L2-002 链表去重

---

题目

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

输入

输入在第一行给出 L 的第一个结点的地址和一个正整数 N（≤10⁵，为结点总数）。一个结点的地址是非负的 5 位整数，空地址 NULL 用 −1 来表示。
随后 N 行，每行按以下格式描述一个结点：

地址 键值 下一个结点

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

输出

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

---

最终代码

#include<iostream>
#include<cmath>
#include<iomanip>
#include<map>
using namespace std;
struct Node{
    int key;
    int nextnode;
};
int main(){
    int head,N,adress,key,next,dnum=0,newnum=0,dnode[100007],newnode[100007];
    Node node[100007];
    map<int,int> KEY;
    cin >> head >> N;
    for(int i=0;i<N;i++){
        cin >> adress >> key >> next;
        node[adress].key=key;
        node[adress].nextnode=next;
    }
    for(int j=head;j>=0;j=node[j].nextnode){
        if(KEY.find(node[j].key)!=KEY.end()||KEY.find(-node[j].key)!=KEY.end()){
            dnode[dnum]=j;
            dnum++;
        }
        else{
            newnode[newnum]=j;
            newnum++;
            KEY[node[j].key]=0;
        }
    }
    for(int i=0;i<newnum;i++){
        cout << setw(5) << setfill('0') << right << newnode[i] << ' ' << node[newnode[i]].key << ' ';
        if(i!=newnum-1)
            cout << setw(5) << setfill('0') << right << newnode[i+1] << endl;
        else
            cout << -1 << endl;
    }
    for(int i=0;i<dnum;i++){
        cout << setw(5) << setfill('0') << right << dnode[i] << ' ' << node[dnode[i]].key << ' ';
        if(i!=dnum-1)
            cout << setw(5) << setfill('0') << right << dnode[i+1] << endl;
        else
            cout << -1 << endl;
    }
    return 0;
}

---

遇过的问题

• 段错误

  数组越界，调用了node[-1]

• 运行超时

  刚开始用的是循环去重，复杂度太大了

  for(int j=head;j>=0;j=node[j].nextnode){
      for(int k=j;k>=0;k=node[k].nextnode){
          while(abs(node[node[k].nextnode].key)==abs(node[j].key)&&k>=0){
              dnode[dnum]=node[k].nextnode;
              dnum++;
              node[k].nextnode=node[node[k].nextnode].nextnode;
          }
      }
  }
  

---

总结

去重用map