题意:将链表去重,输出去重后的新链表 和被去掉的节点组成的新链表
思路:
结构体存储输入的原始数据,vis数组记录是否被去重
初始化结构体中的num变量,前[0, 1e5)记录未被去重的节点,后[1e5, 1e5 * 2)记录被去重的节点,在此之前先全部初始化为1e5 * 2
遍历链表,去重,同时记录未被去重的节点个数和被去重的节点个数
按照num变量的大小排序
输出时注意:每个节点的下一节点地址 是 排序后下一节点地址
#define MAXN 500005
struct Node {
int t; //标记该节点的位次
int val;
int address;
int next;
} node[MAXN];
bool vis[MAXN];
bool cmp(Node a, Node b) { //按照编号排序
return a.t < b.t;
}
int main() {
int start, n;
scanf("%d %d", &start, &n);
for(int i = 0; i < MAXN; i ++) { //初始化 1e5*2
node[i].t = MAXN * 2;
}
for(int i = 0; i < n; i ++) { //输入,用数组号存储首地址
int aa;
scanf("%d", &aa);
node[aa].address = aa;
scanf("%d %d", &node[aa].val, &node[aa].next);
}
int k1 = 0, k2 = 0; //去重,k1,k2记录没被去重的节点个数 和去重的节点个数,将节点序号填入node[i].t
for(int i = start; i != -1; i = node[i].next) {
if(!vis[abs(node[i].val)]) {
vis[abs(node[i].val)] = 1;
node[i].t = k1 ++;
}
else {
node[i].t = MAXN + k2;
k2 ++;
}
}
sort(node, node + MAXN, cmp); //排序,前k1个是没被去重的节点,[k1, k2)是被去重的节点,空节点被排序去后面
int k = k1 + k2;
for(int i = 0; i < k; i ++) {
if(i != k1 - 1 && i != k - 1) {
printf("%05d %d %05d\n", node[i].address, node[i].val, node[i + 1].address); //注意next节点的地址可能被更新,为数组中下一节点的个数
}
else {
printf("%05d %d -1\n", node[i].address, node[i].val);
}
}
return 0;
}