洛谷P1939 【模板】矩阵加速(数列)
生活随笔
收集整理的這篇文章主要介紹了
洛谷P1939 【模板】矩阵加速(数列)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目描述
a[1]=a[2]=a[3]=1
a[x]=a[x-3]+a[x-1] (x>3)
求a數列的第n項對1000000007(10^9+7)取余的值。
輸入格式
第一行一個整數T,表示詢問個數。
以下T行,每行一個正整數n。
輸出格式
每行輸出一個非負整數表示答案。
輸入輸出樣例
輸入 #1
3
6
8
10
輸出 #1
4
9
19
說明/提示
對于30%的數據 n<=100;
對于60%的數據 n<=2*10^7;
對于100%的數據 T<=100,n<=2*10^9;
解析:
\(\displaystyle \begin{array}{{>{\displaystyle}l}} \ 現在我需要求的矩陣是:\\ \begin{bmatrix} f[ i]\\ f[ i-1]\\ f[ i-2] \end{bmatrix}\\ 根據題目中給出的條件:f[ x] =f[ x-1] +f[ x-3]\\ 而我們下一步要求出f[ i+1]\\ 所以f[ i+1] =f[ i] +f[ i-2]\\ 所以求初始矩陣為\\ \begin{bmatrix} 1 & 0 & 1\\ 1 & 0 & 0\\ 0 & 1 & 0 \end{bmatrix}\\ 對初始矩陣進行矩陣快速冪然后輸出a[ 1][ 1] \end{array}\)
#include <cstdio> #include <iostream> #include <cmath> #include <cstring> #include <queue> #include <stack> #define re register #define Max 200000012 #define int long long int n; const int mod=1000000007; struct Mat {int a[4][4];Mat() {memset(a,0,sizeof a);}inline void build() {memset(a,0,sizeof a);for(re int i = 1 ; i <= 3 ; ++ i) a[i][i]=1;} }; Mat operator*(Mat &a,Mat &b) {Mat c;for(re int k = 1 ; k <= 3 ; ++ k)for(re int i = 1 ; i <= 3 ; ++ i)for(re int j = 1 ; j <= 3 ; ++ j)c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j]%mod)%mod;return c; } Mat ans,a,A; void quick_Mat(int x) {ans.build();while(x) {if(x & 1 == 1) ans=ans*a;a=a*a;x>>=1;} } signed main() {scanf("%lld",&n);int x;A.a[1][1]=1;A.a[2][1]=1;A.a[3][1]=1;for(re int i = 1 ; i <= n ; ++ i) {scanf("%lld",&x);if(x<=3 && x>=1) {printf("1\n");continue;}a.a[1][1]=1;a.a[1][2]=0;a.a[1][3]=1;a.a[2][1]=1;a.a[2][2]=0;a.a[2][3]=0;a.a[3][1]=0;a.a[3][2]=1;a.a[3][3]=0;quick_Mat(x-3);ans = ans * A;printf("%lld\n",ans.a[1][1]);}return 0; }轉載于:https://www.cnblogs.com/handsomegodzilla/p/11299441.html
總結
以上是生活随笔為你收集整理的洛谷P1939 【模板】矩阵加速(数列)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: CentOS7安装Zabbix
- 下一篇: 洛谷P1962 斐波那契数列题解