PTA 链表去重

链表去重

分数 5

全屏浏览题目

切换布局

作者 陈越

单位 浙江大学

给定一个带整数键值的链表 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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

c++代码：

#include <stdio.h>
#include <math.h>

// 定义结点结构体
typedef struct Node
{
    int date; // 结点的日期信息
    int next; // 结点的下一个结点索引
} Node;

// 打印链表
void print(Node arr[], int g[], int n)
{
    for (int i = 0; i < n; i++)
    {
        if (i == n - 1)
        {
            printf("%05d %d -1\n", g[i], arr[g[i]].date);
        }
        else
        {
            printf("%05d %d %05d\n", g[i], arr[g[i]].date, g[i + 1]);
        }
    }
}

int first[100005];
int f = 0; // 存储首次出现结点的索引数组的元素数量
int isVisited[100005];
int is = 0;
int last[100005];
int l = 0; // 存储重复出现结点的索引数组的元素数量

int main()
{
    Node arr[100005];
    int head;
    int n;
    int ads;
    
    // 输入链表头结点和结点数量
    scanf("%d %d", &head, &n);
  
    // 输入并初始化链表中每个结点的信息
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &ads); // 输入当前结点索引
        scanf("%d %d", &arr[ads].date, &arr[ads].next); // 输入当前结点的日期和下一个结点索引
    }
  
  // 遍历链表，将首次出现的结点索引存入 first 数组，将重复出现的结点索引存入 last 数组
  int p = head;
  while (p != -1)
  {
      is = abs(arr[p].date); // 获取日期的绝对值
      if (!isVisited[is]) // 如果该日期没有被访问过
      {
          first[f++] = p; // 添加到首次出现的结点索引数组中，更新 f 的值
          isVisited[is] = 1; // 将该日期标记为已访问
      }
      else // 如果日期已经被访问过，则为重复出现的结点
      {
          last[l++] = p; // 添加到重复出现的结点索引数组中，更新 l 的值
      }
      p = arr[p].next; // 移动到下一个结点
  }
    
    // 打印首次出现和重复出现的结点链表
    print(arr, first, f);
    print(arr, last, l);

    return 0;
}