距阵的运用

昨晚大四的一个学长给我们讲了，距阵的使用，很不错，这里总结一下。

一般距阵，用在n很大，导致超时的问题上。它需要快速幂取模的辅助。

例子

求斐波那契数列   a₁=1；    a₂=1；     a_n=a_n-1+a_{n-2；   求 an     （n>=1&&n<=10⁸）    由于n很大结果对999999999取模；}

对于这题n很大你直接推，每次取%999999999，明显会超时，

矩阵做法

这样就可以化为 矩阵（a1 a2）*矩阵（A^n-1）= 矩阵（an，an+1）；

看代码

#include<stdio.h>
#include<string.h>
#define mod 999999999

__int64 a[5];
__int64 b[3][3];

__int64 quick_mod(__int64 n)
{
a[1]=1; a[2]=1;
b[1][1]=0; b[1][2]=1;
b[2][1]=1; b[2][2]=1;
while(n)
{
if(n&1)
{
__int64 s1,s2;
s1=(a[1]*b[1][1]%mod+a[2]*b[2][1]%mod)%mod;
s2=(a[1]*b[2][1]%mod+a[2]*b[2][2]%mod)%mod;
a[1]=s1;
a[2]=s2;
n--;
}
n=n/2;
__int64 k1,k2,k3,k4;
k1=(b[1][1]*b[1][1]%mod+b[1][2]*b[2][1]%mod)%mod;
k2=(b[1][1]*b[2][1]%mod+b[1][2]*b[2][2]%mod)%mod;
k3=(b[2][1]*b[1][1]%mod+b[2][2]*b[2][1]%mod)%mod;
k4=(b[2][1]*b[1][2]%mod+b[2][2]*b[2][2]%mod)%mod;
b[1][1]=k1; b[1][2]=k2; b[2][1]=k3; b[2][2]=k4;
}
return a[1];
}

int main(void)
{
__int64 n;
while(scanf("%I64d",&n)==1)
{
printf("%I64d ",quick_mod(n-1));
}
return 0;
}

这样就运用快速幂取模大大减少了时间。

如果矩阵大一点的话这样写

#include<stdio.h>
#include<string.h>
#define mod 999999999

__int64 a[5];
__int64 b[3][3];

__int64 quick_mod(__int64 n)
{
__int64 i,j,k;
a[1]=1; a[2]=1;
b[1][1]=0; b[1][2]=1;
b[2][1]=1; b[2][2]=1;
while(n)
{
if(n&1)
{
__int64 s[3];
memset(s,0,sizeof(s));
for(i=1;i<=2;i++)
{
for(j=1;j<=2;j++)
{
s[i]=(s[i]+a[j]*b[j][i])%mod;
}
}
for(i=1;i<=2;i++)
a[i]=s[i];
n--;
}
n=n/2;
__int64 kk[5][5];
memset(kk,0,sizeof(kk));
for(i=1;i<=2;i++)
{
for(j=1;j<=2;j++)
{
for(k=1;k<=2;k++)
{
kk[i][j]=(kk[i][j]+b[i][k]*b[k][j])%mod;
}
}
}
for(i=1;i<=2;i++)
{
for(j=1;j<=2;j++)
{
b[i][j]=kk[i][j];
}
}
}
return a[1];
}

int main(void)
{
__int64 n;
while(scanf("%I64d",&n)==1)
{
printf("%I64d ",quick_mod(n-1));
}
return 0;
}