蓝桥杯—垒骰子—Java

1、问题描述

赌圣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

2、代码实现

import java.util.Scanner;

public class Main{
	static int N = 1000000007;
	//对立面
	static int[] opponent = {0,4,5,6,1,2,3};
	//跟上面的代码等价： opponent[1] = 4; opponent[4] = 1;
	//opponent[2] = 5;opponent[5] = 2;opponent[3] = 6;opponent[6] = 3;
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		//骰子的个数，也是层数
		int n = scanner.nextInt();
		//冲突面的个数
		int m = scanner.nextInt();
		//创建冲突面的数组，存储哪个面冲突
		long[][] conflict = new long[7][7];
		//全部初始化为1，默认为不冲突，当某个点为0时，表示对应的x和y冲突
		for(int i = 1;i < 7;i++) {
			for(int j = 1;j < 7;j++) {
				conflict[i][j] = 1;
			}
		}
		//接收冲突面
		for(int i = 0;i < m;i++) {
			int x = scanner.nextInt();
			int y = scanner.nextInt();
			//这里为什么这个写？
			//假如我们输入1和2位冲突面
			//如果第一个骰子的上面是1，那么第二个骰子的上面就不能是5
			//此时第二个骰子在第一个骰子的上方，第二个骰子的上面是5，底面就是2
			//此时1和2就会发生冲突
			conflict[opponent[x]][y] = 0;
			conflict[opponent[y]][x] = 0;
		}
		//此时的矩阵存储的就是n个骰子的摆放次数
		long[][] ans = matrixPow(conflict, n - 1);
		long sum = 0;
		for(int i = 1;i < 7;i++) {
			for(int j = 1;j < 7;j++) {
				sum = (sum + ans[i][j]) % N;
			}
		}
		//每个骰子的侧面都可以旋转4次，有n个骰子，所以要乘上4^n
		System.out.println(sum * pow(4,n) % N);
	}
	//矩阵相乘
	public static long[][] matrixMultiple(long[][] matrix1,long[][] matrix2){
		long[][] ans = 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++) {
					ans[i][j] = (ans[i][j] + matrix1[i][k] * matrix2[k][j]) % N;
				}
			}
		}
		return ans;
	}
	//矩阵的pow次幂
	public static long[][] matrixPow(long[][] matrix,int pow){
		long[][] ans = new long[7][7];
		//初始化为单位矩阵
		for(int i = 1;i < 7;i++) {
			ans[i][i] = 1;
		}
		while(pow != 0) {
			//如果pow为奇数就多乘一次
			if(pow % 2 == 1) {
				ans = matrixMultiple(ans,matrix);
			}
			matrix = matrixMultiple(matrix,matrix);
			//右移一位
			pow >>= 1;
		}
		return ans;
	}
	//num的pow次幂
	public static long pow(long n,int pow) {
		long ans = 1;
		while(pow != 0) {
			//如果pow是奇数就多乘一次
			if(pow % 2 == 1) {
				ans = ans * n % N;
			}
			n = n * n % N;
			//右移一位
			pow >>= 1;
		}
		return ans;
	}
}