Числа Фібоначчі#2421

Числа Фібоначчі (Розбір)

Відомо, що для чисел Фібоначчі виконується рівність:

НОД(F_n, F_m) = F_{НОД(n,m)}

Це означає, що задача зводиться до обчислення -го числа Фібоначчі по модулю 10^8, де

k = НОД(n, m)

Оскільки n, m \le 10^{18}, то k ≤ 10^{18}. Отже, необхідно обчислити значення F_k~mod~10^8 за час O(log_2k).

Теорема. Для швидкого обчислення чисел Фібоначчі зручно використовувати матричне піднесення до степеня. Відомо, що:

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

База індукції. При k = 1 маємо:

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

що правильно, оскільки F_0 = 0, F_1 = 1, F_2 = 1.

Крок індукції. Припустимо, що формула правильна для . Тоді для 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}

Таким чином, формула правильна для всіх k \ge 1.

Залишилось реалізувати піднесення матриці до степеня з використанням швидкого піднесення до степеня за час O(log_2k).

Приклад

Числа Фібоначчі мають вигляд:

Обчислимо числа Фібоначчі за допомогою піднесення до степеня матриці:

\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}

Реалізація алгоритму – перевантаження операторів

Оголосимо константу , по модулю якої будуть проводитись обчислення.

#define MOD 100000000

Функція gcd обчислює найбільший спільний дільник двох чисел і .

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

Оголосимо клас матриці Matrix і опишемо конструктор.

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;
  }

Перевантажимо оператор множення матриць. Всі обчислення виконуються по модулю 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;
  }

Перевантажимо оператор піднесення матриці до степеня . Часова складність алгоритму складає 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);
  }
};

Функція fib повертає -е число Фібоначчі F_n по модулю 10^8.

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

Основна частина програми. Читаємо вхідні дані, обчислюємо та виводимо значення F_{НОД(n, m)}.

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

Реалізація алгоритму – функції

Оголосимо константу , по модулю якої будуть проводитись обчислення.

#define MOD 100000000

Функція gcd обчислює найбільший спільний дільник двох чисел і .

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

Оголосимо структуру Matrix — матрицю розміру 2 \times 2. За замовчуванням створюється одинична матриця, що зручно при піднесенні до степеня.

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_) {}
};

Функція multiply перемножує дві матриці 2 \times 2 за стандартною формулою.

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
  );
}

Функція 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;
}

Функція fib обчислює -е число Фібоначчі по модулю 10^8 з використанням матричного методу.

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)
}

Основна частина програми. Читаємо вхідні дані, обчислюємо та виводимо значення F_{НОД(n, m)}.

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

Реалізація алгоритму – запам'ятовування

Для чисел Фібоначчі відоме наступне тотожність:

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

З нього безпосередньо випливають часткові випадки:

Ці формули дозволяють обчислювати числа Фібоначчі методом "розділяй і володарюй", зводячи обчислення F_n до значень з приблизно вдвічі меншими індексами.

Окремо обробляються базові випадки:

F_0 = 0, F_1 = F_2 = 1

Завдяки такому підходу глибина рекурсії пропорційна log_2n, а часова складність алгоритму складає O(log_2n).

Оголосимо константу , по модулю якої будуть проводитись обчислення.

#define MOD 100000000

Оголосимо змінну типу map для запам'ятовування вже обчислених чисел Фібоначчі.

map<long long, long long> F;

Функція gcd обчислює найбільший спільний дільник двох чисел і .

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

Функція fib обчислює -е число Фібоначчі по модулю 10^8.

long long fib(long long n)
{

Окремо обробляємо базові випадки.

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

Якщо значення F_n вже було обчислено раніше і збережено в , воно повертається одразу, без повторних обчислень.

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

Далі використовуємо розклад числа на:

  long long k = n / 2;

Зберігаємо та повертаємо результат.

  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;
}

Основна частина програми. Читаємо вхідні дані, обчислюємо та виводимо значення F_{НОД(n, m)}.

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

Java реалізація

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();
  }
}