python解决过河问题

只有一艘船，三个商人三个仆人过河，每一次船仅且能坐1~2个人，而且任何一边河岸上仆人比商人多的时候，仆人会杀人越货。

这是一个很经典的过河问题。题解：

#允许状态集合，例num=3
#S={(x,y)|x=0,y=0,1,2,3;x=3,y=0,1,2,3;x=y=1,2} x是此岸的商人数，y是此岸的仆人数
#允许决策集合，例boat_limit=2
#D={(u,v)|1<=u+v<=2,u,v=0,1,2} u是撘载的商人数，v是搭载的仆人数
num=3
boat_limit=2
temp=[]
for i in range(0,num+1):
    if i==0 or i==num:
        for j in range(0,num+1):
            temp.append((i,j))
    else:
        temp.append((i,i))
S=set(temp)
D=[]
for u in range(0,boat_limit+1):
    for v in range(0,boat_limit+1):
        if u+v>=1 and u+v<=boat_limit :
            D.append((u,v))
start=(num,num)
end=(0,0)
queue=[]
queue.append((0,start)) #前面的元素如果是0，说明是船在此岸，是1，说明船在对岸
step_dict={}
flag=0
finish=[]
while len(queue)!=0:
    q_pop=queue.pop(0)
    if q_pop[0]==0:
        for x in D:
            temp_s=(q_pop[1][0]-x[0],q_pop[1][1]-x[1])
            if temp_s not in S:
                continue
            if (1,temp_s) in step_dict:
                continue
            queue.append((1,temp_s))
            step_dict[(1,temp_s)]=q_pop
            if temp_s==end:
                flag=1
                finish=(1,temp_s)
                break
    else:
        for x in D:
            temp_s=(q_pop[1][0]+x[0],q_pop[1][1]+x[1])
            if temp_s not in S:
                continue
            if (0,temp_s) in step_dict:
                continue
            queue.append((0,temp_s))
            step_dict[(0,temp_s)]=q_pop
    if flag==1:
        break
if flag==1:
    print('该问题有解！最短路径：')
    path=[]
    path.append(finish)
    while path[-1]!=(0,start):
        path.append(step_dict[path[-1]])
    path.reverse()
    real_path=list(map(str,path))
    for i in range(len(real_path)):
        if i!=len(real_path)-1:
            print(real_path[i] + '->')
        else:
            print(real_path[i])
else:
    print('该问题无解')

当然，代码已经把扩展到n个人，船的限制是1-m个人的情况了，例中的输出如下：

该问题有解！最短路径：
(0, (3, 3))->
(1, (3, 1))->
(0, (3, 2))->
(1, (3, 0))->
(0, (3, 1))->
(1, (1, 1))->
(0, (2, 2))->
(1, (0, 2))->
(0, (0, 3))->
(1, (0, 1))->
(0, (0, 2))->
(1, (0, 0))

参考资料：《数学模型》第一章