方格填色【牛客小白月赛7 J】【矩阵快速幂】【极其详细的一份解题报告】

题目链接

  题目要求的是左右两两邻接点不能同时为白色，与两两邻接列不能同为黑色，于是可以有这样的定义，白色为1，黑色为0，最初的时候我想到的还不是矩阵快速幂，我用0、1这样的二进制数表示每一列的状态，对于第i列，假如他是最大上限，也就是5个空格，那么就有2^5种可能性，举个例子，对与这样的两列：a{10110}、b{11110}是不满足规则的，你会发现，a&b>0也就是会同时存在两个邻接的1（白色）；还有例如：a{00000}、b{00000}也是不满足规则的，它们同时为0，也就是全黑状态，显然是不行的。那么，怎样的两列是满足规则的呢？{10101}与{11011}就是可以的，像这样的例子有很多，我们需要如何去处理它们呢？并且，还有那么多行，我们又该如何处理才能不T呢？

  于是，我想到了让这样的条件递推下去，于是便联系到了矩阵快速幂的想法，于是，就算每次的计算是极限值，快速幂令N原本的O(2^60)的复杂度下降到了O(60)，然后矩阵的惩罚运算不过最多O(30000)的时间复杂度，于是便可以开工了，我们比较两两列，能不断的往后递推规律，而这样的规律就是有这样的矩阵——从目前这一列的状态推向下一邻接列的状态的种类个数，并且将满足要求的矩阵上的点计作1，不满足则为0，做矩阵乘法就能不断的更新并且得到新的个数，对于n这一例，我们比较它的n-1那一列，也就是前一列，举例这一列的状态，就是从0～（1<<M）-1的全部（我把每个元素看成i），那么对于n-1那列，我们在0～（1<<M）-1里遍历这样的j，需要满足条件(i&j)==0 「这是因为左右两个不能同时为白」、(i==0 && j==0)时候退出——「因为此时两列同时为黑」，然后，对于符合要求的mat[i][j]我们令它=1，其余不符合要求的为0。

  剩余的就是矩阵快速幂了，类比于快速幂，我们还是一样的处理，但是为了简化，我在结构体中operator（重载函数）了“*”乘法这个操作，使得矩阵乘法简单了。

  对了，还有一件事，矩阵的左乘与矩阵的右乘是不同的。

 

完整代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int maxN=50;
ll M, N, tot, cnt=0;      //在二进制中，用0表示黑，用1表示白
struct juzhen
{
    ll a[maxN][maxN];
    juzhen() { memset(a, 0, sizeof(a)); }
    friend juzhen operator * (juzhen x, juzhen y)       //重载函数，矩阵的乘积得到的新矩阵
    {
        juzhen ans;
        for(int i=0; i<=tot; i++)
        {
            for(int j=0; j<=tot; j++)
            {
                ll tmp=0;
                for(int k=0; k<=tot; k++)
                {
                    tmp=(tmp + x.a[i][k]*y.a[k][j]) %mod;       //最后得到的需要取mod
                }
                ans.a[i][j]=tmp;
            }
        }
        return ans;
    }
};
juzhen fast_jz(juzhen x, ll ti)
{
    juzhen ans;
    for(int i=0; i<=tot; i++) ans.a[i][i]=1;        //单位阵
    while(ti)
    {
        if(ti&1)
        {
            ans=ans*x;
        }
        x=x*x;
        ti>>=1;
    }
    return ans;
}
int main()
{
    while(scanf("%lld%lld", &M, &N)!=EOF)
    {
        tot=1<<M;   tot--;  cnt=0;
        juzhen mat;
        for(int i=0; i<=tot; i++)
        {
            for(int j=0; j<=tot; j++)
            {
                if(i==0 && j==0) continue;      //两列全黑情况
                if( (i&j)==0 ) mat.a[i][j]=1;   //相邻不能两个都为白色的缘故
            }
        }
        juzhen basic, ans;
        basic=fast_jz(mat, N-1);
        for(int i=0; i<=tot; i++) ans.a[0][i]=1;
        ans=ans*basic;
        for(int i=0; i<=tot; i++)
        {
            cnt=( cnt+ans.a[0][i] )%mod;
        }
        printf("%lld\n", cnt);
    }
    return 0;
}