L2-002 链表去重

先来一段错误代码

#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;
}