算法筆記_091:藍橋杯練習 遞推求值（Java）

目錄

1 問題描述

2 解決方案

 

---

1 問題描述

問題描述

　　已知遞推公式：

　　F(n, 1)=F(n-1, 2) + 2F(n-3, 1) + 5,

　　F(n, 2)=F(n-1, 1) + 3F(n-3, 1) + 2F(n-3, 2) + 3.

　　初始值為：F(1, 1)=2, F(1, 2)=3, F(2, 1)=1, F(2, 2)=4, F(3, 1)=6, F(3, 2)=5。
　　輸入n，輸出F(n, 1)和F(n, 2)，由於答案可能很大，你只需要輸出答案除以99999999的余數。

輸入格式

　　輸入第一行包含一個整數n。

輸出格式

　　輸出兩行，第一行為F(n, 1)除以99999999的余數，第二行為F(n, 2)除以99999999的余數。

樣例輸入

4

樣例輸出

14

21

數據規模和約定

　　1<=n<=10^18。

 

---

2 解決方案

本題直接用傳統的遞推求解，結果會運行超時。

此處要利用要構造矩陣，來計算相應結果。

首先看一下參考資料1中對於斐波那契數，構造矩陣的示例：

 

其具體相關理解，請參考文末參考資料1哦。

本題構造的矩陣如下：

對應1*8的矩陣為[f(3,1),f(3,2),f(2,1),(2,2),f(1,1),f(1,2),3,5]

public final static long[][] UNIT = {{0,1,1,0,0,0,0,0},
        {1,0,0,1,0,0,0,0},
        {0,0,0,0,1,0,0,0},
        {0,0,0,0,0,1,0,0},
        {2,3,0,0,0,0,0,0},
        {0,2,0,0,0,0,0,0},
        {0,1,0,0,0,0,1,0},
        {1,0,0,0,0,0,0,1}};   //根據遞推公式構造的矩陣

 

具體代碼如下：

import java.util.Scanner;

public class Main {
    public final static long[][] UNIT = {{0,1,1,0,0,0,0,0},
        {1,0,0,1,0,0,0,0},
        {0,0,0,0,1,0,0,0},
        {0,0,0,0,0,1,0,0},
        {2,3,0,0,0,0,0,0},
        {0,2,0,0,0,0,0,0},
        {0,1,0,0,0,0,1,0},
        {1,0,0,0,0,0,0,1}};   //根據遞推公式構造的矩陣
    public final static long[][] ZERO = new long[8][8];  //元素全為0
    public final static long p = 99999999L;
    //獲取矩陣NUIT的n次方結果
    public long[][] getNofMatrix(long n) {
        if(n == 0)
            return ZERO;
        if(n == 1)
            return UNIT;
        if((n & 1) == 0) {  //當n為偶數時
            long[][] matrix = getNofMatrix( n >> 1);
            return multiOfMatrix(matrix, matrix);
        } 
        //當n為奇數時
        long[][] matrix = getNofMatrix((n - 1) >> 1);
        return multiOfMatrix(multiOfMatrix(matrix, matrix), UNIT);
    }
    //計算矩陣A*B取余99999999的值
    public long[][] multiOfMatrix(long[][] A, long[][] B) {
        long result[][] = new long[A.length][B[0].length];
        for(int i = 0;i < A.length;i++) {
            for(int j = 0;j < B[0].length;j++) {
                for(int k = 0;k < A[0].length;k++)
                    result[i][j] = (result[i][j] + A[i][k] * B[k][j]) % p;
            }
        }
        return result;
    }
    
    public void printResult(long n) {
        long[][] start = {{6,5,1,4,2,3,3,5}};
        if(n == 1) {
            System.out.println(start[0][4]+"\n"+start[0][5]);
            return;
        } else if(n == 2) {
            System.out.println(start[0][2]+"\n"+start[0][3]);
            return;
        } else if(n == 3) {
            System.out.println(start[0][0]+"\n"+start[0][1]);
            return;
        }
        long[][] A = getNofMatrix(n - 3);
        start = multiOfMatrix(start, A);
        System.out.println(start[0][0]+"\n"+start[0][1]);
        return;
    }
    
    public static  void main(String[] args) {
        Main test = new Main();
        Scanner in = new Scanner(System.in);
        long n = in.nextLong();
        test.printResult(n);
    }
}

 

 

 

參考資料：

1.矩陣構造方法

2.藍橋杯算法提高 遞推求值 【矩陣快速冪】