NEUQ 1768 一道简单的递推题（矩阵快速幂）

最新推荐文章于 2022-10-30 23:22:34 发布

信鸦

最新推荐文章于 2022-10-30 23:22:34 发布

阅读量343

点赞数

分类专栏：矩阵快速幂

本文链接：https://blog.csdn.net/NEET_Champloo/article/details/73135569

版权

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

4 篇文章 0 订阅

订阅专栏

问题 F: 一道简单的递推题

时间限制: 1 Sec 内存限制: 128 MB
提交: 735 解决: 94
[ 提交][ 状态][ 讨论版]

题目描述

存在如下递推式:

F(n+1)=A1*F(n)+A2*F(n-1)+...+An*F(1)

求第K项的值对1000000007取模的结果

输入

单组测试数据

第一行输入两个整数 n , k (1<=n<=100,n<k<=10000000000)

第二行输入 n 个整数 F(1) F(2) ... F(n)

第三行输入 n 个整数A1 A2 ... An

输出

输出一个整数

样例输入

2   3
1   2
3   4

样例输出

提示

[ 提交][ 状态][ 讨论版] 对 F(n+1)=A1*F(1)+A2*F(2)+...+An*F(n) 的式子可以这样构造矩阵

这题将最后一排反过来即可

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

const int N = 100 + 10;
const int Mod = 1000000007;
long long n, k;
long long a[N], f[N];
struct xx{
    long long a[N][N];
} ori, res;

xx mul(xx x, xx y){
    xx temp;
    memset(temp.a, 0, sizeof(temp.a));
    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            for(int k = 0; k < n; k++){
                if(!x.a[i][k]) continue;
                temp.a[i][j] += x.a[i][k]*y.a[k][j]+Mod;
                temp.a[i][j] %= Mod;
            }
        }
    }
    return temp;
}

void init(){
    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            if(i == j && i < n-1){
                 ori.a[i][j+1] = 1;
            }
        }
    }
    for(int j = 0; j < n; j++){
        ori.a[n-1][j] = a[n-j-1];
    }
    memset(res.a, 0, sizeof(res.a));
    for(int i = 0; i < n; i++){
        res.a[i][0] = f[i];
    }
}

void calc(long long k){
    while(k){
        if(k&1){
            res = mul(ori, res);
        }
        ori = mul(ori, ori);
        k >>= 1;
    }
    printf("%lld\n", res.a[n-1][0]);
}

int main(){
    while(scanf("%lld%lld", &n, &k) == 2){
        for(int i = 0; i < n; i++){
            scanf("%lld", &f[i]);
        }
        for(int i = 0; i < n; i++){
            scanf("%lld", &a[i]);
        }
        init();
        calc(k-n);
    }
}