由于前面的回溯法分割回文串这边文章已经梳理了回溯法的基本思想,所以在这里就不介绍了,直接上我们今天的主题——回溯实现台阶问题。在前面的文章也简单的代码实现了台阶问题的方法数有几种,今天是不仅要得到爬台阶的方法有几种,还要得出具体的爬楼梯方案。
直接上代码:
n=10 # 楼梯阶数
x = [] # 得到的一个解
x_set = [] # 得到满足要求的解得集合
# 冲突检测
def conflict(k):
global n, x, x_set
#如果存在解得步数之和超过总的台阶数,则产生冲突,即返回True
if sum(x) > n:
return True
return False # 无冲突
def climb(k):
global n, x, x_set
# 递归回溯
if sum(x) == n: # 已走的所有步数之和等于楼梯总台阶数
#此时走过的步数之和等于楼梯总台阶数,则输出此解,并将此解放入解集合x_set中
print(x)
x_set.append(x)
else:
for i in [1, 2]:
x.append(i)
#如果当前走的第k步没有检测到冲突,则递归进行下一层即递归处理k+1步
if not conflict(k):
climb(k+1)
x.pop() # 回溯
# 测试
if __name__ == "__main__":
#调用climb函数,返回所有的解,并输出解得个数
climb(1)
print(len(x_set))
大家在学习的过程中基础一定要打牢哦~ 有不懂的地方欢迎评论区留言~(不过加了注释好像也没什么不懂的了吧~哈哈)