该题描述中,存在四个生物,人,狼,羊,白菜。分别按照是否过河,一共有2**4种状态
分析:不妨用0表示未过河,1表示过河。我们用1111和0000分别表示人狼羊白菜均过河及均未过河。因此很容易我们就可以想到把状态压缩到一个数字0~15中。
数据结构设计:利用一个0-15的数字,按照2进制从最高位到最低位,依次表示人,狼,羊,白菜。0表示未过河,1表示过河
st[0] = True
#过程描述
dt = {
0:'人和狼过河了',
1:'人和羊过河了',
2:'人和菜过河了',
3:'人自己过河了'
}
比如或运算为:1001|1100 = 1101,下面匿名函数的处理等价于返回一个四位数即,return abcd
nxt = lambda a,b,c,d:a<<3|b<<2|c<<1|d #对四个变量,a,b,c,d进行或处理
在状态计算中,1和0的互相转化,可以用1-?处理。如1-1 = 0 , 1-0 = 1
st = [False]*16#是否已经遍历过某个状态,初始都没有遍历
ans = []#答案序列
def dfs(cur):
if cur == 15:#15的二进制表示为1111,代表均过河
print(' '.join(ans))#输出答案序列
return
peo,wlof,sheep,cab = ((cur>>i)&1 for i in range(3,-1,-1))#获得所有生物的状态,利用二进制的右移。
sta = (peo,wlof,sheep,cab,peo)#状态元组,人狼羊菜 + 人
if peo!= sheep and (sheep == cab or wlof == sheep): return#当人不在时,狼和羊在一起,或者羊和白菜在一起是无效状态,退出
#num为状态转移元组,下列分别表示,当人和狼一起过河,人和羊,人和菜,人自己过河。之后的状态数字
num= (w,s,c,d) = nxt(1-peo,1-peo,sheep,cab),nxt(1-peo,wlof,1-sheep,cab),nxt(1-peo,wlof,sheep,1-cab),nxt(1-peo,wlof,sheep,cab)
for i,j in enumerate(sta[1:]):
if peo == j and (not st[num[i]]):#如果人和该生物在一边,并且当前状态没有被遍历过
st[num[i]] = True#标记该状态已经遍历
ans.append(dt[i])#加入答案序列
dfs(num[i])#继续搜索该状态
ans.pop()#把之前的答案剔除,搜索下一个可能满足的要求
st[num[i]] = False#标记当前的状态未被遍历。
结果展示:
dfs(0)
人和羊过河了 人自己过河了 人和狼过河了 人和羊过河了 人和菜过河了 人自己过河了 人和羊过河了 人和羊过河了 人自己过河了 人和菜过河了 人和羊过河了 人和狼过河了 人自己过河了 人和羊过河了