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

原创已于 2022-04-08 20:02:50 修改 · 441 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #蓝桥杯 #算法 #宽度优先

于 2022-03-31 20:19:22 首次发布

蓝桥杯专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一道X星球青蛙跳杯子的问题，其中青蛙可以跳到相邻、隔一只或隔两只青蛙的空杯中。为了解决这个问题，提出了采用宽度优先搜索（BFS）而非深度优先搜索（DFS）的原因，因为DFS可能导致深搜，而BFS更适合寻找最优解。通过建立搜索树并使用字典存储已搜索过的状态进行剪枝，最终得出从初始局面到目标局面所需的最少步数。代码中详细展示了如何实现这一算法，包括关键的搜索过程和剪枝操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

*WWWBBB

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

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

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

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

WWW*BBB

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

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

*WWBB
WWBB*

样例输出

二、思路和代码

思路：因为这道题用深度优先可能达到很深的位置（可以不停地生成很长的路径），加上这道题求的是最优解，所以采用宽度优先。
技巧：使用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)