java-优化斐波那契数列

N阶楼梯上楼问题：一次可以走两阶或一阶，问有多少种上楼方式。例如，当楼梯只有一阶时，只有一种方法；当楼梯有两阶时，可以每次跨一阶，跨两次，也可以每次跨两阶，跨一次，因此有两种方法。

输入格式:

输入包括一个整数N,(1<=N<46)。

输出格式:

输出当楼梯阶数是N时的上楼方式总数。

输入样例1:

3

输出样例1:

3

输入样例2:

4

输出样例2:

5

一般来想，我们可能最先想到的是用递归来完成这个斐波那契数列

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        System.out.println(fun1(n));
    }

    public static int fun1(int n) {
        if (n == 1)
            return 1;
        else if (n == 2)
            return 2;
        else
            return (fun1(n - 1) + fun1(n - 2));
    }
}

 

当楼梯层数过多时，我们会发现运行超时，原因就是递归函数的时间复杂度过高

这里我们可以采用滚动数组来解决这个问题

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        System.out.println(fun2(n));
    }
    public static int fun2(int n) {
//        初始位置按照:p-q-r
        int p = 0, q = 1, r = 1;
//        滚动次数:每次滚动p移到q的位置;q移动到r的位置;r则等于p+q；
        for (int i = 0; i < n - 1; i++) {
            p = q;
            q = r;
            r = p + q;
        }
        return r;
    }
}

这样一来，则解决了运行超时这个问题。