【递归】普通递归关系 矩阵快速幂

问题 B: 【递归】普通递归关系

题目描述

#include <bits/stdc++.h>
using namespace std;

typedef vector <int> vec;
typedef vector <vec> mat;
typedef long long ll;

const int M = 10000;

mat mul(mat &A, mat &B){

mat C(A.size(), vec(B[0].size()));
for (int i = 0; i < A.size(); i++){
for (int k = 0; k < B.size(); k++){
for (int j = 0; j < B[0].size(); j++){
C[i][j] = (C[i][j]+A[i][k]*B[k][j]) % M;
}
}
}
return C;
}

mat pow(mat A, ll n){

mat B(A.size(), vec(A.size()));
for (int i = 0; i < A.size(); i++){
B[i][i] = 1;
}
while (n > 0){

if (n & 1) B = mul(B, A);
A = mul(A, A);
n >>= 1;
}
return B;
}
ll  n,a,b,c,d;
void slove(){
mat A(2, vec(2));
A[0][0] = a; A[0][1] = b;
A[1][0] = c; A[1][1] = d;
A = pow(A, n);
printf("%d\n",A[1][0]);
}
int main() {
cin>>a>>b>>c>>d>>n;
slove();
return 0;
}