ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[51nod 1126] 求递推序列的第N项 - 矩阵乘法

2020-02-24 22:04:45  阅读:243  来源: 互联网

标签:std long const matrix 51nod 1126 int ret 递推


#include <bits/stdc++.h>
using namespace std;

#define int long long
const int mod = 7;

struct matrix {
    int a[5][5]={};
    int n,m;
};

matrix I(int n) {
    matrix ret;
    ret.n=n;
    ret.m=n;
    for(int i=1;i<=n;i++) ret.a[i][i]=1;
    return ret;
}

matrix operator * (matrix a, matrix b) {
    matrix ret;
    ret.n = a.n;
    ret.m = b.m;
    for(int i=1;i<=ret.n;i++) {
        for(int j=1;j<=ret.m;j++) {
            for(int k=1;k<=a.m;k++) {
                ret.a[i][j] += a.a[i][k] * b.a[k][j];
                ret.a[i][j] %= mod;
                ret.a[i][j] += mod;
                ret.a[i][j] %= mod;
            }
        }
    }
    return ret;
}

matrix qpow(matrix p, int q) {
    return ((q&1)?p:I(2)) * (q?qpow(p*p,q/2):I(2));
}

signed main() {
    matrix T;
    T.n=2; T.m=2;
    int a,b,n;
    cin>>a>>b>>n;
    T.a[1][1]=a; T.a[1][2]=b;
    T.a[2][1]=1;
    T=qpow(T,n-2);
    matrix A;
    A.n=2; A.m=1;
    A.a[1][1]=1; A.a[2][1]=1;
    if(n==1) cout<<1<<endl;
    else cout<<(T*A).a[1][1];
}

标签:std,long,const,matrix,51nod,1126,int,ret,递推
来源: https://www.cnblogs.com/mollnn/p/12359066.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有