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 = 40;
const int ag[]={0,3,6,12,15,24,27,30};
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,0,sizeof(N));
    }
    Matrix operator*(Matrix& B)const{
        Matrix A=*this,C;
        C.init(A.r,B.c);
        for(int i=0;i<C.r;i++)
           for(int j=0;j<C.c;j++)
              for(int k=0;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=0;i<r;i++) ans.N[i][i]=1;
        while(p) {
            if(p&1) ans=ans*tmp;
            tmp=tmp*tmp;
            p>>=1;
        }
        return ans;
    }
};
int main() {
    scanf("%d%d%d",&n,&m,&MOD);
    Matrix ans;
    int all=1<<m;
    ans.init(all,all);
    for(int i=0;i<all;i++)    
       for(int j=0;j<all;j++)
           if( ((~i)&j) == ((~i)&(all-1))) {
                  int l=0;
                  for(int r=0;r<=8;r++) l=l||((i&j)==ag[r]);
                  ans.N[i][j]=l;
           }
    ans=ans.pow(n);
    printf("%d",ans.N[all-1][all-1]);
    return 0;
}