5届国赛java试题 H: 台阶

台阶

---

有n级台阶。从地面（第0级）出发，首先连续的上台阶，上到不超过第n级的某一个位置后再连续的下台阶，直到回到地面。若每次上下台阶只允许走1级或2级，请问可能的上下台阶的方案数是多少？特别地，在0级站着不动也算一种方案。
数据格式：
输入一行包含两个正整数n和m。
输出一个整数，表示n级台阶有多少种合法的走楼梯方案，答案对m取余。
例如：输入：
2 10007
程序应该输出
6
【样例说明1】
共有6种方案(其中+表示上台阶，-表示下台阶)：
(1) 原地不动
(2) +1 -1
(3) +2 -2
(4) +2 -1 -1
(5) +1 +1 -2
(6) +1 +1 -1 -1
再例如，输入：
3 14
程序应该输出：
1
【样例说明2】
共有15种方案，对14取余后得1。
思路：
和普通的台阶题是一个道理只是他可以到0到n个台阶，那我们for遍历从0到n个台阶的dfs就可以了，这题还有个信息点就是下来也有不同的方法我们当上去到n台阶的方法乘一个平方就可以得到最终的次数了，然后次数全部统计起来就是我们要的结果了。
程序：

n,m=map(int,input().split())
s=0

def dfs(ce):
    global con
    if w<=ce:
        if w==ce:
            con+=1
        return 
    for i in range(1,3):
        dfs(ce+i)
for i in range(n+1):
    w=i
    con=0
    dfs(0)
    s+=(con*con)%m
print(s%m)

禁止转载。仅用于自己学习。对程序错误不负责。