PTA 1105 链表合并(Python3)

给定两个单链表 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

提交结果：

代码（测试点4运行超时）：

import sys

data = sys.stdin.readline().split()
l = []
l1 = [data[0]]
l2 = [data[1]]
l3 = []
l4 = []
for i in range(int(data[2])):
    data1 = input().split()
    l.append(data1)
while l1[-1] != '-1':
    for i in l:
        if i[0] == l1[-1]:
            l1.append(i[2])
            l3.append(i)

while l2[-1] != '-1':
    for i in l:
        if i[0] == l2[-1]:
            l2.append(i[2])
            l4.append(i)

count = 2
if len(l3) > 2 * len(l4):
    l4.reverse()
    for i in l4:
        l3.insert(count, i)
        l3[count - 1][-1] = l3[count][0]
        try:
            l3[count][-1] = l3[count + 1][0]
        except:
            l3[count][-1] = '-1'
        count = count + 3
    for j in l3:
        print(' '.join(j))
else:
    l3.reverse()
    for i in l3:
        l4.insert(count, i)
        l4[count - 1][-1] = l4[count][0]
        try:
            l4[count][-1] = l4[count + 1][0]
        except:
            l4[count][-1] = '-1'
        count = count + 3
    for j in l4:
        print(' '.join(j))