先来一段错误代码
#include <bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
typedef pair<int, int> PII;
const int N = 100010;
struct Node
{
int num;
int next;
} node[N];
int num1[N], num2[N]; // 存储去重之后的数组的地址和被去重的数组地址
int idx1, idx2; // 两个数组的游动指针
set<int> s; // 用于判断每个数的绝对值是否出现过
int first, n; // first存储第一个结点地址
int main()
{
cin >> first >> n;
for (int i = 0; i < n; i++)
{
int adr, num, next;
cin >> adr >> num >> next;
node[adr] = {num, next};
}
for (int i = first; i != -1; i = node[i].next)
{
int x = node[i].num;
if (!s.count(abs(x)))
{
num1[idx1++] = i;
s.insert(abs(x));
}
else
{
num2[idx2++] = i;
}
}
for (int i = 0; i < idx1 - 1; i++)
{
printf("%05d %d %05d\n", num1[i], node[num1[i]].num, num1[i + 1]);
}
printf("%05d %d -1\n", num1[idx1 - 1], node[num1[idx1 - 1]].num);
for (int i = 0; i < idx2 - 1; i++)
{
printf("%05d %d %05d\n", num2[i], node[num2[i]].num, num2[i + 1]);
}
printf("%05d %d -1\n", num2[idx2 - 1], node[num2[idx2 - 1]].num);
return 0;
}
然后是ac代码
切记一定要将末尾元素特判放入for循环遍历内部
#include <bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
typedef pair<int, int> PII;
const int N = 100010;
struct Node
{
int num;
int next;
} node[N];//不需要在结构体里创建地址变量,下标即可表示地址
int num1[N], num2[N]; // 存储去重之后的数组的地址和被去重的数组地址
int idx1, idx2; // 两个数组的游动指针
set<int> s; // 用于判断每个数的绝对值是否出现过
int first, n; // first存储第一个结点地址
int main()
{
cin >> first >> n;
for (int i = 0; i < n; i++)
{
int adr, num, next;
cin >> adr >> num >> next;
node[adr] = {num, next};
}
for (int i = first; i != -1; i = node[i].next)//链表遍历
{
int x = node[i].num;
if (!s.count(abs(x)))//如果此数值的绝对值未出现过
{
num1[idx1++] = i;//将此数值的地址存入num1
s.insert(abs(x));//将此数值的绝对值插入set
}
else
{
num2[idx2++] = i;//如果出现过则将地址存入num2
}
}
for (int i = 0; i < idx1; i++)//末尾元素尽量放在for循环内特判
//例如此题如果其中一个链表结点个数为零则会输出00000 0 -1 不符合题意
{
if (i == idx1 - 1)
printf("%05d %d -1\n", num1[i], node[num1[i]].num);
else
printf("%05d %d %05d\n", num1[i], node[num1[i]].num, num1[i + 1]);
}
// printf("%05d %d -1\n", num1[idx1 - 1], node[num1[idx1 - 1]].num);
for (int i = 0; i < idx2; i++)
{
if (i == idx2 - 1)
printf("%05d %d -1\n", num2[i], node[num2[i]].num);
else
printf("%05d %d %05d\n", num2[i], node[num2[i]].num, num2[i + 1]);
}
// printf("%05d %d -1\n", num2[idx2 - 1], node[num2[idx2 - 1]].num);
return 0;
}