斐波那契（java）矩阵乘法快速幂

原题链接：205. 斐波那契 - AcWing题库

输入样例：

0
9
999999999
1000000000
-1

输出样例：

0
34
626
6875

思路分析：

时间限制在1s,递推的话，时间复杂度是2 * 10^9  *  100 肯定会TIL 

递归代码：

public static long fibonacci(long number) {
        if ((number == 0) || (number == 1))
            return number;
        else
            return fibonacci(number - 1) + fibonacci(number - 2);
    }

我们可以用矩阵快速幂来实现斐波那契的计算

  

实现代码：

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	static final int MOD = 10000;
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		while(true){
			int n = sc.nextInt();
			if (n == -1) {
				break;
			}
			System.out.println(fib(n));
		}
	}
	
	public static int fib(int n) { //快速幂，把整数乘法换成矩阵乘法
		int[][] a = {{0,1},{0,0}}; //把1*2的矩阵换成2*2的矩阵，结果不影响
		int[][] f = {{0,1},{1,1}};
		while(n != 0) {
			if ((n & 1) == 1) {
				mul(a,f);
			}
			mul(f, f);
			n >>= 1;
		}
		return a[0][0];
	}
	
	public static void mul(int[][] a,int[][] b) { //矩阵乘法的实现
		int[][] c = new int[2][2]; //定义一个矩阵c用于保存运算的结果，因为计算过程中a的值           
                                   //不能发生改变
		for (int i = 0;i < 2;i++) {
			for (int j = 0;j < 2;j++) {
				for (int k = 0;k < 2;k++) {
					c[i][j] = (c[i][j] + a[i][k] * b[k][j]) % MOD;
				}
			}
		}
		for (int i = 0;i < 2;i++) a[i] = Arrays.copyOf(c[i], 2);
	}
}