L2-022 重排链表(c++)

题目

给定一个单链表 L 1→L 2→⋯→L n−1→L n，请编写程序将链表重新排列为 Ln→L 1→L n−1→L 2→⋯。例如：给定L为1→2→3→4→5→6，则输出应该为6→1→5→2→4→3。
输入格式：
每个输入包含1个测试用例。每个测试用例第1行给出第1个结点的地址和结点总个数，即正整数N (≤10的5次方)。结点的地址是5位非负整数，NULL地址用−1表示。

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

Address Data Next
其中Address是结点地址；Data是该结点保存的数据，为不超过10
5
的正整数；Next是下一结点的地址。题目保证给出的链表上至少有两个结点。

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

输入

00100 6
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218

输出

68237 6 00100
00100 1 99999
99999 5 12309
12309 2 00000
00000 4 33218
33218 3 -1

思路

利用结构体存链表的结点相关信息，存完之后前后各一个指针依次遍历每一个值即可，唯一需要注意的是当结点数为奇数时需要特判。

代码

#include<iostream>
#include<vector>
#include<cstdio>

using namespace std;

const int N = 100010;

//利用结构体存链表
struct Node{
    int date, next;
}node[N];

int res[N];

int main(){
    int st, n;
    cin >> st >> n;
    
    for (int i = 0; i < n; i ++){
        int a, b, c;
        cin >> a >> b >> c;
        node[a].date = b;
        node[a].next = c;
    }
    
    int cnt = 0;
    //从头节点依次遍历找到每一个结点的地址
    for (int i = st; i != -1; i = node[i].next) res[cnt ++] = i; 
    
    vector<int> v; //用来存放最终的链表
    int l = 0, r = cnt - 1;
    while (l <= r){
        if (l == r) {
            v.push_back(res[l]);
            break;
        } //当结点数为奇数时需要特判一下，不然会使同一个结点存两次
        v.push_back(res[r]);
        r --;
        v.push_back(res[l]);
        l ++;
    }
    
    for (int i = 0; i < v.size(); i ++){
        if (i != v.size() - 1)
            printf("%05d %d %05d\n", v[i], node[v[i]].date, v[i + 1]);
        else
            printf("%05d %d -1\n", v[i], node[v[i]].date);
    }
    
    return 0;
}