python 穷举2个半潜船去a，b两个集结点运输5个军舰艇(往返运输）

这是一个结合选址、路径、装卸的穷举问题，军舰艇到集结点的不是固定的，也就是说每个军舰艇都可以选择1，2两地集结，所以集结方案就有2**5=32种，然后根据1，2两集结点的军舰艇数量穷举半潜船的往返运输路线。每个半潜船的满载量是4个，也就是说半潜船装完4个就必须去终点，终点用3代替。

注：本问题是组合运输，可以由一艘船运输，也可以由两艘船共同完成运输

1 半潜船1运输路线为[1,2,3],半潜船2运输路线为[2,3]

2 半潜船1运输路线为[1,2,3,1,3],半潜船2运输路线[]

.............

import itertools

# 军舰艇集结点
A = [1, 1, 1, 1, 1]
B = [2, 2, 2, 2, 2]

# 每个半潜船的运输路线
routes = []

# 生成军舰艇集结点选择方案
combinations = itertools.product([1, 2], repeat=5)

for combination in combinations:
    # 计算每个集结点中的军舰艇数量
    a_count = combination.count(1)
    b_count = combination.count(2)

    # 根据半潜船满载量 4 计算需要的往返次数
    trips = (a_count + b_count + 3) // 4

    # 根据往返次数将集结点划分为多个运输段
    segments = []
    for i in range(trips):
        start = i * 4
        end = min(start + 4, a_count + b_count)
        segment = (combination[start:end],)
        if i % 2 == 0:
            segment += (1, 2, 3)
        else:
            segment += (2, 3, 1)
        segments.append(segment)

    # 将所有运输段组合为一条完整的路线
    route = []
    for segment in segments:
        route.extend(segment)
    routes.append(route)

# 输出所有半潜船运输路线
for i, route in enumerate(routes):
    print(f"Route {i+1}: {route}")

该代码使用 itertools 库中的 product() 函数生成了所有军舰艇集结点的选择方案，然后根据半潜船的满载量和往返路线将集结点划分为多个运输段，并将所有运输段组合为一条完整的路线，最终得到所有半潜船的运输路线。