蓝桥杯 自行车停放 双向链表

问题描述

　　有n辆自行车依次来到停车棚，除了第一辆自行车外，每辆自行车都会恰好停放在已经在停车棚里的某辆自行车的左边或右边。(e.g.停车棚里已经有3辆自行车，从左到右编号为：3,5,1。现在编号为2的第4辆自行车要停在5号自行车的左边，所以现在停车棚里的自行车编号是：3,2,5,1)。给定n辆自行车的停放情况，按顺序输出最后停车棚里的自行车编号。

输入格式

　　第一行一个整数n。
　　第二行一个整数x。表示第一辆自行车的编号。
　　以下n-1行，每行3个整数x,y,z。
　　z=0时，表示编号为x的自行车恰停放在编号为y的自行车的左边
　　z=1时，表示编号为x的自行车恰停放在编号为y的自行车的右边

输出格式

　　从左到右输出停车棚里的自行车编号

样例输入

4
3
1 3 1
2 1 0
5 2 1

样例输出

3 2 5 1

数据规模和约定

　　n<=100000
　　自行车编号为不超过100000的正整数。

解题思路：用数组模拟双向链表的模板题目，基础不扎实，调试了1小时。

在模板的基础上增加一个数组e_idx[]，若e[idx] = x，则e_idx[x] = idx，互为反函数（代码的12,13行）。便于通过值找到其下标（也就是要插入的位置）

#include <bits/stdc++.h>
using namespace std;
const int N = 200010;
int e[N], l[N], r[N], idx;
int e_idx[N]; //e_idx(x) = idx;即表示值为x的节点的下标是idx 
void init() {
    r[0] = 1;
    l[1] = 0;
    idx = 2;
}
void add(int k, int x) {
    e[idx] = x;
    e_idx[x] = idx;
    r[idx] = r[k];
    l[idx] = k;
    l[r[k]] = idx;
    r[k] = idx;
    //cout << "e:" << x << " " << "idx:" << idx << endl;
    idx++;
}
int main() {
    int n;
    cin >> n;
    init();
    int a;
    cin >> a;
    add(0, a);
    n--;
    /*cout << "结果为：";
    for (int i = r[0]; i != 1; i = r[i]) {
        cout << e[i] << " ";
    }
    cout << endl;*/
    while (n--) {
        int x, y, z;
        cin >> x >> y >> z;
        if (z == 0) { //左边插入 
            add(l[e_idx[y]], x);
        } else { //右边插入 
            add(e_idx[y], x);
        }
        /*cout << "结果为：";
        for (int i = r[0]; i != 1; i = r[i]) {
            cout << e[i] << " ";
        }
        cout << endl;*/
    }
    for (int i = r[0]; i != 1; i = r[i]) {
        cout << e[i] << " ";
    }
    return 0;
}