L2-002. 链表去重

from:http://www.voidcn.com/article/p-ttkypqyj-bhh.html

给一个链表，去重（去掉值或者绝对值相等的），先输出删除后的链表，再输出删除了的链表。

分析：用结构体数组存储这个链表，大小为maxn = 100000，node[i]表示地址为i的结点。在结构体中定义一个num变量，将num变量先初始化为2 * maxn。通过改变num变量的值最后sort排序来改变链表的顺序。

将没有删除的结点的num标记为cnt1，cnt1为当前没有删除的结点的个数；将需要删除的结点的num标记为maxn + cnt2，cnt2表示当前删除了的结点的个数，因为一开始初始化为了2 * maxn，所以我们可以通过对num排序达到：num = 0~maxn为不删除结点，num = maxn~2maxn为删除结点，num = 2maxn为无效结点

这样sort后就会按照需要输出的顺序将结点排序，我们只需要输出前cnt1+cnt2个结点即可~~~~

#include <bits/stdc++.h>
using namespace std;

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

struct Node{
int add;
int key;
int next;
int num;
bool operator <(Node obj) const{
return num<obj.num;
}
}node[maxn];

bool exist[maxn];
int begin,n,cnt1,cnt2;

int main(){

cin >> begin >> n;
for (int i=0;i<maxn;i++){
node[i].num=2*maxn;
}

for (int i=0;i<n;i++){
int a;
cin >> a;
cin >> node[a].key >> node[a].next;
node[a].add=a;
}

for (int i=begin;i!=-1;i=node[i].next){
if (!exist[abs(node[i].key)]){
exist[abs(node[i].key)]=1;
node[i].num=cnt1;
cnt1++;
}
else{
node[i].num=maxn+cnt2;
cnt2++;
}
}

sort(node,node+maxn);

int cnt=cnt1+cnt2;

for (int i=0;i<cnt;i++){
if (i!=cnt1-1&&i!=cnt-1){
printf("%05d %d %05d\n",node[i].add,node[i].key,node[i+1].add);
}
else{
printf("%05d %d -1\n",node[i].add,node[i].key);
}
}
}