PTA L2-002链表去重

L2-002 链表去重

分数 25

全屏浏览

切换布局

作者 陈越

单位 浙江大学

给定一个带整数键值的链表 L，你需要把其中绝对值重复的键值结点删掉。即对每个键值 K，只有第一个绝对值等于 K 的结点被保留。同时，所有被删除的结点须被保存在另一个链表上。例如给定 L 为 21→-15→-15→-7→15，你需要输出去重后的链表 21→-15→-7，还有被删除的链表 -15→15。

输入格式：

输入在第一行给出 L 的第一个结点的地址和一个正整数 N（≤105，为结点总数）。一个结点的地址是非负的 5 位整数，空地址 NULL 用 −1 来表示。

随后 N 行，每行按以下格式描述一个结点：

地址 键值 下一个结点

其中地址是该结点的地址，键值是绝对值不超过104的整数，下一个结点是下个结点的地址。

输出格式：

首先输出去重后的链表，然后输出被删除的链表。每个结点占一行，按输入的格式输出。

输入样例：

00100 5
99999 -7 87654
23854 -15 00000
87654 15 -1
00000 -15 99999
00100 21 23854

输出样例：

00100 21 23854
23854 -15 99999
99999 -7 -1
00000 -15 87654
87654 15 -1

思路比较简单，用数组分别存地址，值，下一个符号地址即可，竞赛中这样相较于写一个正式的链表并操作会节省很多时间 

 上代码

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int addr,data;
    int next;
};
const int N = 1e5+5;
int main()
{
    int start,n;cin>>start>>n;
    node nc[N];
    for(int i=0;i<n;i++)
    {
        int add;cin>>add;
        cin>>nc[add].data>>nc[add].next;
        nc[add].addr=add;
    }
    int cnt[N]={0};
    vector<node> ans1,ans2;
    for(int i=start;i!=-1;i=nc[i].next)
    {
        cnt[abs((nc[i].data))]++;
        if(cnt[abs((nc[i].data))]!=1)
        {
            ans2.push_back(nc[i]);
        }
        else 
        {
            ans1.push_back(nc[i]);
        }
    }
    int m1=ans1.size(),m2=ans2.size();
    for(int i=0;i<m1;i++)
    {
        printf("%05d %d ",ans1[i].addr,ans1[i].data);
        if(i==m1-1) 
        {
            cout<<"-1";
        }
        else 
        {
            printf("%05d",ans1[i+1].addr);
        }
        cout<<"\n";
    }
    for(int i=0;i<m2;i++)
    {
        printf("%05d %d ",ans2[i].addr,ans2[i].data);
        if(i==m2-1) 
        {
            cout<<"-1";
        }
        else 
        {
            printf("%05d",ans2[i+1].addr);
        }
         cout<<"\n";
    }
    return 0;
}