等比数列二分求和

最新推荐文章于 2019-11-24 17:13:32 发布

原创最新推荐文章于 2019-11-24 17:13:32 发布 · 636 阅读

1 ·

CC 4.0 BY-SA版权

------矩阵快速幂专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用矩阵快速幂算法高效求解矩阵求和问题的方法。通过递归分解求和范围并利用矩阵乘法及快速幂技巧加速计算过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天我们学习如何有效地求表达式的值。对于这个问题，用二分解决比较好。

（1）当时，

（2）当时，那么有

（3）当时，那么有

题目：http://poj.org/problem?id=3233

题意：矩阵求和

#include <iostream>
#include <string.h>
#include <stdio.h>

using namespace std;
const int N = 35;

struct Matrix
{
    int m[N][N];
};

Matrix I;
int n,k,M;

Matrix add(Matrix a,Matrix b)
{
    Matrix c;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            c.m[i][j] = a.m[i][j] + b.m[i][j];
            c.m[i][j] %= M;
        }
    }
    return c;
}

Matrix multi(Matrix a,Matrix b)
{
    Matrix c;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            c.m[i][j] = 0;
            for(int k=0; k<n; k++)
                c.m[i][j] += a.m[i][k] * b.m[k][j];
            c.m[i][j] %= M;
        }
    }
    return c;
}

Matrix power(Matrix A,int n)
{
    Matrix ans = I,p = A;
    while(n)
    {
        if(n & 1)
        {
            ans = multi(ans,p);
            n--;
        }
        n >>= 1;
        p = multi(p,p);
    }
    return ans;
}

Matrix sum(Matrix A,int k)
{
    if(k == 1) return A;
    Matrix t = sum(A,k/2);
    if(k & 1)
    {
        Matrix cur = power(A,k/2+1);
        t = add(t,multi(t,cur));
        t = add(t,cur);
    }
    else
    {
        Matrix cur = power(A,k/2);
        t = add(t,multi(t,cur));
    }
    return t;
}

int main()
{
    while(scanf("%d%d%d",&n,&k,&M)!=EOF)
    {
        Matrix A;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                scanf("%d",&A.m[i][j]);
                A.m[i][j] %= M;
                I.m[i][j] = (i==j);
            }
        }
        Matrix ans = sum(A,k);
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
                printf("%d ",ans.m[i][j]);
            puts("");
        }
    }
    return 0;
}