矩阵乘法 贴个新模板

都怪自己脑残,忘记处理负数情况,WA了一上午

矩阵乘法满足结合率

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#include<math.h>
using namespace std;
#define maxn 100010
#define mod  1000000007
#define mx 4
typedef long long ll;
struct matrix
{
    int n,m;
    ll mat[4][4];
    matrix()
    {
        n=m=mx;
        memset(mat,0,sizeof(mat));
    }
    void init(int r,int c)
    {
        n=r,m=c;
        memset(mat,0,sizeof(mat));
    }
    void init2()
    {
        for(int i=0;i<mx;i++)
        for(int j=0;j<mx;j++)
         mat[i][j]=(i==j);
    }
};
matrix operator*(matrix a,matrix b)
{
    matrix ret;
    int i,j,k;
    for(i=0;i<mx;i++)
    {
     for(k=0;k<mx;k++)
     {
         if(a.mat[i][k])
         for(int j=0;j<mx;j++)
         {
          if(b.mat[k][j])
          ret.mat[i][j]=(ret.mat[i][j]+a.mat[i][k]*b.mat[k][j])%mod;
         }
     }
    }
    return ret;
}
matrix operator^(matrix a,int k)
{
    matrix ret;
    ret.init2();
    for(;k;k>>=1)
    {
        if(k&1)
        ret=ret*a;
        a=a*a;
    }
    return ret;
}
matrix res;
void init()
{
    res.mat[0][0]=3,res.mat[0][1]=1,res.mat[0][2]=-3,res.mat[0][3]=0;
    res.mat[1][0]=1,res.mat[1][1]=0,res.mat[1][2]=0,res.mat[1][3]=0;
    res.mat[2][0]=0,res.mat[2][1]=1,res.mat[2][2]=0,res.mat[2][3]=0;
    res.mat[3][0]=0,res.mat[3][1]=0,res.mat[3][2]=1,res.mat[3][3]=0;
}
int main()
{
    freopen("in.txt","r",stdin);
    ll a,b,c,d;
    while(scanf("%lld%lld%lld%lld",&a,&b,&c,&d)==4)
    {
        int ans[4];
        ans[0]=a,ans[1]=b,ans[2]=c,ans[3]=d;
        ll day,x;
        scanf("%lld%lld",&day,&x);
        init();
        matrix ret;
        ret=res^(day-1);
        ll sum,a1=0,b1=0,c1=0,d1=0;
        for(int i=0;i<mx;i++)
        {
            a1=(a1+ans[i]*ret.mat[0][i])%mod;
            b1=(b1+ans[i]*ret.mat[1][i])%mod;
            c1=(c1+ans[i]*ret.mat[2][i])%mod;
            d1=(d1+ans[i]*ret.mat[3][i])%mod;
        }
        ll x2=(x*x)%mod;
        ll x3=(x2*x)%mod;
        sum=((a1*x3)%mod+(b1*x2)%mod+(c1*x)%mod+d1)%mod;
        printf("%lld\n",(sum+mod)%mod);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值