pku 3150(修改后的矩阵乘法)

起初直接有矩阵乘法做,结果TLE。后来参考网上代码,发现他们巧妙利用转移矩阵的性质。从而降低复杂度。

转移矩阵的下一列是前一列的下移而来。因次计算出第一列,其余的就可推出。

code:

#include<stdio.h>
#include<string.h>

int n,m,k,d;
__int64 ans[510]={0};
__int64 E[501];

int multiply(__int64 (*A),__int64 (*B))
{  
 int i,j,t;
    __int64 C[502]={0};
 for(i=0;i<n;i++)
 {  
  j=(n-i)%n;
  for(t=0;t<n;t++)
  {  
      C[i]+=A[t]*B[j];
   j=(j+1)%n;
  }
 }
 for(i=0;i<n;i++)  
   A[i]=C[i]%m; 

 return 1;
}
int erfen()
{
   while(k)
   {
    if(k&1)multiply(ans,E);
    multiply(E,E);
    k>>=1;
   }
   return 0;
}
int main()
{
    int i,cnt=0;
 while(EOF!=scanf("%d%d%d%d",&n,&m,&d,&k))
 {     
        for(i=0;i<n;i++)
   scanf("%I64d",&ans[i]);
  memset(E,0,sizeof(E));
  for(E[0]=i=1;i<=d;i++)
  { 
   E[i]=E[n-i]=1;
  }
        erfen();
  for(i=0;i<n;i++)
   printf("%I64d ",ans[i]);
  printf("/n");
 }
 return 0;
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值