vijosP1194 Domino

vijosP1194 Domino

链接：https://vijos.org/p/1194

【思路】

矩阵相乘。

参考Matrix67的文章：

【代码】

 #include<cstdio>

 #include<cstring>

 #include<iostream>

 #define FOR(a,b,c) for(int a=(b);a<(c);a++)

 using namespace std;

 const int maxn = ;

 const int ag[]={,,,,,,,};

 typedef long long LL;

 int n,m,MOD;

 struct Matrix{

     int r,c;

     LL N[maxn][maxn];

     void init(int r,int c) {

         this->r=r, this->c=c;

         memset(N,,sizeof(N));

     }

     Matrix operator*(Matrix& B)const{

         Matrix A=*this,C;

         C.init(A.r,B.c);

         for(int i=;i<C.r;i++)

            for(int j=;j<C.c;j++)

               for(int k=;k<A.c;k++)

                  C.N[i][j] = (C.N[i][j]+A.N[i][k]*B.N[k][j])%MOD;

         return C;

     }

     Matrix pow(int p) {

         Matrix tmp=*this;

         Matrix ans;

         ans.init(r,r);

         for(int i=;i<r;i++) ans.N[i][i]=;

         while(p) {

             if(p&) ans=ans*tmp;

             tmp=tmp*tmp;

             p>>=;

         }

         return ans;

     }

 };

 int main() {

     scanf("%d%d%d",&n,&m,&MOD);

     Matrix ans;

     int all=<<m;

     ans.init(all,all);

     for(int i=;i<all;i++)

        for(int j=;j<all;j++)

            if( ((~i)&j) == ((~i)&(all-))) {

                   int l=;

                   for(int r=;r<=;r++) l=l||((i&j)==ag[r]);

                   ans.N[i][j]=l;

            }

     ans=ans.pow(n);

     printf("%d",ans.N[all-][all-]);

     return ;

 }