题目:
求斐波那契数列的第n项。
写一个函数,输入n,求斐波那契(Fibonacci)数列的第n项。
分析:
根据斐波那契数列的定义,最容易想到的解法就是递归,但是递归的效率很慢,因为存在大量的重复计算问题。继而可以想到递推公式。
解法:
效率很低的解法,挑剔的面试官不会喜欢
package com.wsy;
public class Main {
public static void main(String[] args) {
System.out.println(fun(10));
}
public static int fun(int n) {
if (n <= 0) {
return 0;
}
if (n == 1) {
return 1;
}
return fun(n - 1) + fun(n - 2);
}
}
面试官期待的实用解法
package com.wsy;
public class Main {
public static void main(String[] args) {
System.out.println(fun(10));
}
public static int fun(int n) {
if (n <= 0) {
return 0;
}
if (n == 1) {
return 1;
}
int f1 = 0;
int f2 = 1;
int result = 0;
for (int i = 2; i <= n; i++) {
result = f1 + f2;
f1 = f2;
f2 = result;
}
return result;
}
}
时间复杂度O(logn)但不够实用的解法
首先需要了解到一个数学公式(如下),如果我们要求f(n),只需要求矩阵的n-1次幂即可,然后取出矩阵的a[0][0]就是结果了。再来看矩阵的n-1次幂,还是需要O(n)的时间复杂度,这时候,又要用到另一个公式(如下),至此,我们可以开始写代码了。
package com.wsy;
public class Main {
public static void main(String[] args) {
int[][] init = {{1, 1}, {1, 0}};
System.out.println(fun(init, 10 - 1)[0][0]);
}
public static int[][] fun(int[][] a, int n) {
if (n <= 1) {
return a;
}
int[][] result = matrixMultiplication(fun(a, n >> 1), true);
return n % 2 == 0 ? result : matrixMultiplication(result, false);
}
public static int[][] matrixMultiplication(int[][] a, boolean flag) {
int[][] b = {{a[0][0], a[0][1]}, {a[1][0], a[1][1]}};
if (flag) {
a[0][0] = b[0][0] * b[0][0] + b[0][1] * b[1][0];
a[0][1] = b[0][0] * b[0][1] + b[0][1] * b[1][1];
a[1][0] = b[1][0] * b[0][0] + b[1][1] * b[1][0];
a[1][1] = b[1][0] * b[0][1] + b[1][1] * b[1][1];
} else {
a[0][0] = b[0][0] * 1 + b[0][1] * 1;
a[0][1] = b[0][0] * 1 + b[0][1] * 0;
a[1][0] = b[1][0] * 1 + b[1][1] * 1;
a[1][1] = b[1][0] * 1 + b[1][1] * 0;
}
return a;
}
}
题目:
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个n级台阶总共有多少种跳法。
分析:
先考虑简单的情况,假设只有1级台阶,显然只有1种跳法。如果有2级台阶,就有2种跳法:分两次跳,每次跳1级;分一次跳,每次跳2级。
再来分析一般情况,把n级台阶的跳法看做n的函数,记作f(n)。当n>2时,第一次跳就有两种选择:第一次跳1级,此时后面的跳法是f(n-1)种;第一次跳2级,此时后面的跳法是f(n-2)种,于是f(n)=f(n-1)+f(n-2)。
解法:
因为我们分析出f(n)是斐波那契数列了,所以解法同上。