PAT天梯赛 L2-002. 链表去重 【STL】

题目链接

https://www.patest.cn/contests/gplt/L2-002

思路

用结构体 存储 一个结点的地址 值 和下一个地址
然后从首地址开始 往下走 并且每个值的绝对值 都标记一下
并且 每次往下走的时候 都判断一下 其值的绝对值 是否 已经被标记
如果被标记过 那么 它就要加入到 重复的序列当中
如果 没有被标记过 就要标记 然后加入到 未重复的序列

然后 最后记得 将最后一个结点的下一个地址 指为-1

AC代码

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define CLR(a) memset(a, 0, sizeof(a))
#define pb push_back

using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair <int, int> pii;
typedef pair  pll;
typedef pair<string, int> psi;
typedef pair<string, string> pss;

const double PI = 3.14159265358979323846264338327;
const double E = exp(1);
const double eps = 1e-6;

const int INF = 0x3f3f3f3f;
const int maxn = 1e2 + 5;
const int MOD = 1e9 + 7;

struct Node
{
    int add;
    int value;
    int next;
}temp;

map <int, Node> m;
map <int, int> vis;

vector  v[2];

int ini, n;

void dfs(int add)
{
    if (vis[abs(m[add].value)] == 0)
    {
        v[0][v[0].size() - 1].next = m[add].add;
        v[0].pb(m[add]);
        vis[abs(m[add].value)] = 1;
    }
    else
    {
        if (v[1].size())
            v[1][v[1].size() - 1].next = m[add].add;
        v[1].pb(m[add]);
    }
    if (m[add].next != -1)
        dfs(m[add].next);
}

int main()
{

    scanf("%d%d", &ini, &n);
    for (int i = 0; i < n; i++)
    {
        scanf("%d%d%d", &temp.add, &temp.value, &temp.next);
        m[temp.add] = temp;
    }
    v[0].pb(m[ini]);
    vis[abs(m[ini].value)] = 1;
    ini = m[ini].next;
    if (ini != -1)
        dfs(ini);
    if (v[0].size())
    {
        v[0][v[0].size() - 1].next = -1;
        vector ::iterator it;
        for (it = v[0].begin(); it != v[0].end(); it++)
        {
            printf("%05d %d ", (*it).add, (*it).value);
            if ((*it).next != -1)
                printf("%05d\n", (*it).next);
            else
                cout << -1 << endl;
        }
    }
    if (v[1].size())
    {
        v[1][v[1].size() - 1].next = -1;
        vector ::iterator it;
        for (it = v[1].begin(); it != v[1].end(); it++)
        {
            printf("%05d %d ", (*it).add, (*it).value);
            if ((*it).next != -1)
                printf("%05d\n", (*it).next);
            else
                cout << -1 << endl;
        }
    }
}