czy的后宫——矩阵快速幂优化DP

题意

有 n 个位置排成一行，可以放 m 种妹子。每个位置可以放也可以不放，规定某些妹子不能相邻，求方案数。

分析

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

ll qmul(ll x,ll y,ll p){        //快速乘
    x%=p;
    y%=p;
    ll ans=;
    while(y){
        if(y&){
            ans+=x;
            if(ans>=p) ans-=p;  //这样写不能有负数
        }
        x<<=;
        if(x>=p) x-=p;
        y>>=;
    }
    return ans;
}

struct Mat{
    int r,c;
    ll m[][];
    Mat(){
        memset(m,,sizeof(m));
    }
};

Mat mmul(Mat x,Mat y,ll p){
    Mat ans;
    ans.r=x.r;
    ans.c=y.c;
    for(int i=;i<x.r;i++)
        for(int k=;k<x.c;k++)
            for(int j=;j<y.c;j++){
                ans.m[i][j]+=qmul(x.m[i][k],y.m[k][j],p);
                if(ans.m[i][j]>=p) ans.m[i][j]-=p;
            }
    return ans;
}
Mat mpow(Mat x,ll y,ll p){
    Mat ans;
    ans.r=x.r;
    ans.c=x.c;
    for(int i=;i<ans.c;i++) ans.m[i][i]=;
    while(y){
        if(y&) ans=mmul(ans,x,p);
        x=mmul(x,x,p);
        y>>=;
    }
    return ans;
}

const ll mod = ;
Mat T, A;
int n, m;
char s[];

int main(){
    scanf("%d%d", &n, &m);
    T.r = T.c = m+;
    for(int i = ;i < m;i++)
    {
        scanf("%s", s);
        for(int j = ;j < m;j++)
            T.m[i][j] = - (s[j] - '');
        T.m[i][m] =  ;
    }
    for(int i = ;i <= m;i++)  T.m[m][i] = ;   //补充一类”空妹子“

    T = mpow(T, n-, mod);
    A.r = , A.c = m+;
    for(int i = ;i <= m;i++)  A.m[][i] = ;
    A = mmul(A, T, mod);
    ll ans = ;
    for(int i = ;i <= m;i++)  ans = (ans + A.m[][i]) % mod;
    printf("%lld\n", ans);

    return ;
}

没处交题，只能找别人题解的代码对拍，应该没错吧？？

参考链接：https://blog.csdn.net/FSAHFGSADHSAKNDAS/article/details/52745912