Fibonacci numbers#2421

Fibonacci numbers (Editorial)

It is known that Fibonacci numbers satisfy the following identity:

GCD(F_n, F_m) = F_{GCD(n,m)}

This means that the problem reduces to computing the -th Fibonacci number modulo 10^8, where

k = GCD(n, m)

Since n, m \le 10^{18}, we have k ≤ 10^{18}. Therefore, it is necessary to compute the value of F_k~mod~10^8 in O(log_2k) time.

Theorem. To efficiently compute Fibonacci numbers, it is convenient to use matrix exponentiation. It is known that:

\begin{pmatrix} 1 & 1 \\ 1 & 0 \end{pmatrix}^k = \begin{pmatrix} F_{k+1} & F_k \\ F_k & F_{k-1} \end{pmatrix}

Base case. For k = 1, we have:

\begin{pmatrix} 1 & 1 \\ 1 & 0 \end{pmatrix} = \begin{pmatrix} F_2 & F_1 \\ F_1 & F_0 \end{pmatrix}

which is true since F_0 = 0, F_1 = 1, F_2 = 1.

Inductive step. Assume that the formula holds for . Then, for k + 1:

\begin{pmatrix}1 & 1 \\ 1 & 0\end{pmatrix}^{k+1} = \begin{pmatrix}F_{k+1} & F_k \\ F_k & F_{k-1}\end{pmatrix} \begin{pmatrix}1 & 1 \\ 1 & 0\end{pmatrix} = \begin{pmatrix}F_{k+1}+F_k & F_{k+1} \\ F_k+F_{k-1} & F_k\end{pmatrix} = \begin{pmatrix}F_{k+2} & F_{k+1} \\ F_{k+1} & F_k\end{pmatrix}

Thus, the formula is valid for all k \ge 1.

It remains to implement raising a matrix to the power using fast exponentiation in O(log_2k) time.

Example

Fibonacci numbers are defined as follows:

Let's compute the Fibonacci numbers using matrix exponentiation:

\begin{pmatrix}1 & 1 \\ 1 & 0\end{pmatrix}^2 = \begin{pmatrix}1 & 1 \\ 1 & 0\end{pmatrix} \cdot \begin{pmatrix}1 & 1 \\ 1 & 0\end{pmatrix} = \begin{pmatrix}2 & 1 \\ 1 & 1\end{pmatrix} = \begin{pmatrix}F_3 & F_2 \\ F_2 & F_1\end{pmatrix}
\begin{pmatrix}1 & 1 \\ 1 & 0\end{pmatrix}^4 = \begin{pmatrix}2 & 1 \\ 1 & 1\end{pmatrix}^2 = \begin{pmatrix}2 & 1 \\ 1 & 1\end{pmatrix} \cdot \begin{pmatrix}2 & 1 \\ 1 & 1\end{pmatrix} = \begin{pmatrix}5 & 3 \\ 3 & 2\end{pmatrix} = \begin{pmatrix}F_5 & F_4 \\ F_4 & F_3\end{pmatrix}
\begin{pmatrix}1 & 1 \\ 1 & 0\end{pmatrix}^8 = \begin{pmatrix}5 & 3 \\ 3 & 2\end{pmatrix}^2 = \begin{pmatrix}5 & 3 \\ 3 & 2\end{pmatrix} \cdot \begin{pmatrix}5 & 3 \\ 3 & 2\end{pmatrix} = \begin{pmatrix}34 & 21 \\ 21 & 13\end{pmatrix} = \begin{pmatrix}F_9 & F_8 \\ F_8 & F_7\end{pmatrix}

Algorithm implementation – operators overloading

Let us declare the constant , modulo which all computations will be performed.

#define MOD 100000000

The function gcd computes the greatest common divisor of two numbers and .

long long gcd(long long a, long long b)
{
  return (!b) ? a : gcd(b, a % b);
}

Let us declare a Matrix class and write its constructor.

class Matrix
{
public:
  long long a, b, c, d;
  Matrix(long long a = 1, long long b = 0, 
         long long c = 0, long long d = 1) 
  {
    this->a = a; this->b = b;
    this->c = c; this->d = d;
  }

Let us overload the matrix multiplication operator. All computations are performed modulo MOD = 10^8.

  Matrix operator* (const Matrix &x)
  {
    Matrix res;
    res.a = (a * x.a + b * x.c) % MOD;
    res.b = (a * x.b + b * x.d) % MOD;
    res.c = (c * x.a + d * x.c) % MOD;
    res.d = (c * x.b + d * x.d) % MOD;
    return res;
  }

Next, overload the operator for raising a matrix to the power . The time complexity of the algorithm is O(log_2n).

  Matrix operator^ (long long n)
  {
    Matrix x(*this);
    if (n == 0) return Matrix();
    if (n & 1) return x * (x ^ (n - 1));
    return (x * x) ^ (n/2);
  }
};

The function fib returns the -th Fibonacci number F_n modulo 10^8.

long long fib(long long n)
{
  Matrix res(1,1,1,0);
  res = res ^ n;
  return res.b;
}

The main part of the program. Read the input data, compute, and print the value of F_{GCD(n, m)}.

while(scanf("%lld %lld",&n,&m) == 2)
{
  d = gcd(n,m);
  printf("%lld\n",fib(d));
}

Algorithm implementation – functions

Let us declare the constant , modulo which all computations will be performed.

#define MOD 100000000

The function gcd computes the greatest common divisor of two numbers and .

long long gcd(long long a, long long b)
{
  return (!b) ? a : gcd(b, a % b);
}

Declare the Matrix structure — a 2 \times 2 matrix. By default, the identity matrix is created, which is convenient for exponentiation.

struct Matrix 
{
  long long a, b, c, d;
  Matrix(long long a_ = 1, long long b_ = 0,
         long long c_ = 0, long long d_ = 1)
         : a(a_), b(b_), c(c_), d(d_) {}
};

The function multiply multiplies two 2 \times 2 matrices using the standard formula.

Matrix multiply(Matrix &x, Matrix &y) 
{
  return Matrix(
    (x.a * y.a + x.b * y.c) % MOD,
    (x.a * y.b + x.b * y.d) % MOD,
    (x.c * y.a + x.d * y.c) % MOD,
    (x.c * y.b + x.d * y.d) % MOD
  );
}

The function power implements binary exponentiation of the matrix to the power .

Matrix power(Matrix base, long long exp) 
{
  Matrix result;
  while (exp > 0)
  {
    if (exp & 1)
      result = multiply(result, base);
    base = multiply(base, base);
    exp >>= 1;
  }
  return result;
}

The function fib computes the -th Fibonacci number modulo 10^8 using the matrix method.

long long fib(long long n) 
{
  if (n == 0) return 0;
  Matrix m(1, 1, 1, 0);
  Matrix res = power(m, n);
  return res.b; // F(n)
}

The main part of the program. Read the input data, compute, and print the value of F_{GCD(n, m)}.

while (scanf("%lld %lld", &n, &m) == 2) 
{
  d = gcd(n, m);
  printf("%lld\n", fib(d));
}

Algorithm implementation – memoization

The following identity is known for Fibonacci numbers:

F_{n+m} = F_m \cdot F_{n+1} + F_{m-1} \cdot F_n

From it, the following special cases directly follow:

These formulas allow Fibonacci numbers to be computed using a "divide and conquer" approach, reducing the computation of F_n to values with indices approximately half as large.

The base cases are handled separately:

F_0 = 0, F_1 = F_2 = 1

With this approach, the recursion depth is proportional to log_2n, and the time complexity of the algorithm is O(log_2n).

Declare the constant , modulo which all computations will be performed.

#define MOD 100000000

Declare a variable of type map to store (memoize) the Fibonacci numbers that have already been computed.

map<long long, long long> F;

The function gcd computes the greatest common divisor of two numbers and .

long long gcd(long long a, long long b)
{
  return (!b) ? a : gcd(b,a % b);
}

The function fib computes the -th Fibonacci number modulo 10^8.

long long fib(long long n)
{

The base cases are handled separately.

  if (n == 0) return 0;
  if (n == 1) return 1;
  if (n == 2) return 1;

If the value of F_n has already been computed and stored in , it is returned immediately without recomputation.

  if (F[n]) return F[n];

Next, we use the decomposition of the number into:

  long long k = n / 2;

Store and return the result.

  if (n % 2 == 1) // n = 2*k + 1
    return F[n] = (fib(k) * fib(k) + fib(k+1) * fib(k+1)) % MOD;
  else // n = 2*k
    return F[n] = (fib(k) * fib(k+1) + fib(k-1) * fib(k)) % MOD;
}

The main part of the program. Read the input data, compute, and print the value of F_{GCD(n, m)}.

while(scanf("%lld %lld",&a,&b) == 2)
{
  d = gcd(a,b);
  printf("%lld\n",fib(d));
}

Java implementation

import java.util.*;

public class Main
{
  static HashMap<Long, Long> F = new HashMap<Long, Long>();
  static long MOD = 100000000;
  
  static long gcd(long a, long b)
  {
    return (b == 0) ? a : gcd(b,a % b);
  }

  static long fib(long n)
  {
    if (n == 0) return 0;
    if (n == 1) return 1;
    if (n == 2) return 1;
    if (F.containsKey(n)) return F.get(n);

    long k = n / 2;
    if (n % 2 == 1)
    {
      F.put(n, (fib(k) * fib(k) + fib(k+1) * fib(k+1)) % MOD);
      return F.get(n);
    }
    else
    {
      F.put(n, (fib(k) * fib(k+1) + fib(k-1) * fib(k)) % MOD);    	
      return F.get(n);
    }
  }
  
  public static void main(String[] args)
  {
    Scanner con = new Scanner(System.in);
    while(con.hasNext())
    {
      long a = con.nextLong();
      long b = con.nextLong();
      long d = gcd(a,b);
      System.out.println(fib(d));  
    }
    con.close();
  }
}