hiho 43 骨牌覆盖问题·三

问题

http://hihocoder.com/contest/hiho43/problem/1

代码

程序递归查找状态转移矩阵。

#include <bits/stdc++.h>
using namespace std;
enum {maxk = 7, max2k = 1<<maxk, mod=12357};
int M[32][max2k][max2k];
int d[max2k][max2k];
int t[max2k][max2k];
int msize;
int width;
void mul(int c[max2k][max2k], int a[max2k][max2k], int b[max2k][max2k])
{
    for(int i=0; i<msize; ++i)
    {
        for(int j=0; j<msize; ++j)
        {
            t[i][j]=0;
            for (int k=0; k<msize; ++k)
            {
                t[i][j] = (t[i][j] + a[i][k]*b[k][j])%mod;
            }
        }
    }
    memcpy(c, t, sizeof(t));
}
void setId(int a[max2k][max2k])
{
    memset(a, 0, sizeof(a));
    for (int i=0; i<msize; ++i)
        a[i][i]=1;
}
void dfs(int x, int y, int col)
{
    if (col ==width)
    {
        M[0][y][x]=1;
        return ;
    }

    dfs(x<<1, y<<1|1, col+1);
    dfs(x<<1|1, y<<1, col+1);
    if (col+2<=width)
    {
        dfs(x<<2|3, y<<2|3, col+2);
    }
}
int main()
{
    int n;
    scanf("%d %d", &width, &n);
    msize= 1<<width;
    memset(M, 0, sizeof(M[0]));
    dfs(0, 0, 0);
    for(int i=1; i<32; ++i)
        mul(M[i], M[i-1], M[i-1]);
    setId(d);
    for (int i=0; i<32 && n; ++i, n>>=1)
    {
        if (n&0x01)
        {
            mul(d, d, M[i]);
        }
    }
    printf("%d\n", d[msize-1][msize-1]);
    return 0;
}