Gena Playing Hanoi

思路:首先思路是BFS,要把结果都放到第一个柱子上,所以后面3个的状态顺序无所谓,可以可以快6倍

from collections import deque

def serialize(a):
#    return tuple(sorted([tuple(t) for t in a]))
    l=[tuple(t) for t in a]
    l[1:]=sorted(l[1:])
    return tuple(l)
def deserialize(a):
    return deque([deque(t) for t in a])
def copy(a):
    return deque([deque(t) for t in a])
    
def helper(n,aa):
    a=[deque() for _ in range(4)]
    for i,v in enumerate(aa): a[v-1].append(i+1)
    step=1
    q,qq=[a],[]
    vis=set()
    vis.add(serialize(a))
    target=serialize([deque(range(1,n+1)),deque(),deque(),deque()])
    while q:
        while q:
            a=q.pop()
            for i in range(4):
                for j in range(4):
                    if a[i] and (not a[j] or a[i][0]<a[j][0]):
                        # a2=deserialize(a)
                        a2=copy(a)
                        a2[j].appendleft(a2[i].popleft())
                        a3=serialize(a2)
                        if a3==target: return step
                        if a3 in vis: continue
                        vis.add(a3)
                        qq.append(a2)
        q,qq=qq,q
        step+=1
                        
    
if __name__ == '__main__':
    N = int(input())
    a = list(map(int, input().rstrip().split()))
    print(helper(N,a))

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值