PTAL2-002 链表去重

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

输入格式：

输入在第一行给出 L 的第一个结点的地址和一个正整数 N（≤105，为结点总数）。一个结点的地址是非负的 5 位整数，空地址 NULL 用 −1 来表示。

随后 N 行，每行按以下格式描述一个结点：

地址 键值 下一个结点

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

输出格式：

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

输入样例：

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;
const int mod = 1e5 + 10;
int main() {
  map<string, pair<int, string>> mp, mp1;  // 模拟链表
  string head;                             // 记录头节点
  int n;
  cin >> head >> n;
  int a[mod] = {0};  // 记录键值出现次数
  for (int i = 0; i < n; i++) {
    string p, next;
    int v;
    cin >> p >> v >> next;
    mp[p] = {v, next};
  }
  string p = head, bp, bp1, head1 = "-1";  // bp，dp1是上一个节点，p是当前节点
  while (p != "-1") {
    if (a[abs(mp[p].first)]) {
      if (mp1.empty()) {
        mp[bp].second = mp[p].second;
        mp1[p].first = mp[p].first;
        mp1[p].second = "-1";
        bp1 = p;
        head1 = p;  // 记录删除链表的头节点
      } else {
        mp[bp].second = mp[p].second;  // 更新原链表
        mp1[p].first = mp[p].first;    // 加入删除链表
        mp1[p].second = "-1";          // 令新节点为空节点
        mp1[bp1].second = p;           // 更新删除链表
        bp1 = p;
      }
      mp.erase(p);        // 删除
      p = mp[bp].second;  // 更新节点
    } else {
      a[abs(mp[p].first)] = 1;  // 记录键值
      bp = p;
      p = mp[p].second;
    }
  }
  p = head;
  while (p != "-1") {  // 遍历链表并输出
    cout << p << " " << mp[p].first << " " << mp[p].second << endl;
    p = mp[p].second;
  }
  p = head1;
  while (p != "-1") {
    cout << p << " " << mp1[p].first << " " << mp1[p].second << endl;
    p = mp1[p].second;
  }
  return 0;
}