导航

18.上楼梯

Posted on 2015-09-09 14:54  骄阳照林  阅读(136)  评论(0编辑  收藏  举报

题目描述

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

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

测试样例:1
返回:1
代码如下:
import java.util.*;

public class GoUpstairs {
    public int countWays(int n) {
        int n1=1;
        int n2=2;
        int n3=4;
        int sum=0;
       if(n==1) return n1;
       if(n==2) return n2;
       if(n==3) return n3;
       for(int i=4;i<=n;i++){
            sum = ((n3+n2)%1000000007+n1)%1000000007;
            n1 = n2;
            n2 = n3;
            n3 = sum;
        }
        return sum;
    }

}