HDU1757 A Simple Math Problem (矩阵快速幂)

Tw_light

于 2020-01-18 16:31:55 发布

阅读量149

点赞数

本文链接：https://blog.csdn.net/Tw_light/article/details/104031304

版权

刷题专栏收录该内容

18 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用矩阵快速幂方法解决HDU1757编程竞赛题目，该题目涉及递归定义的函数f(x)，在x大于等于10时由前10个f(x-1)的线性组合构成，要求计算f(k)对m取模的结果。

摘要由CSDN通过智能技术生成

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1757
题目描述：函数f(x)满足如下条件：
$x < 10 时， f (x) = x;$
$x >= 10时， f(x)=a_0*f(x-1)+ a_1 * f(x-2) + ... + a_9 * f_(x-10)$
$a_i(0 <= i<=9), a_i只能为0或1$
输入k和m，输出 f(k)%m
解题方法：矩阵快速幂
code：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAXN = 1e5 + 7;
const int N = 11;
int num[N][N];
int A[N][N];
int ans[N][N];
int temp[N][N];
int mod;
void init()
{
    memset(num, 0, sizeof(num));
    memset(A, 0, sizeof(A));
}
void multi(int a[][N], int b[][N])
{
    memset(temp, 0, sizeof(temp));
    for(int i = 0; i < N; i++)
        for(int j = 0; j < N; j++)
        for(int k = 0; k < N; k++)
        temp[i][j] = (temp[i][j] + a[i][k] * b[k][j]) % mod;

    for(int i = 0; i < N; i++)
        for(int j = 0; j < N; j++)
        a[i][j] = temp[i][j];

}
void MatPow(int a[][N], int n)
{
    memset(ans, 0, sizeof(ans));
    for(int i = 0; i < N; i++) ans[i][i] = 1;

    while(n)
    {
        if(n & 1)
            multi(ans, a);
        multi(a, a);
        n >>= 1;
    }
}
int main()
{
    int k;
    while(~scanf("%d %d", &k, &mod))
    {
        init();
        for(int i = 0; i < 10; i++)
            scanf("%d", &A[0][i]);
        if(k < 10) printf("%d\n", k%mod);
        else
        {
            for(int i = 0; i < 10; i++) num[i][0] = 9 - i;

            for(int i = 1; i < 10; i++) A[i][i - 1]= 1;

            MatPow(A, k - 9);
            int res = 0;

            for(int i = 0; i < 10; i++)
            {
                res += (ans[0][i] * num[i][0]) % mod;
                res = res % mod;
            }
            cout << res << endl;
        }
    }
    return 0;
}