求解SJTU OJ-1022 过程中的收获

一 Fibonacci数列的矩阵表示

由Fibonacci数列递推式，设数列中连续两项a=F(i+1),b=F(i),由a，b得到递推到下两项（a’,b’)的式子为a’=a+b,b’=a,如果把(a,b)看作二维向量，则递推过程等价于乘上一个二阶向量
1 1
1 0
该矩阵同时是
F(2) F(1)
F(1) F(0)
由分块矩阵的乘积可以得到Fibonacci数列的矩阵表示

（图片来自POJ）

这样一来，求Fibonacci数列第n项转化为求n次幂的问题

二 快速幂

利用位移运算符求底数的n次幂，将时间复杂度由普通的O(n)减小到O(log2(n))
思路：按照指数的二进制表示循环求幂；

int quick_pow(int a, int b)
{
    int ans = a, base = b;
    while(b != 0)
    {
        if(b & 1) //若b的二进制最低位为1，执行a*=base
            a *= base;
        base *= base; //变为下一位的base
        b >>= 1;
    }
    return ans;
}

OJ-1022 Fib数列

Description
定义Fib数列：1,1,2,3,5,8,13,…
求第N项除以2010的余数

Input Format
输入仅一行，为一个整数N
Output Format
输出仅一行，为第N项除以2010的余数

Sample Input
3
Sample Output
2
Limits:
对于70%的数据 N≤1,000,000
对于100%的数据 N≤210,000,000,000

#include <iostream>
#include <stdlib.h>
using namespace std;
class matrix2
{
    friend matrix2 operator* (const matrix2& obj1, const matrix2& obj2);
private:
    int arr[2][2];
public:
    matrix2(int a = 0, int b = 0, int c = 0, int d = 0)
    {
        arr[0][0] = a;
        arr[0][1] = b;
        arr[1][0] = c;
        arr[1][1] = d;
    }
    matrix2(const matrix2& obj)
    {
        for(int i = 0; i < 2; ++i)
        for(int j = 0; j < 2; ++j)
            arr[i][j] = obj.arr[i][j];
    }
    matrix2& operator%= (int a);
    int fetch(int a, int b);
};
matrix2 operator* (const matrix2& obj1, const matrix2& obj2)
{
    matrix2 tmp;
    for(int i = 0; i < 2; ++i)
        for(int j = 0; j < 2; ++j)
        {
            for(int k = 0; k < 2; ++k)
                tmp.arr[i][j] += obj1.arr[i][k] * obj2.arr[k][j];
        }
    return tmp;
}
matrix2& matrix2::operator%= (int a)
{
    for(int i = 0; i < 2; ++i)
        for(int j = 0; j < 2; ++j)
            arr[i][j] %= a;
    return *this;
}
int matrix2::fetch(int a, int b)
{
    if(a < 0 && a > 1 && b < 0 && b > 1)
        exit(-233333);
    else
        return arr[a][b];
}
int fib_mod(long long power, int MOD)
{
    matrix2 ans(1,0,0,1), base(1,1,1,0);
    while(power != 0)
    {
        if(power & 1)
        {
            ans = ans * base;
            ans %= MOD;
        }
        base = base * base;
        base %= MOD;
        power >>= 1;
    }
    return (ans.fetch(0,1));
}
int main()
{
    long long power;
    cin >> power;
    cout << fib_mod(power,2010);
    return 0;
}