matrix_exponentiation.cpp File Reference

Matrix Exponentiation. More...

#include <iostream>
#include <vector>

Include dependency graph for matrix_exponentiation.cpp:

Go to the source code of this file.

Macros
#define	ll   int64_t
#define	endl   std::endl
#define	pb   push_back
#define	MOD   1000000007

Functions
vector< vector< ll > >	multiply (const vector< vector< ll > > &A, const vector< vector< ll > > &B)
vector< vector< ll > >	power (const vector< vector< ll > > &A, ll p)
ll	ans (ll n)
int	main ()

Variables
ll	mat_size
vector< ll >	fib_b
vector< ll >	fib_c

Detailed Description

Matrix Exponentiation.

The problem can be solved with DP but constraints are high.
\(a_i = b_i\) (for \(i <= k\))
\(a_i = c_1 a_{i-1} + c_2 a_{i-2} + ... + c_k a_{i-k}\) (for \(i > k\))
Taking the example of Fibonacci series, \(k=2\)
\(b_1 = 1,\; b_2=1\)
\(c_1 = 1,\; c_2=1\)
\(a = \begin{bmatrix}0& 1& 1& 2& \ldots\end{bmatrix}\)
This way you can find the \(10^{18}\) fibonacci numberMOD. I have given a general way to use it. The program takes the input of B and C matrix.

Steps for Matrix Expo

1. Create vector F1 : which is the copy of B.
2. Create transpose matrix (Learn more about it on the internet)
3. Perform \(T^{n-1}\) [transpose matrix to the power n-1]
4. Multiply with F to get the last matrix of size (1 \(\times\)k).

The first element of this matrix is the required result.

Definition in file matrix_exponentiation.cpp.

Macro Definition Documentation

endl

#define endl   std::endl

shorthand definition for std::endl

Definition at line 36 of file matrix_exponentiation.cpp.

ll

#define ll   int64_t

shorthand definition for int64_t

Definition at line 33 of file matrix_exponentiation.cpp.

MOD

#define MOD   1000000007

Definition at line 40 of file matrix_exponentiation.cpp.

pb

#define pb   push_back

shorthand definition for int64_t

Definition at line 39 of file matrix_exponentiation.cpp.

Function Documentation

ans()

ll ans

(

ll

n

)

Wrapper for Fibonacci

Parameters

[in]

n

\(n^\text{th}\) Fibonacci number

Returns

\(n^\text{th}\) Fibonacci number

Definition at line 91 of file matrix_exponentiation.cpp.

             {
    if (n == 0)
        return 0;
    if (n <= mat_size)
        return fib_b[n - 1];
    // F1
    vector<ll> F1(mat_size + 1);
    for (ll i = 1; i <= mat_size; i++) F1[i] = fib_b[i - 1];
 
    // Transpose matrix
    vector<vector<ll>> T(mat_size + 1, vector<ll>(mat_size + 1));
    for (ll i = 1; i <= mat_size; i++) {
        for (ll j = 1; j <= mat_size; j++) {
            if (i < mat_size) {
                if (j == i + 1)
                    T[i][j] = 1;
                else
                    T[i][j] = 0;
                continue;
            }
            T[i][j] = fib_c[mat_size - j];
        }
    }
    // T^n-1
    T = power(T, n - 1);
 
    // T*F1
    ll res = 0;
    for (ll i = 1; i <= mat_size; i++) {
        res = (res + (T[1][i] * F1[i]) % MOD) % MOD;
    }
    return res;
}

main()

int main

(

void

)

Main function

Definition at line 126 of file matrix_exponentiation.cpp.

           {
    cin.tie(0);
    cout.tie(0);
    ll t;
    cin >> t;
    ll i, j, x;
    while (t--) {
        cin >> mat_size;
        for (i = 0; i < mat_size; i++) {
            cin >> x;
            fib_b.pb(x);
        }
        for (i = 0; i < mat_size; i++) {
            cin >> x;
            fib_c.pb(x);
        }
        cin >> x;
        cout << ans(x) << endl;
        fib_b.clear();
        fib_c.clear();
    }
    return 0;
}

multiply()

vector< vector< ll > > multiply	(	const vector< vector< ll > > &	A,
		const vector< vector< ll > > &	B )

To multiply 2 matrices

Parameters

[in]	A	matrix 1 of size (m \(\times\)n)
[in]	B	matrix 2 of size (p \(\times\)q)

Note

\(p=n\)

Returns

matrix of dimension (m \(\times\)q)

Definition at line 57 of file matrix_exponentiation.cpp.

                                                         {
    vector<vector<ll>> C(mat_size + 1, vector<ll>(mat_size + 1));
    for (ll i = 1; i <= mat_size; i++) {
        for (ll j = 1; j <= mat_size; j++) {
            for (ll z = 1; z <= mat_size; z++) {
                C[i][j] = (C[i][j] + (A[i][z] * B[z][j]) % MOD) % MOD;
            }
        }
    }
    return C;
}

power()

vector< vector< ll > > power	(	const vector< vector< ll > > &	A,
		ll	p )

computing integer power of a matrix using recursive multiplication.

Note

A must be a square matrix for this algorithm.

Parameters

[in]	A	base matrix
[in]	p	exponent

Returns

matrix of same dimension as A

Definition at line 76 of file matrix_exponentiation.cpp.

                                                            {
    if (p == 1)
        return A;
    if (p % 2 == 1) {
        return multiply(A, power(A, p - 1));
    } else {
        vector<vector<ll>> X = power(A, p / 2);
        return multiply(X, X);
    }
}

Variable Documentation

fib_b

vector<ll> fib_b

global vector variables used in the ans function.

Todo

@stepfencurryxiao add documetnation

Definition at line 50 of file matrix_exponentiation.cpp.

fib_c

vector<ll> fib_c

Definition at line 50 of file matrix_exponentiation.cpp.

mat_size

ll mat_size

global variable mat_size

Todo

@stepfencurryxiao add documetnation

Definition at line 45 of file matrix_exponentiation.cpp.