蓝桥杯第四届第九题：垒骰子——矩阵解法

蓝桥杯第四届第九题：垒骰子——矩阵解法

题目

赌圣atm晚年迷恋上了垒骰子，就是把骰子一个垒在另一个上边，不能歪歪扭扭，要垒成方柱体。
经过长期观察，atm 发现了稳定骰子的奥秘：有些数字的面贴着会互相排斥！
我们先来规范一下骰子：1 的对面是 4，2 的对面是 5，3 的对面是 6。
假设有 m 组互斥现象，每组中的那两个数字的面紧贴在一起，骰子就不能稳定的垒起来。
atm想计算一下有多少种不同的可能的垒骰子方式。
两种垒骰子方式相同，当且仅当这两种方式中对应高度的骰子的对应数字的朝向都相同。
由于方案数可能过多，请输出模 10^9 + 7 的结果。

不要小看了 atm 的骰子数量哦～

「输入格式」
第一行两个整数 n m
n表示骰子数目
接下来 m 行，每行两个整数 a b ，表示 a 和 b 数字不能紧贴在一起。

「输出格式」
一行一个数，表示答案模 10^9 + 7 的结果。

「样例输入」
2 1
1 2

「样例输出」
544

「数据范围」
对于 30% 的数据：n <= 5
对于 60% 的数据：n <= 100
对于 100% 的数据：0 < n <= 10^9, m <= 36

题解

用一个$6\times 6$的矩阵表示$B$，$B$的列标代表了加上一个骰子之后底部的标号，列标代表了骰子堆的底部下面的数字，乘以一个矩阵$B$代表在骰子堆的下面加上一个骰子。

用一个$6\times 6$的矩阵表示$A$，表示初始矩阵，列标代表了底部的数字，行标代表了顶部的数字。

如果我们要堆$n$层，那么结果就是矩阵$B^{n-1}A$所有元素的和。

根据矩阵相乘的含义：

$$C_{ij}=\sum _{k=1}^{6}b[i][k]\times c^{\prime }[k][j]$$

$k$就代表了两个压紧的数字。

代码

import java.util.*;

public class Main
{
	public static final long MOD = 1000000007;
	public static int[] pair = {0,4,5,6,1,2,3};
	public static long[][] mult6(long[][] a,long[][] b)
	{
		long[][] c = new long[7][7];
		
		for(int i = 1;i < 7;i++)
			for(int j = 1;j < 7;j++)
				for(int k = 1;k < 7;k++)
					c[i][j] = (((a[i][k] * b[k][j]) % MOD) + c[i][j]) % MOD;
					
		return c;
	}
	public static long[][] add6(long[][] a,long[][] b)
	{
		long[][] c = new long[7][7];

		for(int i = 1;i < 7;i++)
			for(int j = 1;j < 7;j++)
				c[i][j] = (a[i][j] + b[i][j]) % MOD;

		return c;
	}
	public static long[][] m(int[][] mutex)
	{
		long[][] c = new long[7][7];
		for(int i = 1;i < 7;i++)
		for(int j = 1;j < 7;j++)
			if(mutex[pair[i]][j] == 0)
				c[i][j] = 4;
				
		return c;
	}
	public static long sum(long[][] a)
	{
		long ans = 0;
		for(int i = 1;i < 7;i++)
		for(int j = 1;j < 7;j++)
			ans = (a[i][j] + ans) % MOD;
		return ans;
	}
	
	public static void main(String[] args)
	{
		long[][] A = {
			{0,0,0,0,0,0,0},
			{0,0,0,0,4,0,0},
			{0,0,0,0,0,4,0},
			{0,0,0,0,0,0,4},
			{0,4,0,0,0,0,0},
			{0,0,4,0,0,0,0},
			{0,0,0,4,0,0,0}
		};
		
		int[][] mutex = {
			{0,0,0,0,0,0,0},
			{0,0,1,0,0,0,0},
			{0,1,0,0,0,0,0},
			{0,0,0,0,0,0,0},
			{0,0,0,0,0,0,0},
			{0,0,0,0,0,0,0},
			{0,0,0,0,0,0,0}
		};
		
		long[][] B = m(mutex);
		
		long n = 1000000000;
		long[][] EX = new long[7][7];
		for(long e = n-1;e > 0;e>>=1,B = mult6(B,B))
		{
			if((e & 1) == 1)
				EX = add6(EX,B);
		}
		EX = mult6(EX,A);
		System.out.println(sum(EX));
	}
}