Java与算法之(3) - 斐波那契数列

斐波那契数列问题：如果一对兔子每月能生1对小兔子，而每对小兔在它出生后的第三个月里，又能开始生1对小兔子，假定在不发生死亡的情况下，由一对初生的兔子开始，1年后能繁殖出多少对兔子？

首先手工计算来总结规律，如下表

Java与算法之(3) - 斐波那契数列-LMLPHP

注意总数这一列

1+1=2

1+2=3

2+3=5

3+5=8

5+8=13

可以得出规律，第n个斐波那契数=第n-1个斐波那契数+第n-2个斐波那契数

为了计算n，必须计算n-1和n-2；为了计算n-1，必须计算n-2和n-3；直到n-x的值为1为止，这显示是递归大显身手的地方。来看代码

public class Fibonacci {
public static long calc(long n) {
if(n < 0) {
return 0;
}
if(n == 0 || n == 1) {
return n;
} else {
return calc(n - 1) + calc(n - 2);
}
}
}

这真是极短的，测试代码

public static void main(String[] args) {
long n = 50;
long begin = System.nanoTime();
long f = Fibonacci.calc(n);
long end = System.nanoTime();
System.out.println("第" + n + "个斐波那契数是" + f + ", 耗时" + TimeUnit.NANOSECONDS.toMillis(end - begin) + "毫秒");
}

运行输出

第50个斐波那契数是12586269025, 耗时66024毫秒

注意看消耗的时间，在我的电脑上耗时66秒，真是个相当耗时的操作。既然整个过程都是在不断重复相同的计算规则，那我们可以采用分而治之的思想来优化代码。

import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.RecursiveTask;
import java.util.concurrent.TimeUnit;
public class Fibonacci extends RecursiveTask<Long> {
long n;
public Fibonacci(long n) {
this.n = n;
}
public Long compute() {
if(n <= 10) { //小于10不再分解
return Fibonacci.calc(n);
}
Fibonacci f1 = new Fibonacci(n - 1); //分解出计算n-1斐波那契数的子任务
f1.fork(); //由ForkJoinPool分配线程执行子任务
Fibonacci f2 = new Fibonacci(n - 2); //分解出计算n-2斐波那契数的子任务
return f2.compute() + f1.join();
}
public static long calc(long n) {
if(n < 0) {
return 0;
}
if(n == 0 || n == 1) {
return n;
} else {
return calc(n - 1) + calc(n - 2);
}
}
public static void main(String[] args) {
long n = 50;
long begin = System.nanoTime();
Fibonacci fibonacci = new Fibonacci(n);
ForkJoinPool pool = new ForkJoinPool();
long f = pool.invoke(fibonacci);
long end = System.nanoTime();
System.out.println("第" + n + "个斐波那契数是" + f + ", 耗时" + TimeUnit.NANOSECONDS.toMillis(end - begin) + "毫秒");
}
}

运行输出

第50个斐波那契数是12586269025, 耗时20461毫秒

虽然时间缩短了2/3，但是仍然不理想。回头重新看计算方法，用递归方式虽然代码简短，但是存在很严重的重复计算，下面用非递归的方式改写，过程中每个数只计算一次。

public static long calcWithoutRecursion(long n) {
if(n < 0)
return 0;
if(n == 0 || n == 1) {
return n;
}
long fib = 0;
long fibOne = 1;
long fibTwo = 1;
for(long i = 2; i < n; i++) {
fib = fibOne + fibTwo;
fibTwo = fibOne;
fibOne = fib;
}
return fib;
}

测试

第50个斐波那契数是12586269025, 耗时0毫秒

斐波那契数的另一个经典题目是青蛙跳台阶问题：

一只青蛙一次可以条一级或两级台阶，求该青蛙跳上n级的台阶共有多少种跳法。

假设计算第n级台阶跳法的函数是f(n)，当n>2时，第一步选择跳一级有X种跳法，第一步选择跳两级有Y种跳法，f(n)=X+Y。如何计算X呢，站在青蛙的位置考虑，面对的是一个全新的n-1级台阶，有f(n-1)种跳法，那么Y就是n-2级台阶的跳法，那么f(n)=f(n-1)+f(n-2)，即斐波那契数列公式。