51nod 1033 骨牌覆盖 V2

题意：

1033 骨牌覆盖 V2 

基准时间限制：1 秒 空间限制：131072 KB 分值: 320  难度：7级算法题

 收藏

 关注

在m*n的一个长方形方格中，用一个1*2的骨牌排满方格。问有多少种不同的排列方法。（n <= 5)

例如：3 * 2的方格，共有3种不同的排法。（由于方案的数量巨大，只输出 Mod 10^9 + 7 的结果）

Input

2个数M N，中间用空格分隔（2 <= m <= 10^9，2 <= n <= 5）

Output

输出数量 Mod 10^9 + 7

Input示例

2 3

Output示例

3

sol:

不用维护联通分量信息，只需要维护轮廓线上的m个格子是否填充。

由于n很大，必须使用矩阵快速幂。

可以看到按行转移形式是固定的，与具体第几行无关，所以可以求出转移一整行后的转移矩阵，然后进行快速幂。

矩阵中某个状态k的贡献可以让其为1，其他状态初值为0，逐格转移之后各状态h计数的值就是转移矩阵 C(k,h)的值了。

code：

 

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cstdio>

using namespace std;

const int maxn = 1 << 5;
typedef long long ll;
const int mod = 1e9+7;

using namespace std;

ll dp[6][maxn];

typedef vector<ll> vec;
typedef vector<vec> mat;

mat mul(mat &A, mat &B)
{
    mat C(A.size(), vec(B[0].size()));
    for (int i = 0; i < A.size(); i++)
    {
        for (int k = 0; k < B.size(); k++)
            for (int j = 0; j < B[0].size(); j++)
            {
                C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % mod;
            }
    }
    return C;
}

mat pow(mat A, ll n)
{
    mat B(A.size(), vec(A.size()));
    for (int i = 0; i < A.size(); i++)
        B[i][i] = 1;

    while (n)
    {
        if (n & 1)
            B = mul(B, A);
        A = mul(A, A);
        n >>= 1;
    }
    return B;
}

int main()
{
    ll n, m;
    scanf("%lld%lld", &n, &m);
    int maxs = 1 << m;
    mat C(maxs, vec(maxs));
    for (int g = 0; g < maxs; g++)
    {
        memset(dp, 0, sizeof(dp));
        dp[0][g] = 1;
        for (int j = 0; j < m; j++)
        {
            for (int k = 0; k < maxs; k++)
            {
                int s_u = 1 << j;
                int p_u = k & s_u;
                if (j == 0)
                {
                    if (p_u == 0)
                        dp[j + 1][k + s_u] = (dp[j+1][k+s_u]+dp[j][k])%mod;
                    else
                        dp[j + 1][k - s_u] = (dp[j+1][k-s_u]+dp[j][k])%mod;
                    continue;
                }
                int s_l = 1 << (j - 1);
                int p_l = k & s_l;
                if (p_u == 0 && p_l == 0)
                {
                    dp[j+1][k+s_u]=(dp[j+1][k+s_u]+dp[j][k])%mod;
                }
                else if (p_u > 0 && p_l > 0)
                {
                    dp[j+1][k-s_u]=(dp[j+1][k-s_u]+dp[j][k])%mod;    
                }
                else if (p_u > 0 && p_l == 0)
                {
                    dp[j+1][k+s_l]=(dp[j+1][k+s_l]+dp[j][k])%mod;
                    dp[j+1][k-s_u]=(dp[j+1][k-s_u]+dp[j][k])%mod;
                }
                else if (p_u == 0 && p_l > 0)
                {
                    dp[j+1][k+s_u]=(dp[j+1][k+s_u]+dp[j][k])%mod;
                }
            }
        }
        for (int i = 0; i < maxs; i++)
        {
            C[g][i] = dp[m][i];
        }
    }
    mat A(1, vec(maxs));
    A[0][maxs - 1] = 1;
    C = pow(C, n);
    A = mul(A, C);
    ll ans = 0;
    ans=A[0][maxs-1];
    printf("%lld\n",ans);
    return 0;
}