AcWing 1217. 垒骰子 简单dp+快速幂

AcWing 1217. 垒骰子

赌圣atm晚年迷恋上了垒骰子，就是把骰子一个垒在另一个上边，不能歪歪扭扭，要垒成方柱体。

经过长期观察，atm 发现了稳定骰子的奥秘：有些数字的面贴着会互相排斥！

我们先来规范一下骰子：1 的对面是 4，2 的对面是 5，3 的对面是 6。

假设有 m 组互斥现象，每组中的那两个数字的面紧贴在一起，骰子就不能稳定的垒起来。

atm想计算一下有多少种不同的可能的垒骰子方式。

两种垒骰子方式相同，当且仅当这两种方式中对应高度的骰子的对应数字的朝向都相同。

由于方案数可能过多，请输出模 109+7 的结果。

输入格式
第一行包含两个整数 n,m，分别表示骰子的数目和排斥的组数。

接下来 m 行，每行两个整数 a,b，表示 a 和 b 数字不能紧贴在一起。

输出格式
共一个数，表示答案模 109+7 的结果。

数据范围
1≤n≤109,
1≤m≤36,
1≤a,b≤6
输入样例：

2 1
1 2

输出样例：

544

这道题其实还可以，我想到了一个简单dp的方法
$dp[i,j]$表示前i个最后一个色子朝上的是j。

状态计算：
$dp[i][j]=(dp[i-1][k])$k为能与j对上的编号。
st保存是否能够配对。

代码如下

for (int j = 1; j <= 6; j ++ )
{
    for (int k = 1; k <= 6; k ++ )
    {
        if (st[k][op[j]]) continue;
        f[i][j] = (f[i][j] + f[i - 1][k] * 4) % mod;
    }
}

这个方法肯定过不了1e9的数据。

想过数据我们必须了解一个算法，矩阵相乘+快速幂，之前做题做斐波那契数列的时候用到了矩阵相乘，也学了不少线性代数，所以这道题很容易解决了，
我们先建立一个矩阵

我们需要注意一下，在一个面上面放一个色子是有四种，我们可以转四个面。

然后代码如下：

#include<iostream>
#include<cstring>

using namespace std;
const int N=6,MOD=1e9+7;

int A[N][N];
int n,m;

void multi(int a[][N],int b[][N],int c[][N])
{
    int temp[N][N]={};
    for(int i=0;i<N;i++)
     for(int j=0;j<N;j++) 
      for(int k=0;k<N;k++)
       temp[i][j]=(temp[i][j]+(long long)b[i][k]*c[k][j])%MOD;
       memcpy(a,temp,sizeof temp);
}

int get_op(int x)
{
    if (x >= 3) return x - 3;
    return x + 3;
}

int main(void)
{
    scanf("%d%d",&n,&m);
    int fi[N][N]={{4,4,4,4,4,4}};
    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][b]=0;
        A[b][a]=0;
    }
    n--;
    while(n)
    {
        if(n&1) multi(fi,fi,A);
        multi(A,A,A);
        n>>=1;
    }
    int res=0;
    for(int i=0;i<N;i++)
    res=(res+fi[0][i])%MOD;
    cout<<res<<endl;
}