[藍橋杯訓練系統] ALGO-199 奇異的蟲羣 （矩陣快速冪） Apare_xzc

[藍橋杯訓練系統] ALGO-199 奇異的蟲羣 （矩陣快速冪） Apare_xzc

---

---

題面：

---

分析：

就是求斐波那契數列第n項

可用矩陣快速冪

A1 = 1   B1 = 1
A2 = 2   B2 = 1
A3 = 3   B3=  2
A4 = 5   B4 = 3
A5 = 8   B5 = 5
...
A[i+1] = A[i]+B[i]
B[i+1] = A[i]

矩陣遞推式如下：

| A[i+1]  |        =      |1  1|     *      |  A[i]  |
| B[i+1]  |               |1  0|            |  B[i]  |

---

我的AC代碼

#include <bits/stdc++.h>
using namespace std;
#define LL long long
const int N = 5;
const int mod = 1000000007;
LL arr[N][N];
void Mul(LL a[N][N],LL b[N][N],int n) /// a(n,n) = a(n,n) * b(n,n)
{
	LL c[N][N] = {0};
	for(int row=0; row<n; ++row)
		for(int col=0; col<n; ++col)
			for(int k=0;k<n;++k)
				c[row][col] = (a[row][k]*b[k][col]%mod+c[row][col])%mod;
	for(int i=0; i<n; ++i)
		for(int j=0; j<n; ++j)
			a[i][j] = c[i][j];
} 
void solve(LL a[N][N],int n,LL m)
{
	LL ans[N][N] = {0};
	for(int i=0; i<n; ++i) ans[i][i] = 1;
	while(m>0)
	{
		if(m&1) Mul(ans,a,n);
		Mul(a,a,n);
		m >>= 1;
	}
	printf("%lld\n",ans[0][0]); 
}
int main()
{
	LL m;
	while(cin>>m)
	{
		arr[0][0] = arr[0][1] = arr[1][0] = 1;
		arr[1][1] = 0;
		solve(arr,2,m);
	}
	return 0;
}

---

---

矩陣快速冪入門水題
xzc
2020.1.13 16:09

---