L2-002. 链表去重 - 天梯赛练习题

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

解题思路

用结构体模拟一下就好，node【i】，  i 表示地址。从根节点开始遍历，重复的放一个数组，不重复的放另一个数组，就是要注意数组下标越界的问题

代码

#include <iostream>
#include <cstdio>
#include <map>
#include <cmath>
#include <string.h>
#include <algorithm>
#include <set>
#include <sstream>
#include <vector>
using namespace std;

const int maxn = 100010;
const int INF = 0x3f3f3f3f;

struct node{
int next = -1;
int key;
}data[maxn];

vector<int> v1;
vector<int> v2;

bool exist[maxn];

int main()
{
int root,n;
int x;
scanf("%d%d",&root,&n);
for(int i = 0; i<n; ++i){
scanf("%d",&x);
scanf("%d%d",&data[x].key,&data[x].next);
}
for(int i = root; i != -1; i = data[i].next){
if(exist[abs(data[i].key)]){
v2.push_back(i);
}
else{
v1.push_back(i);
exist[abs(data[i].key)] = 1;
}
}

int len1 = v1.size();
int len2 = v2.size();

for(int i = 0; i<len1-1; ++i){
printf("%05d %d %05d\n",v1[i],data[v1[i]].key,v1[i+1]);
}
///必须当len >= 1,才能写这句
if(len1 >= 1)
printf("%05d %d -1\n",v1[len1-1],data[v1[len1-1]].key);

for(int i = 0; i<len2-1; ++i){
printf("%05d %d %05d\n",v2[i],data[v2[i]].key,v2[i+1]);
}
if(len2 >= 1)
printf("%05d %d -1\n",v2[len2-1],data[v2[len2-1]].key);

return 0;
}