Skip to content

1075 链表元素分类

Statement

Metadata

  • 作者: CHEN, Yue
  • 单位: 浙江大学
  • 代码长度限制: 16 KB
  • 时间限制: 400 ms
  • 内存限制: 64 MB

给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而 [0, K] 区间内的元素都排在大于 K 的元素前面。但每一类内部元素的顺序是不能改变的。例如：给定链表为 18→7→-4→0→5→-6→10→11→-2，K 为 10，则输出应该为 -4→-6→-2→7→0→5→10→18→11。

输入格式

每个输入包含一个测试用例。每个测试用例第 1 行给出：第 1 个结点的地址；结点总个数，即正整数N (\le 10^5)；以及正整数K (\le 10^3)。结点的地址是 5 位非负整数，NULL 地址用 -1 表示。

接下来有 N 行，每行格式为： 

Address Data Next
其中 Address 是结点地址；Data 是该结点保存的数据，为 [-10^5, 10^5] 区间内的整数；Next 是下一结点的地址。题目保证给出的链表不为空。

输出格式

对每个测试用例，按链表从头到尾的顺序输出重排后的结果链表，其上每个结点占一行，格式与输入相同。

输入样例 

00100 9 10
23333 10 27777
00000 0 99999
00100 18 12309
68237 -6 23333
33218 -4 00000
48652 -2 -1
99999 5 68237
27777 11 48652
12309 7 33218

输出样例 

33218 -4 68237
68237 -6 48652
48652 -2 12309
12309 7 00000
00000 0 99999
99999 5 23333
23333 10 00100
00100 18 27777
27777 11 -1

Solution

#include <bits/stdc++.h>
using namespace std;
#define maxn 100000
struct node {
    int add, data, next, level;
} temp, total[maxn];
typedef vector<node> no;
void print(node x) {
    printf("%05d %d %05d\n", x.add, x.data, x.next);
}
void print1(node x) {
    printf("%05d %d -1\n", x.add, x.data);
}
int main() {
    int first, N, K, len;
    cin >> first >> N >> K;
    int i, j, k;
    map<int, node> vin;
    no zero, one, two;
    for (i = 0; i < N; i++) {
        scanf("%d%d%d", &temp.add, &temp.data, &temp.next);
        if (temp.data < 0)
            temp.level = 0;
        else if (temp.data <= K)
            temp.level = 1;
        else
            temp.level = 2;
        vin[temp.add] = temp;
    }
    int addnext = first;
    if (first == -1)
        cout << "-1\n";
    else {
        for (; addnext != -1; addnext = vin[addnext].next) {
            if (vin[addnext].level == 0)
                zero.push_back(vin[addnext]);
            else if (vin[addnext].level == 1)
                one.push_back(vin[addnext]);
            else
                two.push_back(vin[addnext]);
        }
        for (len = zero.size(), i = 0, j = 0; i < len; i++) total[j++] = zero[i];
        for (len = one.size(), i = 0; i < len; i++) total[j++] = one[i];
        for (len = two.size(), i = 0; i < len; i++) total[j++] = two[i];
        for (i = 0; i < j - 1; i++) {
            total[i].next = total[i + 1].add;
            print(total[i]);
        }
        print1(total[i]);
    }
}
Last update: May 4, 2022
Back to top