矩阵快速幂模板+例题

矩阵快速幂这个算法,理解起来很容易,但是我之前自己写的代码有bug,也是因为上课不听课,对形参和实参没理解,平常用的都是全局变量,是不是实参影响不大,这次定义一个结构体的矩阵,矩阵需要初始化为0,然后,因为形参和实参没怎么理解,导致输出的答案差异很大,前提是矩阵快速幂,矩阵需要初始化数组,并不会默认为0

重要的事情讲三遍:

矩阵需要初始化,并不会默认为0!
矩阵需要初始化,并不会默认为0!
矩阵需要初始化,并不会默认为0!

例题:Fibonacci

AC代码:

#include<iostream>
#include<cstring>
using namespace std;
const int mod=1e4;
struct node
{
    int a[2][2];//矩阵
};
void mes(node&d)
{
    for(int i=0; i<2; i++)
    {
        for(int j=0; j<2; j++)
        {
            d.a[i][j]=0;
        }
    }
}
void init(node &res)
{
    mes(res);
    for(int i=0; i<2; i++)
        res.a[i][i]=1;
}
node mul(node b,node c,int mod)
{
    node d;
    mes(d);
    for(int i=0; i<2; i++)
    {
        for(int j=0; j<2; j++)
        {
            for(int k=0; k<2; k++)
            {
                d.a[i][j]=(d.a[i][j]+(b.a[i][k]*c.a[k][j])%mod)%mod;
            }
        }
    }
    return d;
}
node quick(node a,int b)
{
    node res;
    init(res);
    while(b)
    {
        if(b&1)
            res=mul(res,a,mod);
        a=mul(a,a,mod);
        b>>=1;
    }
    return res;
}
int main()
{
    int n;
    while(cin>>n&&(n!=-1))
    {
        node b;
        b.a[0][0]=b.a[0][1]=b.a[1][0]=1;
        b.a[1][1]=0;
        node res=quick(b,n);
        cout<<res.a[0][1]<<endl;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值