[python回归]Python回溯法子集树模板解决爬楼梯问题

时间:2021-08-16  来源:python  阅读:

Python的使用现在是很受欢迎的,它有着非常不错的使用功能。下面文章就给大家介绍下Python回溯法子集树模板怎么解决爬楼梯,感兴趣的下面我们具体来看看。

问题

某楼梯有n层台阶,每步只能走1级台阶,或2级台阶。从下向上爬楼梯,有多少种爬法?

分析

这个问题之前用分治法解决过。但是,这里我要用回溯法子集树模板解决它。

祭出元素-状态空间分析大法:每一步是一个元素,可走的步数[1,2]就是其状态空间。不难看出,元素不固定,状态空间固定。

直接上代码。

代码

"""爬楼梯"""
n = 7 # 楼梯阶数
x = []  # 一个解(长度不固定,1-2数组,表示该步走的台阶数)
X = []  # 一组解
# 冲突检测
def conflict(k):
  global n, x, X
  # 部分解步的步数之和超过总台阶数
  if sum(x[:k+1]) > n:
    return True
  return False # 无冲突
# 回溯法(递归版本)
def climb_stairs(k): # 走第k步
  global n, x, X
  if sum(x) == n: # 已走的所有步数之和等于楼梯总台阶数
    print(x)
    #X.append(x[:]) # 保存(一个解)
  else:
    for i in [1, 2]: # 第k步这个元素的状态空间为[1,2]
      x.append(i)
      if not conflict(k): # 剪枝
        climb_stairs(k+1)
      x.pop()       # 回溯
# 测试
climb_stairs(0) # 走第0步

效果图

[python回归]Python回溯法子集树模板解决爬楼梯问题

http://m.bbyears.com/jiaocheng/136346.html

推荐访问:python函数返回多个值 python怎么读 python教程 python能做什么
相关阅读 猜你喜欢
本类排行 本类最新