矩阵快速幂 (模板)

另斐波那契数列的矩阵为 1 1

                    1 0

/*定义矩阵<pre name="code" class="cpp"> Matrix A;
    A.clear();
    /*改*/
    A.n = A.m = 2;
    A.a[0][0] = 1;
    A.a[0][1] = 1;
    A.a[1][0] = 1;
    A.a[1][1] = 0;

接口:Matrix res = Matrix_pow(A, n - 1);
cout<<res.a[0][0]<<endl;
*/
#include <cstdio>
#include <cstring>

typedef long long ll;
using namespace std;
/*改*/
const int maxn = 5;
const int maxm = 5;
const int mod = 10000;
struct Matrix
{
    int n, m;
    ll a[maxn][maxm];
    void clear()
    {
        n = m = 0;
        memset(a, 0, sizeof(a));
    }
    Matrix operator * (const Matrix &b) const
    {
        Matrix tmp;
        tmp.clear();
        tmp.n = n;
        tmp.m = b.m;
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++)
            {
                if (!a[i][j]) continue;  //稀疏矩阵乘法优化
                for (int k = 0; k < b.m; k++)
                {
                    tmp.a[i][k] += a[i][j] * b.a[j][k];
                    tmp.a[i][k] %= mod;
                }
            }
        return tmp;
    }
};
int n;
Matrix Matrix_pow(Matrix A, int k)
{
    Matrix res;
    res.clear();
    res.n = res.m = 2;//改
    for (int i = 0; i < 2; i++) //改
        res.a[i][i] = 1;
    while(k)
    {
        if (k & 1) res = res * A;
        k >>= 1;
        A = A * A;
    }
    return res;
}
int main ()
{

    //freopen("text.in","r",stdin);
    Matrix A;
    A.clear();
    /*改*/
    A.n = A.m = 2;
    A.a[0][0] = 1;
    A.a[0][1] = 1;
    A.a[1][0] = 1;
    A.a[1][1] = 0;
    while(1)
    {
        int n;
        scanf("%d", &n);
        if(n == -1)
            break;
        if(n == 0)
        {
            printf("0\n");
            continue;
        }
        Matrix res = Matrix_pow(A, n - 1);
        printf("%lld\n", res.a[0][0]);
    }
    return 0;
}

 




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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值