蓝桥杯-青蛙跳杯子（bfs， python）

题目 1878: 蓝桥杯2017年第八届真题-青蛙跳杯子
时间限制: 1Sec 内存限制: 128MB 提交: 602 解决: 265
题目描述
X星球的流行宠物是青蛙，一般有两种颜色：白色和黑色。
X星球的居民喜欢把它们放在一排茶杯里，这样可以观察它们跳来跳去。
如下图，有一排杯子，左边的一个是空着的，右边的杯子，每个里边有一只青蛙。

*WWWBBB

其中，W字母表示白色青蛙，B表示黑色青蛙，*表示空杯子。

X星的青蛙很有些癖好，它们只做3个动作之一：

1. 跳到相邻的空杯子里。
2. 隔着1只其它的青蛙（随便什么颜色）跳到空杯子里。
3. 隔着2只其它的青蛙（随便什么颜色）跳到空杯子里。

对于上图的局面，只要1步，就可跳成下图局面：

WWW*BBB

本题的任务就是已知初始局面，询问至少需要几步，才能跳成另一个目标局面。

输入
输入为2行，2个串，表示初始局面和目标局面。
输出
输出要求为一个整数，表示至少需要多少步的青蛙跳。
样例输入

*WWBB
WWBB*

样例输出

2

二、思路和代码

思路：因为这道题用深度优先可能达到很深的位置（可以不停地生成很长的路径），加上这道题求的是最优解，所以采用宽度优先。
技巧：使用dict存搜索过的结点，用以剪枝。（其实用set可能更快，因为查看是否在set当中是用的哈希查找，时间是O（n））。

begin = list(input())
end = list(input())
n = len(begin)
# 深度优先搜索错！因为这道题广度不大，深度很大！
# searched记录该局面到结果的最优步数
searched = dict()

# 宽度优先,存当前状态和到该状态的步数【state， steps】
qu = []
qu.append(begin+[0])
res = 0
while len(qu) != 0:
    state = qu.pop()
    state, steps = state[0:-1], state[-1]
    stateStr = "".join(state)
    # 到达目标
    if state == end:
        res = steps
        break
    # 搜索过
    elif stateStr in searched.keys():
        continue
    else:
        # 加入搜索过
        searched[stateStr] = 1
        # *和i位置交换,将[state,steps]加入qu
        pos = state.index("*")
        for i in range(max(0, pos - 3), min(n - 1, pos + 3) + 1):
            state[pos], state[i] = state[i], state[pos]
            qu.insert(0, state[:]+[steps+1])
            state[pos], state[i] = state[i], state[pos]
                    
print(res)