The 2019 Asia Nanchang First Round Online Programming Contest-H-The Nth Item

矩阵快速幂的应用
由关系式有:
F (n) =3 *F(n - 1) + 2 * F(n - 2)
F(n - 1) =1 * F(n - 1) + 0 * F(n - 2)
则选取矩阵(b):
b[1][1] = 3;
b[1][2] = 2;
b[2][1] = 1;
b[2][2] = 0;

代码:

    #include<bits/stdc++.h>
    using namespace std;
    const int mod = 998244353;
    typedef long long ll;
    struct mat {
       ll a[3][3];
    } ans, base;
    void init() {
        memset(ans.a, 0, sizeof(ans.a));
        ans.a[1][1] = 1;
        memset(base.a, 0, sizeof(base.a));
        base.a[1][1] = 3;
        base.a[1][2] = 2;
        base.a[2][1] = 1;
     } //初始化矩阵
     mat yunsuan(mat j1, mat j2) {
        mat m;
        m.a[1][1] = ((j1.a[1][1]%mod)*(j2.a[1][1]%mod)%mod+(j1.a[1][2]%mod)*(j2.a[2][1]%mod)%mod)%mod;
        m.a[1][2] = ((j1.a[1][1]%mod)*(j2.a[1][2]%mod)%mod+(j1.a[1][2]%mod)*(j2.a[2][2]%mod)%mod)%mod;
        m.a[2][1] = ((j1.a[2][1]%mod)*(j2.a[1][1]%mod)%mod+(j1.a[2][2]%mod)*(j2.a[2][1]%mod)%mod)%mod;
        m.a[2][2] = ((j1.a[2][1]%mod)*(j2.a[1][2]%mod)%mod+(j1.a[2][2]%mod)*(j2.a[2][2]%mod)%mod)%mod;
        return m;
     } //矩阵乘法
    map<ll,ll>pp;
    ll ksm(ll n) {
        if(pp.count(n)) return pp[n]; //已经计算过则直接返回值
        init(); //别忘记 忘了就是个WA
        ll num = n;
        while(n) {
	       if(n & 1) ans = yunsuan(ans, base);
	       base = yunsuan(base, base);
	       n >>= 1; 
         }
         return pp[num] = ans.a[1][1] % mod; //用map记录
    }
    const int maxn = 1e7 + 5;
    ll rec[maxn];
    int main() {
        ll Q,N,num,q;
        cin >> Q >> N;
        q = N;
        for(int i=0; i<Q; i++){
              num = ksm(q - 1);
              rec[i] = num;
              q ^= (num * num);
         }
         q = rec[0];
          for(int i = 1; i < Q; i ++)
              q ^= rec[i];
          cout << q << endl;
          return 0;
     }

总结:
矩阵快速幂的程序是固定的模板,变化的只是矩阵,即上面的ans和base;
所以我们只需要 由递归式得到base(相当于Aa^b中的a),由初值得到ans(Aa ^ b中的A);

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值