华为OD-2024年C卷D卷-学生重新排队[200分]Python3实现100%通过率

题目描述

n 个学生排成一排，学生编号分别是 1 到 n，n 为 3 的整倍数。

老师随机抽签决定将所有学生分成 m 个 3 人的小组（n == 3 * m） ，

为了便于同组学生交流，老师决定将小组成员安排到一起，也就是同组成员彼此相连，同组任意两个成员之间无其它组的成员。

因此老师决定调整队伍，老师每次可以调整任何一名学生到队伍的任意位置，计为调整了一次， 请计算最少调整多少次可以达到目标。

注意：对于小组之间没有顺序要求，同组学生之间没有顺序要求。

输入描述

两行字符串，空格分隔表示不同的学生编号。

• 第一行是学生目前排队情况
• 第二行是随机抽签分组情况，从左开始每 3 个元素为一组

n 为学生的数量，n 的范围为 [3, 900]，n 一定为 3 的整数倍

第一行和第二行元素的个数一定相同

输出描述

老师调整学生达到同组彼此相连的最小调整次数

备注

同组相连：同组任意两个成员之间无其他组的成员，比如有两个小组 [4, 5, 6] 和 [1, 2, 3]，

以下结果都满足要求：

1,2,3,4,5,6；

1,3,2,4,5,6；

2,3,1,5,6,4；

5,6,4,1,2,3；

以下结果不满足要求：

1,2,4,3,5,6；（4与5之间存在其他组的成员3）

用例1

输入

7 9 8 5 6 4 2 1 3
7 8 9 4 2 1 3 5 6

输出

1

说明

学生目前排队情况：7 9 8 5 6 4 2 1 3

学生分组情况：7 8 9 4 2 1 3 5 6

将3调整到4之前，队列调整为：7 9 8 5 6 3 4 2 1，那么三个小组成员均彼此相连 [7 9 8] [5 6 3] [4 2 1]

输出1

用例2

输入

8 9 7 5 6 3 2 1 4
7 8 9 4 2 1 3 5 6

输出

0

说明

学生目前排队情况：8 9 7 5 6 3 2 1 4

学生分组情况：7 8 9 4 2 1 3 5 6

无需调整，三个小组成员均彼此相连：

[7 8 9] [4 2 1] [3 5 6]

输出0

arr = [int(_) for _ in input().split()]
n = len(arr)
id = [0] * (n + 1)
arr2 = [int(_) for _ in input().split()]
def work(q, id):
    dq = []
    while len(q) > 0:
        f = q.pop(0)
        a, b = f
        if a != id:
            if not len(dq) or dq[-1][0] != a:
                dq.append([a, b])
                continue
            dq[-1][1] += b
    return dq
for i in range(n):
    id[arr2[i]] = i // 3
q = []
for num in arr:
    if len(q) == 0 or q[-1][0] != id[num]:
        q.append([id[num], 1])
    else:
        q[-1][1] += 1
d = 0
while len(q) > 0:
    id1,cnt = q.pop(0)
    if cnt == 2:
        d += 1
        q = work(q, id1)
        continue
    if cnt==3:
        continue
    id2,cnt1 = q[0]
    while cnt1 == 3:
        q.pop(0)
        id2,cnt1 = q[0]
    if id2 == id1 and cnt1 == 2:
        d += 1
        q.pop(0)
        continue
    d += 2
    q = work(q, id1)
print(d)

# 7 9 8 5 6 4 2 1 3
# 7 8 9 4 2 1 3 5 6
# 1
# 8 9 7 5 6 3 2 1 4
# 7 8 9 4 2 1 3 5 6
# 0