矩阵快速幂:数列考验

题面:

给定一个新的数列 f(x) 来考一考大家。数列 f(x) 定义如下:、

当 x < 10 时,f(x) = x;
当 x ≥ 10 时,f(x) = a0 * f(x-1) + a1 * f(x-2) + a2 * f(x-3) + …… + a9 * f(x-10),ai 只能为 0 或 1。
现在给定 a0~a9,以及两个正整数 k m,询问 f(k) % m 的数值大小

输出文件包含多组测试用例,每组测试用例格式如下:
第一行给定两个正整数 k m。(k < 2e9, m < 1e5)
第二行给定十个整数,分别表示 a0~a9

对于每一组测试用例输出一行,表示 f(k) % m 的数值大小

sample intput:
10 9999
1 1 1 1 1 1 1 1 1 1
20 500
1 0 1 0 1 0 1 0 1 0

sample output:
45
104

思路:

因为数据量比较大,所以使用矩阵快速幂来求解
根据题目要求,只要根据输入初始化原始矩阵并且 k-9 的快速幂操作并最终得出要求的f %m即可

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int n=10;
int k,m;
struct matrix
{
 int a[10][10];
 matrix()
 {
  memset(a,0,sizeof(a));
 }
 matrix(matrix& ma)
 {
  memcpy(a,ma.a,sizeof(a));
 }
 matrix operator*(const matrix& ma) const
 {
  matrix temp;
  for(int i=0;i<10;i++)
  {
   for(int j=0;j<10;j++)
   {
    temp.a[i][j]=0;
    for(int k=0;k<10;k++)
    {
     temp.a[i][j]+=a[i][k]*ma.a[k][j]%m;
     temp.a[i][j]%=m;
    }
   }
  }
  return temp;
 }
 static matrix quick_pow(matrix a,int x)
 { 
 matrix temp;
 for(int i=0;i<10;i++)
  temp.a[i][i]=1;
 while(x)
 {
  if(x&1) temp=temp*a;
  a=a*a;
  x>>=1;
 }
 return temp;
 }
};
int main()
{
 while(scanf("%d%d",&k,&m)!=EOF)
 {
  int ans=0;
  matrix mat;
  for(int i=0;i<10;i++)
   cin>>mat.a[0][i];
  for(int i=1;i<10;i++)
   mat.a[i][i-1]=1;
  if(k<10)
   ans=k%m;
  else if(k>=10)
  {
   mat=matrix::quick_pow(mat,k-9);
   for(int i=0;i<10;i++)
   {
    ans+=(mat.a[0][i]*(9-i))%m;
    ans%=m;
   }  
  }
  printf("%d\n",ans);
 }
 return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值