题解 | #换队伍#

换队伍，思路，如果暴力穷举会超时，因此需要使用字典降低时间复杂度。将字典的数据结构表示为：{排队顾客编号:[队编号(为了方便处理取1或者-1),用来排序的号数]}，如下图所示：

然后遍历需要变动的数组y，每次变动后，编号都要增大。
假设y为[3,10,6]，则字典变成：

变动完之后将队伍抽取出来，按照排序号数排序就行了。
届时duiwu1=[[排序号数，顾客编号]]，将duiwu1和duiwu2分别按照排序号数排队就行了。

x = list(map(int, input().split()))
y = list(map(int, input().split()))

duiwu1 = list(range(1, x[0] + 1))
duiwu2 = list(range(x[0] + 1, x[0] + x[1] + 1))

zidian = {}

for i in range(len(duiwu1)):
    zidian[duiwu1[i]] = [-1, i]
for i in range(len(duiwu2)):
    zidian[duiwu2[i]] = [1, x[0] + i]

n = x[0] + x[1]
for i in y:
    zidian[i][0] *= -1
    n += 1
    zidian[i][1] = n

duiwu1 = []
duiwu2 = []
for key, value in zidian.items():
    if value[0] == -1:
        duiwu1.append([value[1], key])
    else:
        duiwu2.append([value[1], key])
duiwu2.sort()
duiwu1.sort()

res = ''
for i in duiwu1:
    res += str(i[1]) + ' '
print(res[:-1])

res = ''
for i in duiwu2:
    res += str(i[1]) + ' '
print(res[:-1])