java:台阶问题
题目描述
前几个月放映的头号玩家简直火得不能再火了,作为一个探索终极AI的研究人员,月神自然去看了此神剧。
由于太过兴奋,晚上月神做了一个奇怪的梦,月神梦见自己掉入了一个被施放了魔法的深渊,月神想要爬上此深渊。
已知深渊有N层台阶构成(1 <= N <= 1000),并且每次月神仅可往上爬2的整数次幂个台阶(1、2、4、…),请你编程告诉月神,月神有多少种方法爬出深渊
输入描述:
输入共有M行,(1<=M<=1000)
第一行输入一个数M表示有多少组测试数据,
接着有M行,每一行都输入一个N表示深渊的台阶数
输出描述:
输出可能的爬出深渊的方式
示例1
输入:
4
1
2
3
4
输出:
1
2
3
6
解题思路:
动态规划的思想,和换硬币问题思路差不多,只不过换硬币问题组合不考虑顺序,而跳台阶问题需要考虑跳台阶的先后顺序,即需要考虑组合的顺序
java代码如下:
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner in=new Scanner(System.in);
int n=in.nextInt();
int[] w=new int[n];
int[] step={1,2,4,8,16,32,64,128,256,512};
int[] res=new int[n];
for(int i=0;i<n;i++){
w[i]=in.nextInt();
res[i]=result(w[i],step,step.length);
}
for(int i:res) System.out.println(i);
}
public static int result(int k,int[] step,int n){
int[] dp=new int[k+1];
dp[0]=1;
// for(int i=0;i<n;i++){
// for(int j=step[i];j<=k;j++){
// dp[j]=dp[j]+dp[j-step[i]];
// }
// }
for (int i = 1; i <= k; ++i) {
for (int j = 0; j < step.length && i >= step[j]; ++j) {
dp[i] += dp[i - step[j]];
dp[i] %= 1000000003L;
}
}
return dp[k];
}
}