首页 > > 程序设计 > C/C++ >

L2-002 链表去重

2019-03-06 07:14:51来源：博客园阅读 ()

题目：

给定一个带整数键值的链表

输入格式：

输入在第一行给出 L 的第一个结点的地址和一个正整数

随后

地址 键值 下一个结点

其中地址是该结点的地址，键值是绝对值不超过

输出格式：

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

输入样例：

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

思路：

定义一个结构体数组表示初始的链表，直接以下标作为地址。因为题目说一个结点的地址是非负的五位整数，所以定义数组长度为100000。还需要一个数组用于标记链表中每一个键值的绝对值是否存在过了，以数组下标表示键值的绝对值，根据题目，数组长度为10010够用了。再定义两个vector，用于存放删除后的链表的每个键值的地址和被删除的每个键值的地址。再根据这两个vector存储的地址去初始链表中找到，输出。

上代码：

#include <iostream>
#include <cmath>
#include <vector>
using namespace std;

struct node
{
    int value;
    int next;
}a[100000];

vector<int> v1;
vector<int> v2;
int main() {    
    int start,sum;
    cin>>start>>sum;
    int add,v,next,ov;
    int count[100010]={0};
    for(int i=0;i<sum;i++){
        cin>>add>>v>>next;
        a[add].value=v;
        a[add].next=next;    
    }
    
    for(int i=start;i!=-1;i=a[i].next)
    {
        ov=abs(a[i].value);
        if(count[ov]==0)
        {
            count[ov]=1;
            v1.push_back(i);
        }else if(count[ov]==1){
            v2.push_back(i);
        }
    }
    int len1 = v1.size();
    int len2 = v2.size();
 
    for(int i = 0; i<len1-1; ++i){
        printf("%05d %d %05d\n",v1[i],a[v1[i]].value,v1[i+1]);
    }
    ///必须当len >= 1时
    if(len1 >= 1)
        printf("%05d %d -1\n",v1[len1-1],a[v1[len1-1]].value);
 
 
    for(int i = 0; i<len2-1; ++i){
        printf("%05d %d %05d\n",v2[i],a[v2[i]].value,v2[i+1]);
    }
    if(len2 >= 1)
        printf("%05d %d -1\n",v2[len2-1],a[v2[len2-1]].value);

    return 0;
}