等比数列取模求和公式(数字,矩阵皆可)

假设是2*2的矩阵吧,数字类比即可

代码:

struct Matrix
{
    long long ma[2][2];
};
Matrix res;
Matrix mult(Matrix A,Matrix B)
{
    Matrix C;
    memset(C.ma,0,sizeof(C.ma));
    for(int i=0; i<2; i++)
        for(int j=0; j<2; j++)
            for(int k=0; k<2; k++)
                C.ma[i][j]=(C.ma[i][j]+A.ma[i][k]*B.ma[k][j])%mod;
    return C;
}
Matrix pow_mod(Matrix A,long long n)
{
    Matrix B;
    B.ma[0][0]=B.ma[1][1]=1;
    B.ma[0][1]=B.ma[1][0]=0;
    while(n)
    {
        if(n&1) B=mult(B,A);
        A=mult(A,A);
        n>>=1;
    }
    return B;
}
Matrix add(Matrix a,Matrix b){
    for(int i=0;i<2;i++){
        for(int j=0;j<2;j++){
            a.ma[i][j]=(a.ma[i][j]+b.ma[i][j])%mod;
        }
    }
    return a;
}
Matrix cal(Matrix A,long long k) ///用二分法求矩阵和,递归实现 计算 a^1+a^2.....+a^p
{
    if(k==1)
        return A;
    else if(k&1)
        return add(cal(A,k-1),pow_mod(A,k));
    else
        return mult(cal(A,k>>1),add(pow_mod(A,k>>1),res));
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值