AcWing 4273. 链表合并

摆烂了

#include <iostream>

using namespace std;

const int N=100010;
int num[N],ne[N];
int a_begin,b_begin,n;

//将a转置
void reverse_a() {
    int p=ne[a_begin];//存一下后继的位置
    ne[a_begin]=-1;//首端变尾端，置-1
    while(p!=-1) {
        int p2=ne[p];//存后继
        ne[p]=a_begin;//当前指向前驱
        a_begin=p;//首端后移
        p=p2;//后继变当前
    }
}

//判断a和b哪个更长
bool check(int a,int b) {
    int a_num=0,b_num=0;
    while(a!=-1) {
        a_num++;
        a=ne[a];
    }
    while(b!=-1) {
        b_num++;
        b=ne[b];
    }
    return b_num<a_num;
}
int main() {
    cin >> a_begin >> b_begin >> n;
    for(int i=0; i<n; i++) {
        int address;
        cin >> address;
        cin >> num[address] >> ne[address];
    }

    //让a作为较短的那个，再将a转置
    if(check(a_begin,b_begin)) swap(a_begin,b_begin);
    reverse_a();

    //从b开始输出
    while(a_begin!=-1) {
        //依次输出b的位置、元素、和下一个位置地址
        printf("%05d %d %05d\n",b_begin,num[b_begin],ne[b_begin]);
        b_begin=ne[b_begin];//更新地址
        //输出后一个位置地址、元素和下一个a的地址
        printf("%05d %d %05d\n",b_begin,num[b_begin],a_begin);
        //a的地址，元素，和下一个b的地址
        if(ne[b_begin]!=-1) printf("%05d %d %05d\n",a_begin,num[a_begin],ne[b_begin]);
        else printf("%05d %d %d\n",a_begin,num[a_begin],ne[b_begin]);
        b_begin=ne[b_begin];
        a_begin=ne[a_begin];

    }
    if(b_begin!=-1) {
        while(ne[b_begin]!=-1) {
            printf("%05d %d %05d\n",b_begin,num[b_begin],ne[b_begin]);
            b_begin=ne[b_begin];
        }
        printf("%05d %d %d\n",b_begin,num[b_begin],ne[b_begin]);
    }
    return 0;
}