斐波那契加强版

一、题目

题目描述

对于斐波拉契经典问题，我们都非常熟悉，通过递推公式F(n) = F(n - 1) + F(n - 2)，我们可以在线性时间内求出第n项F(n)，现在考虑斐波拉契的加强版，我们要求的项数n的范围为int范围内的非负整数，请设计一个高效算法，计算第n项F(n-1)。第一个斐波拉契数为F(0) = 1。

给定一个非负整数，请返回斐波拉契数列的第n项，为了防止溢出，请将结果Mod 1000000007。
测试样例：

3

返回：2

二、分析

如果想要解决本问题,有两种解决方案,一种是使用矩阵快速幂的方法,另一种是通过数学公式。
(一)、矩阵快速幂求斐波那契数。
看懂下面代码所需的数学知识:
1、矩阵的相乘—>nXm的矩阵A和mXn的矩阵B相乘得到的是nXn的矩阵C,矩阵C中的(i,j)是由矩阵A中的第i行和矩阵B中的第j列逐个乘积的和。
2、单位矩阵—->一个单位矩阵成以另一个矩阵A,得到的乘积还是矩阵A。

public class 矩阵快速幂 {
    public static int[][] mul(int[][] x, int[][] y) {
        int[][] res = new int[2][2];//res用于存储x矩阵和y矩阵的乘积
        for (int i = 0; i < 2; i++)
            for (int j = 0; j < 2; j++)
                for (int k = 0; k < 2; k++) {
                    res[i][j] += x[i][k] * y[k][j];
                    res[i][j] %= 1000007;//完成矩阵x与y的相乘
                }
        return res;
    }

    public static int pow(int n) {
        int[][] c = new int[2][2];
        int[][] res = new int[2][2];
        c[0][0] = 1;//矩阵初始化
        c[0][1] = 1;
        c[1][0] = 1;
        c[1][1] = 0;
        for (int i = 0; i < 2; i++)
            res[i][i] = 1;// 单位矩阵；
        while (n != 0) {
            if ((n & 1) != 0)
                res = mul(res, c);//矩阵相乘
            c = mul(c, c);
            n = n >> 1;
        }
        for(int i=0;i<2;i++){
            for(int j=0;j<2;j++){
                System.out.print(res[i][j]+" ");
            }
            System.out.println();

        }
        return res[0][0];
    }

    public static void main(String[] args) {
        pow(3);
        //System.out.println(pow(3));
    }

}

那么斐波那契树和矩阵快速幂有什么关联呢?
我们可以把斐波那契数的公式写成下面的形式:
F(n)=1*F(n-1)+1*F(n-2)
F(n-1)=1*F(n-1)+0*F(n-2)
如果我们用矩阵表示上面的公式:

上面矩阵还可以写成下面的形式

至此就可以发现斐波那契数的求解可以转化为两个矩阵的乘积的求解。

public int getNthNumber(int n) {
        long[][] base = { { 1, 1 }, { 1, 0 } };//图片中的矩阵--->求它的n-1次幂
        long[][] ret = { { 1, 0 }, { 0, 1 } };//单位矩阵
        while (n != 0) {
            //n%2==1
            if ((n&1)!= 0)
                ret = MultMetri(ret, base);
            base = MultMetri(base, base);
            n = n >> 1;
        }
        return (int) (ret[0][0]);//F(n)
    }

    public static long[][] MultMetri(long[][] ret, long[][] base) {
        long[][] tmp = new long[2][2];
        for (int i = 0; i < 2; i++)
            for (int j = 0; j < 2; j++)
                tmp[i][j] = (ret[i][0] * base[0][j] + ret[i][1] * base[1][j]) % 1000000007;//得到矩阵的乘积
        for (int k = 0; k < 2; k++)
            for (int q = 0; q < 2; q++)
                ret[k][q] = tmp[k][q];
        return ret;
    }

以上参考:http://blog.csdn.net/nyist_tc_lyq/article/details/52981353

(二)、数学公式求斐波那契数列
思路：
初始值：f(0) = 1, f(1) = 1, f(2) = 2, f(3) = 3, f(4) = 5…
以f(10)为例子找规律：f(10)=f(9)+f(8)=2f(8)+f(7)=3f(7)+2f(6)=5f(6) + 3f(5)=f(4)f(6)+f(3)f(5) …
发现规律：F(n) = F(k)F(n-k) + F(k-1)F(n-k-1), k>=1
推导出如下公式:
F(2n) = F(n) F(n) + F(n-1) F(n-1), n>=1
F(2n+1) = F(n+1) F(n) + F(n) F(n-1), n>=1
有了这个公式便可以很容易地写出代码。

Map<Integer, Integer> mp = new HashMap<Integer, Integer>();

    public int getNthNumber(int n) {
        if (n == 1 || n == 0) {
            return 1;
        }
        if (n == 2) {
            return 2;
        }
        if (mp.containsKey(n)) {//已求解过该斐波那契数
            return mp.get(n);
        }
        long fn1 = getNthNumber(n / 2 + 1), fn_1 = getNthNumber(n / 2 - 1), fn = getNthNumber(n / 2);
        if (n % 2 == 1) {//奇数时
            mp.put(n, (int) ((fn * fn1 + fn * fn_1) % 1000000007));
        } else {//偶数时
            mp.put(n, (int) ((fn * fn + fn_1 * fn_1) % 1000000007));
        }
        return mp.get(n);
    }

以上参考:https://www.nowcoder.com/questionTerminal/2393c500d43a4293aa7a662274aff4d1

Don’t let the past steal your present