[Acwing|蓝桥] 1217. 垒骰子 dp求方案数+矩阵乘法

前言

传送门 :

题意

有多个骰子,竖成一列的队列,规定$a_i{b}_i$表示如果$a_i$是下面那个骰子的顶,那么$b_i$就不能是上面那个骰子的底。反之亦然,求合法方案数

这里的骰子 1的对面不是6注意,题面有明确规定

思路

首先常规做法考虑$dp$ :
状态表示$f[i][j]:$从前$i$个骰子选且第$i$个骰子数字$j$朝上的所有方案数

状态转移$f[i][j]=f[i-1][k]\ast 4(st[k][op[j]]!=0)$ ($st[k][j]$表示两个数字是否互斥

这里讲的$op$表示对面的数字,因为当你枚举到第$i$层的时候,判断矛盾是$i$ 和 $i-1$层进行判断,所以也需要知道当前以$j$为顶的对面的数字

显然这是会超时的,因为数据范围$1e9$

因此我们考虑使用矩阵乘法进行优化 :

令$F(n)=[f[n][1],f[n][2],f[n][3],f[n][4],f[n][5],f[n][6]]$
则$F(n+1)=[f[n+1][1],f[n+1][2],f[n+1][3],f[n+1][4],f[n+1][5],f[n+1][6]]$

考虑$F(n)\ast A[6][6]=F(n+1)$

$F(n)\left[\begin{array}{cccccc}4& 0& 4& 4& 0& 4\\ 4& 4& 0& 0& 4& 4\\ 0& 4& 4& 4& 4& 4\\ 4& 4& 4& 4& 4& 0\\ 4& 0& 4& 0& 4& 4\\ 4& 4& 4& 4& 0& 4\end{array}\right]=F(n+1)(条件:\left[\begin{array}{cc}1& 2\\ 1& 5\\ 2& 6\\ 3& 4\\ 5& 5\end{array}\right])当前行与当前列互斥$
(数据借 : 蔗)

观察发现,只有互斥的时候,$A_{ij}=0$
$F_n=F_0\ast A^n=F_1\ast A^{n-1}$
所以我们就构造出了,$A$,下面就可以愉快的进行$dp$了

Mycode

const int N  = 6 ,mod = 1e9+7;



int op[N] = {3,4,5,0,1,2};
int n,m;
int A[N][N];

void mul(int c[][N],int a[][N],int b[][N]){
	int  t[N][N] = {0};
	
	for(int i=0;i<N;i++){
		for(int j=0;j<N;j++){
			for(int k=0;k<N;k++){
				t[i][j] = (t[i][j] +1ll*a[i][k]*b[k][j])%mod;
			}
		}
	}
	memcpy(c,t,sizeof t);
}
void solve()
{
	cin>>n>>m;
	
	for(int i=0;i<N;i++){
		for(int j =0;j<N;j++){
			A[i][j] = 4;
		}
	}
	for(int i=1;i<=m;i++){
		int a,b;cin>>a>>b;
		a--,b--;
		A[a][op[b]] = 0;
		A[b][op[a]] = 0;
	}
	
	int f1[N][N] = {4,4,4,4,4,4};
	//表示只有一共骰子的时候 那么肯定就是 4 了


    
	for(int k = n-1;k;k>>=1){
		if(k&1) mul(f1,f1,A);//F =  F*A;
		mul(A,A,A);// A = A*A
	}


	int res = 0;
	for(int i = 0;i<N;i++){
		res =  (res+f1[0][i])%mod;
	}
	cout<<res<<endl;
	
	
}
/**mYHeart is my algorithm**/

int main()
{
    //int t;cin>>t;while(t -- )
    solve();
    return 0;
}