蓝桥杯垒骰子题解：矩阵快速幂优化动态规划

陈慈龙

1. 问题背景与需求分析

蓝桥杯作为国内知名的程序设计竞赛，其题目往往融合了数学建模与算法设计的双重挑战。2015年省赛AB组的这道"垒骰子"题目，表面看是一个排列组合问题，实则考察选手对动态规划优化和矩阵快速幂的掌握程度。

题目要求计算n个骰子垂直垒放时相邻骰子接触面数字不互斥的排列总数。具体约束条件包括：

每个骰子有6个面，数字1-6
给出m组互斥数字对（如1与2互斥）
相邻骰子接触面的数字不能是互斥对
结果需要对10^9+7取模

2. 基础解法与性能瓶颈

2.1 暴力递归解法

最直观的解法是采用DFS递归枚举所有可能的排列组合。对于每个骰子，考虑其朝上面数字与下方骰子朝下面数字是否满足非互斥条件。这种方法时间复杂度为O(6^n)，当n较大时（如题目中的n≤10^9）完全不可行。

python复制def dfs(current, top, n):
    if current == n:
        return 1
    count = 0
    for num in range(1,7):
        if not conflict[top][num]:
            count += dfs(current+1, num, n)
    return count % MOD

2.2 动态规划优化

观察到每个骰子的状态只依赖于前一个骰子的朝上面数字，可以使用DP进行优化：

python复制dp = [[0]*7 for _ in range(n+1)]
# 初始化第一个骰子
for num in range(1,7):
    dp[1][num] = 1

for i in range(2,n+1):
    for top in range(1,7):
        for num in range(1,7):
            if not conflict[top][num]:
                dp[i][num] += dp[i-1][top]
        dp[i][num] %= MOD

这种DP解法时间复杂度为O(n*6^2)，对于n=10^9仍然无法承受。

3. 矩阵快速幂解法详解

3.1 状态转移矩阵构建

将骰子数字之间的关系表示为6×6的转移矩阵T：

T[i][j] = 1 当数字i与j不互斥
T[i][j] = 0 当数字i与j互斥

例如，若数字1与2互斥，则T[1][2]=T[2][1]=0

3.2 矩阵快速幂原理

观察到DP递推式可以表示为：
dp[n] = T^(n-1) * dp[1]

利用矩阵快速幂可以在O(log n)时间内计算矩阵的高次幂。具体步骤：

将转移矩阵T表示为二维数组
实现矩阵乘法函数
实现快速幂算法，每次将指数折半

python复制def matrix_mult(a, b):
    res = [[0]*6 for _ in range(6)]
    for i in range(6):
        for j in range(6):
            for k in range(6):
                res[i][j] = (res[i][j] + a[i][k]*b[k][j]) % MOD
    return res

def matrix_pow(mat, power):
    result = [[1 if i==j else 0 for j in range(6)] for i in range(6)]
    while power > 0:
        if power % 2 == 1:
            result = matrix_mult(result, mat)
        mat = matrix_mult(mat, mat)
        power //= 2
    return result

3.3 完整计算流程

构建转移矩阵T
计算T^(n-1)
初始向量dp[1]是全1向量
结果等于T^(n-1)各元素之和乘以4（每个骰子有4种旋转方式）

python复制def solve():
    n, m = map(int, input().split())
    conflict = [[False]*7 for _ in range(7)]
    for _ in range(m):
        a, b = map(int, input().split())
        conflict[a][b] = conflict[b][a] = True
    
    # 构建转移矩阵
    T = [[0]*6 for _ in range(6)]
    for i in range(6):
        for j in range(6):
            if not conflict[i+1][j+1]:
                T[i][j] = 1
    
    # 矩阵快速幂
    mat_n = matrix_pow(T, n-1)
    
    # 计算结果
    total = 0
    for i in range(6):
        for j in range(6):
            total = (total + mat_n[i][j]) % MOD
    
    # 考虑骰子旋转 (4^n种可能)
    rotation = pow(4, n, MOD)
    ans = (total * rotation) % MOD
    print(ans)