Gym - 101061B 简单矩阵快速幂

题意 有三种颜色的花 ，

每朵红色的花第二天会变成1朵红花2朵绿花3朵蓝花

每朵绿色的花第二天会变成4朵红花5朵绿花6朵蓝花

蓝色花第二天会变成7朵红花8朵绿花9朵蓝花

一开始红绿蓝的花各一朵 问你n天后花园里有多少多花

#include<iostream>
#include<queue>
#include<cstdio>
#include<vector>
#include<cmath>
#include<string>
#include<string.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof a)
typedef long long LL;
const LL MOD=1e9+7;
const int MAXN=100000+3;
  int a[MAXN];
  struct Matrix
  {
     LL a[4][4];//矩阵大小根据需求修改
      Matrix()
      {
          memset(a,0,sizeof(a));
      }
      void init()
      {
          for(int i=0;i<4;i++)
              for(int j=0;j<4;j++)
                  a[i][j]=(i==j);
      }
      Matrix operator + (const Matrix &B)const
      {
          Matrix C;
          for(int i=0;i<4;i++)
              for(int j=0;j<4;j++)
                  C.a[i][j]=(a[i][j]+B.a[i][j])%MOD;
          return C;
      }
      Matrix operator * (const Matrix &B)const
      {
          Matrix C;
          for(int i=0;i<4;i++)
              for(int k=0;k<4;k++)
                  for(int j=0;j<4;j++)
                      C.a[i][j]=(C.a[i][j]+1LL*a[i][k]*B.a[k][j])%MOD;
          return C;
      }
      Matrix operator ^ (const int &t)const
      {
          Matrix A=(*this),res;
          res.init();
         int p=t;
          while(p)
          {
              if(p&1)res=res*A;
              A=A*A;
             p>>=1;
       }
        return res;
    }
 };
int main()
{
    Matrix e,q;
    e.a[0][0]=1;e.a[0][1]=4;e.a[0][2]=7;
    e.a[1][0]=2;e.a[1][1]=5;e.a[1][2]=8;
    e.a[2][0]=3;e.a[2][1]=6;e.a[2][2]=9;
    int n;
    cin>>n;
    q.a[0][0]=q.a[1][0]=q.a[2][0]=1;
    while(n--)
    {
        int cnt;
        scanf("%d",&cnt);
        Matrix ss=(e^(cnt-1))*q;
        cout<<(ss.a[0][0]+ss.a[1][0]+ss.a[2][0])%MOD<<endl;
    }
}