XDOJ1197 - 科协训练-斐波那契数列

Description

   闲的无聊的jxy有一天对斐波那契数列感兴趣了，希望能知道任意一位的斐波那契数，但因为能力不足，不会计算，所以希望聪明的你能帮助他解决这个问题。具体描述见输入说明。

Input

多组数据，输入exit结束
第一行为一个字符串x 代表选择序号
x为exit时，程序退出
x为1时 ，存在第二行 ，第二行为一个数字n ，代表要输出第n个斐波那契数(n<10^18)
x为2时，输出 第1，10,100……，1000000000位斐波那契数，共十个，一行一个
x为其他数字，输出none
x为其他字符，输出error
所有结果对1000000007取余。
每组数据后输出10个-。

Output

输出详见样例。

Sample Input

1
2
1
1
2
8
n
1
2
exit

Sample Output

1
----------
1
----------
1
X
X
X
X
X
X
X
X
X
X（X为那一位的斐波那契数）
----------
none
----------
error
----------
1
----------

解题思路:

Fibonacci数列是一个很常见的数列，定义很简单
Fib(0) = 1
Fib(1) = 1
Fib(n) = Fib(n-1) + Fib(n-2)
常见的计算方法主要有两种：递归和循环。
其中递归的方法最容易理解，但复杂度高。可以用备忘录的方式解决重复计算，从而取得更好的结果
第二种办法的循环，从小到大算起，fib(2)=fib(0)+fib(1), fib(3)=fib(2)+fib(1)... fib(n)=fib(n-1)+fib(n-2)，只要O(n)的复杂度，是一种很另人满意的算法。
但是在编程珠玑上看到了第三种方法：矩阵相乘
Fib(i+2) = Fib(i+1) + Fib(i)
[Fib(i+2), Fib(i+1)] = [Fib(i+1)+Fib(i), Fib(i+1)] = [Fib(i+1)*1+Fib(i)*1, Fib(i+1)*1+Fib(i)*0]
=[Fib(i+1), Fib(i)] * [1, 1]
                             [1, 0]
设A=[1,1]
        [1,0]
则 [Fib(n), Fib(n-1)] = [Fib(n-1), Fib(n-2)]*A = [Fib(n-2), Fib(n-3)] * A * A
=[Fib(1), Fib(0)] * A^(n-1)
这个问题就转化成了求A^n的问题，而A^n可以表示成A^(2^i1) * (A^(2^i2)) * (A^(2^i3))的形式，因此复杂度为O(log2N)

#include<iostream>
#include<string>
using namespace std;

const int modu = 1000000007;
const int D = 2;
struct Matrix
{
    long long m[D][D];
};
Matrix X,A;
void ZeroMatrix(Matrix& M)
{
    for(int i=0;i<D;++i)
        for(int j=0;j<D;++j)
            M.m[i][j] = 0;
}
void init()
{
    A.m[0][0] = 1;A.m[0][1] = 1;
    A.m[1][0] = 1;A.m[1][1] = 0;
    ZeroMatrix(X);
    X.m[0][0] = 1;
}
Matrix mul(const Matrix& M,const Matrix& N)
{
    Matrix T;
    T.m[0][0] = (M.m[0][0]*N.m[0][0]+M.m[0][1]*N.m[1][0])%modu;
    T.m[0][1] = (M.m[0][0]*N.m[0][1]+M.m[0][1]*N.m[1][1])%modu;
    T.m[1][0] = (M.m[1][0]*N.m[0][0]+M.m[1][1]*N.m[1][0])%modu;
    T.m[1][1] = (M.m[1][0]*N.m[0][1]+M.m[1][1]*N.m[1][1])%modu;
    return T;
}
void fastPow(long long n)
{
    while(n!=0)
    {
        if(n&1)
            X = mul(X,A);
        n = n>>1;
        A = mul(A,A);
    }
}
int strToInt(string str)
{
    int len = str.length();
    int num = 0;
    int factor = 1;
    for(int i=len-1;i>=0;--i)
    {
        if(str[i]-'0'<0||str[i]-'0'>9)
        {
            cout<<"error"<<endl;
            cout<<"----------"<<endl;
            return -1;
        }
        else
        {
            num += factor*(str[i]-'0');
            factor *= 10;
        }
    }
    return num;
}

long long getFib(long long n)
{
    if(n==0)
        return 0;
    if(n==1)
        return 1;
    init();
    fastPow(n-1);

    return X.m[0][0];
}
int main()
{
    string str;
    while(cin>>str)
    {
        long long n;
        if(str.compare("exit")==0)
            break;
        int t = strToInt(str);
        if(t==-1)
            continue;
        if(t>2)
        {
            cout<<"none"<<endl;
            cout<<"----------"<<endl;
            continue;
        }
        if(t==1)
        {
            cin>>n;
            cout<<getFib(n)<<endl;
            cout<<"----------"<<endl;
            continue;
        }
        if(t==2)
        {
            int t = 1;
            for(int i=1;i<=10;++i)
            {
                cout<<getFib(t)<<endl;
                t = t*10;

            }
            cout<<"----------"<<endl;
        }
    }
    return 0;
}

 

最后欢迎大家访问我的个人网站: 1024s