京东 数据分析实习生笔试编程第二题 跳格子的方法

def dfs(visited,current,h0):
    if current<=n-1:
        if current==n-1: 
            if visited not in final_visited:
                final_visited.append(visited[:])
        else:            
            for i in [h0,ai[current]]:                                             
                xx=current+i # follow h0
                h0=i
                visited.append(xx)
                dfs(visited,xx,h0)
                visited.pop()
n,h0=[int(s) for s in input().split()]
ai=[int(s) for s in input().split()]
visited=[0]
current=0
current=0
final_visited=[]
dfs(visited,current,h0)
print(len(final_visited))