【HDOJ 1021】 Fibonacci Again (矩阵快速幂(裸))

来源：互联网发布：拜特软件编辑：程序博客网时间：2024/05/19 11:50

【HDOJ 1021】 Fibonacci Again

没错我又来发水题了。。。感觉做矩快会上瘾。。。这种原本绝壁超时的东西突然秒秒钟出来。。。这感觉。。。

爽上代码

#include <iostream>#include <cstdio>#include <cstring>using namespace std;typedef struct Matrix Matrix;struct Matrix{    int mx[2][2];    void Out()    {        for(int i = 0; i < 2; ++i)        {            for(int j = 0; j < 2; ++j)                printf("%d ",mx[i][j]);            puts("");        }    }    void Init()    {        mx[0][0] = mx[0][1] = mx[1][0] = 1;        mx[1][1] = 0;    }    void Emp()    {        mx[0][0] = mx[1][1] = 1;        mx[0][1] = mx[1][0] = 0;    }    Matrix operator * (const Matrix a)const    {        Matrix x;        memset(x.mx,0,sizeof(x.mx));        for(int i = 0; i < 2; ++i)            for(int j = 0; j < 2; ++j)                for(int k = 0; k < 2; ++k)                    x.mx[i][j] = (x.mx[i][j]+mx[i][k]*a.mx[k][j])%3;        return x;    }};bool pow(int n){    Matrix a,b;    a.Emp();    b.Init();    while(n)    {        if(n&1) a = a*b;        b = b*b;        n >>= 1;    }    return !((a.mx[0][0]*7+a.mx[0][1])%3);}int main(){    int n;    while(~scanf("%d",&n))    {        printf("%s\n",pow(n)? "yes": "no");    }    return 0;}

0 0