HDU 5015 233 Matrix 矩阵递推

题意:给出一个矩阵,第一行从左到右分别是233,2333,23333,2333333。。。,给出第一列的元素。其他的元素是它上面和左面的元素的和。求右下角元素的值。

思路:在比赛的时候,总是想从右下角的元素递推回去,找到公式,直接计算。发现没法高效的求和。

           这道题正确的解法是利用矩阵从第二列开始递推,每次得到下一列每个元素的值,最终的得到右下角的元素的值。

          这样,我们需要构造出一个矩阵表示和题目中给出的和的关系。同时还要递推233.

代码如下:

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
typedef long long ll;
typedef vector<ll> vec;
typedef vector<vec> mat;

const ll MOD = 1e7+7;

mat mul(mat & A, mat &B)
{
    mat C(A.size(), vec(B[0].size()));
    for (int i = 0; i < A.size(); ++i)
        for (int k = 0; k < B.size(); k++)
            for (int j = 0; j < B[0].size(); ++j)
                C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD;
    return C;
}
mat pow(mat A, ll n)
{
    mat B(A.size(), vec(A.size()));
    for (int i = 0; i < A.size(); ++i)
        B[i][i] = 1;
    while (n){
        if (n & 1) B = mul(B, A);
        A = mul(A, A);
        n >>= 1;
    }
    return B;
}

int N,M;
ll a[100005];


int main(void)
{
    while(scanf("%d %d", &N,&M) !=  EOF){
        for(int i=0; i<N; i++){
            scanf("%I64d",&a[i]);
        }
        a[N]=233; a[N+1]=3;
        mat s(N+2, vec(N+2));
        for(int i=0; i<N; i++){
            for(int j=0; j<=i; j++){
                s[i][j]=1LL;
            }
            s[i][N]=1;
        }
        s[N][N]=10; s[N][N+1]=1;
        s[N+1][N+1]=1;
        s=pow(s,M);
        ll sum=0;
        for(int i=0; i<N+2; i++){
            sum=(sum+s[N-1][i]*a[i])%MOD;
        }
        printf("%I64d\n",sum);
    }

    return 0;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值