牛客网台阶问题源码实现 动态规划

来源:互联网 发布:皇牌空战 知乎 编辑:程序博客网 时间:2024/05/21 17:01

有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶、3阶。请实现一个方法,计算小孩有多少种上楼的方式。为了防止溢出,请将结果Mod 1000000007

给定一个正整数int n,请返回一个数,代表上楼的方式数。保证n小于等于100000。

题目不解释了,最简单的动态规划

那么问题来了,为什么写出来的例子只过了10% 原因 溢出了 每两个数想加都会溢出 这时候就要取一下模

源代码如下:

import java.util.*;public class GoUpstairs {    public int countWays(int n) {        // write code here        int[] dp=new int[n+1];        if(n<0) return 0;        if(n==1) return 1;        if(n==2) return 2;        dp[0]=1;        dp[1]=1;        dp[2]=2;        for(int i=3;i<=n;i++)        {            dp[i]=((dp[i-1]+dp[i-2])%1000000007+dp[i-3])%1000000007;        }        return dp[n];    }}


0 0
原创粉丝点击