码迷,mamicode.com
首页 > 其他好文 > 详细

POJ3070 矩阵快速幂模板

时间:2018-05-02 21:05:44      阅读:148      评论:0      收藏:0      [点我收藏+]

标签:names   cst   ons   cstring   tar   i++   模板   传引用   code   

题目:http://poj.org/problem?id=3070

矩阵快速幂模板。mod写到乘法的定义部分就行了。

别忘了 I ( ) 和 i n i t ( ) 要传引用!

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,p=10000;
struct Matrix{
    int a[2][2];
    Matrix operator *(const Matrix &b)const
    {
        Matrix c;
        c.a[0][0]=c.a[1][0]=c.a[0][1]=c.a[1][1]=0;
        for(int i=0;i<=1;i++)
            for(int j=0;j<=1;j++)
                for(int k=0;k<=1;k++)
                    (c.a[i][j]+=a[i][k]*b.a[k][j])%=p;
//                        printf("i=%d j=%d c=%d\n",i,j,b.a[i][j]);
        return c;
    }
};
Matrix r,u;
void I(Matrix &k)
{
    k.a[0][0]=k.a[1][1]=1;
    k.a[0][1]=k.a[1][0]=0;
}
void init(Matrix &k)
{
    k.a[0][0]=k.a[0][1]=k.a[1][0]=1;
    k.a[1][1]=0;
}
int main()
{
    while(1)
    {
        scanf("%d",&n);if(n==-1)return 0;
        if(!n){printf("0\n");continue;}
        n--;I(r);init(u);
        while(n)
        {
            if(n&1)r=r*u;
            n>>=1;u=u*u;
        }
        printf("%d\n",(r.a[1][0]+r.a[1][1])%p);
    }
}

 

POJ3070 矩阵快速幂模板

标签:names   cst   ons   cstring   tar   i++   模板   传引用   code   

原文地址:https://www.cnblogs.com/Narh/p/8981926.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!