十进制矩阵快速幂牛客暑期多校第五场B题

https://ac.nowcoder.com/acm/contest/885/B
这是题目链接,就是简单的矩阵快速幂模板题,唯一特殊的就是那个n是个大数,但是他是个10的10的六次幂,只能用字符串来输入,其实赛后一看题解,十进制快速幂也就那样很简单的,举个例子
就是把n拆分成10的几次幂,然后就加一个单位矩阵,依次迭代承上就可以啦。

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 5;
ll mod;
ll a,b,x1,x0;
char n[N];
struct matrix{
    ll a[2][2];
};
matrix Matrix(matrix a,matrix b)
{
    matrix c;
    for(int i = 0;i < 2;i++)
    {
        for(int j = 0;j < 2;j++)
        {
            c.a[i][j] = 0;
        }
    }
    for(int i = 0;i < 2;i++)
    {
        for(int j = 0;j < 2;j++)
        {
            for(int k = 0;k < 2;k++)
            {
                c.a[i][k] = (c.a[i][k] + a.a[i][j] * b.a[j][k]) % mod;
            }
        }
    }
    return c;
}
matrix quick_pow(matrix a,ll x)
{
    matrix A;
    memset(A.a,0,sizeof(A.a));
    for(int i = 0;i < 2;i++)
    {
        A.a[i][i] = 1;
    }
    while(x)
    {
        if(x&1)
        {
            A = Matrix(A,a);
        }
        a = Matrix(a,a);
        x/=2;
    }
    return A;
}
int main()
{
    scanf("%lld%lld%lld%lld",&x0,&x1,&a,&b);
    scanf("%s%lld",n,&mod);
    matrix ans;
    ans.a[0][0] = a;ans.a[0][1] = b;
    ans.a[1][0] = 1;ans.a[1][1] = 0;
    matrix ret;
    ret.a[0][1] = 0;ret.a[1][0] = 0;
    ret.a[0][0] = 1;ret.a[1][1] = 1;
    for(int l = strlen(n) - 1;l >= 0;l--)
    {
        ret = Matrix(ret,quick_pow(ans,n[l] - '0'));
        ans = quick_pow(ans,10);
    }
    printf("%lld\n",(1ll*ret.a[1][0] * x1 % mod + 1ll*ret.a[1][1] * x0 % mod)%mod);//输出的是第n-1项。
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值