[Power] rapid matrix multiplication

Quick Start Video: av56433157

1> p1926 Fibonacci

#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
long long n;
const int mod=1000000007;
long long nw[2][2],ans[2][2];
long long t[2][2];

void mul1()
{
    memset(t,0,sizeof(t));
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            for(int k=0;k<2;k++)
                t[i][j]+=(ans[i][k]*nw[j][k])%mod;
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            ans[i][j]=t[i][j];
}
void mul2()
{
    memset(t,0,sizeof(t));
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            for(int k=0;k<2;k++)
                t[i][j]+=(nw[i][k]*nw[j][k])%mod;
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            nw[i][j]=t[i][j];
}
void mul3()
{
    long long as=0;
    for(int i=0;i<2;i++) as=(as+ans[0][i])%mod;
    printf("%lld\n",as);
}

int main()
{
    scanf("%lld",&n);
    n-=2;
    if(n<=0) 
    {
        printf("1\n");
        return 0;
    }
    
    nw[0][0]=nw[1][0]=nw[0][1]=1;
    ans[0][0]=ans[1][1]=1;
    while(n)
    {
        if(n&1) mul1();
        n>>=1,mul2();
    }
    mul3();
    
    return 0;
}

2> generalized Fibonacci number

 

 

3> ch30 placed flowers

Fast matrix power accelerated dp, optimization of time complexity and space complexity

60->100

#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
int n,m;
const int N=103,mod=1e9+7;
long long nw[N][N],ans[N],t[N][N],tt[N];

void mul1()
{
    memset(tt,0,sizeof(tt));
    for(int i=0;i<=n;i++)
            for(int k=0;k<=n;k++)
                tt[i]=(tt[i] +nw[i][k]*ans[k]%mod) %mod;
    for(int i=0;i<=n;i++)
            ans[i]=tt[i];
}
void mul2()
{
    memset(t,0,sizeof(t));
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            for(int k=0;k<=n;k++)
                t[i][j]=(t[i][j] +nw[i][k]*nw[k][j]%mod)%mod;
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            nw[i][j]=t[i][j];
}
void mul3()
{
    long long as=0;
    for(int i=0;i<=n;i++) as=(as+ans[i])%mod;
    printf("%lld\n",as);
}

int main()
{
    scanf("%d%d",&m,&n);
    int x;
    for(int i=0;i<=n;i++) nw[i][0]=nw[0][i]=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            scanf("%1d",&x);
            nw[i][j]=x^1;
        }
    
    ans[0]=1;
    while(m)
    {
        if(m&1) mul1();
        m>>=1,mul2();
    }
    mul3();
    
    return 0;
}

 

Guess you like

Origin www.cnblogs.com/xwww666666/p/11519655.html