The Little Architect----矩阵求幂

最新推荐文章于 2020-07-04 21:24:40 发布

wk476855

最新推荐文章于 2020-07-04 21:24:40 发布

阅读量608

点赞数

文章标签： struct

本文链接：https://blog.csdn.net/wk476855/article/details/7874618

版权

题目：http://acm.hdu.edu.cn/showproblem.php?pid=3493

递推方程：F(n)=5F(n-1) - 7F(n-2) + 4F(n-3).

源代码：

#include <stdio.h>
#define M 9997

int a[10],n;

typedef  struct
{
    int ma[5][5];
}node;

node re;

void init(node &s,node &b)
{
   for(int i=0;i<3;i++)
      for(int j=0;j<3;j++)
       {
           s.ma[i][j]=0;
           b.ma[i][j]=0;
       }
    s.ma[0][0]=1;
    s.ma[1][1]=1;  s.ma[2][2]=1;
    b.ma[0][0]=5;  b.ma[0][1]=-7;  b.ma[0][2]=4;  b.ma[1][0]=1; b.ma[2][1]=1;
}

node multiply(node s,node t)
{
    node r;
    for(int i=0;i<3;i++)
      for(int j=0;j<3;j++)
       r.ma[i][j]=0;

    for(int i=0;i<3;i++)
      for(int j=0;j<3;j++)
       for(int k=0;k<3;k++)
         r.ma[i][j]=(r.ma[i][j]+s.ma[i][k]*t.ma[k][j])%M;
     return r;
}

node quickPow(int x)
{
  node s,b;
  init(s,b);
  while(x>0)
  {
      if(x%2==1)  s=multiply(s,b);
      x=x/2;
      b=multiply(b,b);
  }
  return s;
}

int main()
{
   while(true)
   {
       scanf("%d",&n);
       a[1]=1; a[2]=2;  a[3]=6; a[4]=19;
       if(n==0)  break;
       if(n<=4)  {
            printf("%d\n",a[n]);
             continue; }
       re=quickPow(n-4);
       printf("%d\n",((re.ma[0][0]*a[4]+re.ma[0][1]*a[3]+re.ma[0][2]*a[2])%M+M)%M);
   }
}