练习题7:袋鼠过河(使用了动态规划求解)
题目描述
一只袋鼠要从河这边跳到河对岸,河很宽,但是河中间打了很多桩子,每隔一米就有一个,每个桩子上都有一个弹簧,袋鼠跳到弹簧上就可以跳的更远。每个弹簧力量不同,用一个数字代表它的力量,如果弹簧力量为5,就代表袋鼠下一跳最多能够跳5米,如果为0,就会陷进去无法继续跳跃。河流一共N米宽,袋鼠初始位置就在第一个弹簧上面,要跳到最后一个弹簧之后就算过河了,给定每个弹簧的力量,求袋鼠最少需要多少跳能够到达对岸。如果无法到达输出-1。
输入描述:
输入分两行,第一行是数组长度N (1 ≤ N ≤ 10000),第二行是每一项的值,用空格分隔。
输出描述:
输出最少的跳数,无法到达输出-1。
Python解法:
# -*- coding:utf-8 -*-
"""
袋鼠过河
"""
import sys
def main():
dp = [0]
for i in range(1, n+1):
dp.append(-1)
print(th)
print(dp)
for i in range(n):
if dp[i] == -1:
continue
j = 1
while j <= th[i] and i + j <= n:
if dp[i+j] == -1 or dp[i+j] >= dp[i] + 1:
dp[i + j] = dp[i] + 1
j += 1
print(dp)
if dp[n] == -1:
print(-1)
else:
print(dp[n])
if __name__ == '__main__':
n = int(sys.stdin.readline().strip())
th = list(map(int, sys.stdin.readline().strip().split()))
main()