1105 链表合并分数 25

给定两个单链表 L1​=a1​→a2​→⋯→an−1​→an​ 和 L2​=b1​→b2​→⋯→bm−1​→bm​。如果 n≥2m，你的任务是将比较短的那个链表逆序，然后将之并入比较长的那个链表，得到一个形如 a1​→a2​→bm​→a3​→a4​→bm−1​⋯ 的结果。例如给定两个链表分别为 6→7 和 1→2→3→4→5，你应该输出 1→2→7→3→4→6→5。

输入格式：

输入首先在第一行中给出两个链表 L1​ 和 L2​ 的头结点的地址，以及正整数
N (≤105)，即给定的结点总数。一个结点的地址是一个 5 位数的非负整数，空地址 NULL 用 -1 表示。

随后 N 行，每行按以下格式给出一个结点的信息：

Address Data Next

其中 Address 是结点的地址，Data 是不超过 105 的正整数，Next 是下一个结点的地址。题目保证没有空链表，并且较长的链表至少是较短链表的两倍长。

输出格式：

按顺序输出结果链表，每个结点占一行，格式与输入相同。

输入样例：

00100 01000 7
02233 2 34891
00100 6 00001
34891 3 10086
01000 1 02233
00033 5 -1
10086 4 00033
00001 7 -1

输出样例：

01000 1 02233
02233 2 00001
00001 7 34891
34891 3 10086
10086 4 00100
00100 6 00033
00033 5 -1

代码如下：

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

struct node {
	int data;
	int next;
	int pre;
}res[100000];

void last(int head, int rear) {
	int sum = 0;
	int p = head, q = rear;
	
	vector<int> answer;

	while (p != -1) {
		answer.push_back(p);
		p = res[p].next;
		sum++;

		if (sum == 2 && q != -1) {
			answer.push_back(q);
			q = res[q].pre;
			sum = 0;
		}
	}

	int flag = 1;
	for (int i = 0; i < answer.size(); i++) {
		if (flag > 1) {
			printf("%05d\n%05d %d ", answer[i], answer[i], res[answer[i]].data);
		}
		else {
			printf("%05d %d ", answer[i], res[answer[i]].data);
			flag++;
		}
	}

	cout << "-1" << endl;
}

int main() {
	int start1, start2, N, address, i, data, next;
	int length1 = 0, length2 = 0;

	cin >> start1 >> start2 >> N;

	for (i = 0; i < N; i++) {
		cin >> address;

		cin >> res[address].data >> res[address].next;
	}

	int p1 = start1;
	int pre1 = -1;
	while (p1 != -1) {
		res[p1].pre = pre1;
		pre1 = p1;
		p1 = res[p1].next;
		length1++;
	}

	int p2 = start2;
	int pre2 = -1;
	while (p2 != -1) {
		res[p2].pre = pre2;
		pre2 = p2;
		p2 = res[p2].next;
		length2++;
	}

	if (length1 >= length2 * 2) {
		last(start1, pre2);
	}
	else if (length1 <= length2 * 2) {
		last(start2, pre1);
	}

	return 0;
}