hdu 2276 Kiki & Little Kiki 2（矩阵快速幂+循环矩阵）

jpphy0

于 2021-04-05 19:50:16 发布

阅读量196

点赞数 1

分类专栏： # 矩阵快速幂

本文链接：https://blog.csdn.net/jpphy0/article/details/115446314

版权

矩阵快速幂专栏收录该内容

10 篇文章 0 订阅

订阅专栏

Kiki & Little Kiki 2

分析

转移方程：设某两相邻的灯的编号为i-1和i，则在t时刻，L_i-1和L_i的状态组合有4种，在t+1时刻，L_i的可能状态及其状态转移方程如下：
在这里插入图片描述

转移矩阵
在这里插入图片描述

代码

非优化【456MS】

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define MXM 110
#define mod(x) ((x)%2)
int N, M;
string L;
struct mat{ 
    LL d[MXM][MXM];
    mat operator*(const mat x){
        mat ret;
        LL tmp;
        for(int i = 0; i < N; i++){
            for(int j = 0; j < N; j++){
                tmp = 0;
                for(int k = 0; k < N; k++){
                    tmp = mod(tmp + d[i][k]* x.d[k][j]);
                }
                ret.d[i][j] = tmp;
            }
        }
        return ret;
    }
    void init_unit(){
        for(int i = 0; i < N; i++)
            for(int j = 0; j < N; j++) d[i][j] = i == j ? 1 : 0;
    }
    void init(){
        memset(d, 0, sizeof d);
        for(int i = 0; i < N; i++)
            for(int j = 0; j < N; j++){
                if(i == j) d[i][j] = 1, d[i][(N+j-1)%N] = 1;
            }
    }
}ma;
mat matrixPow(mat base, int pow){
    mat res;
    res.init_unit();
    while(pow){
        if(pow & 1) res = res * base;
        base = base * base;
        pow >>= 1;
    }
    return res;
}
void solve(){
    ma.init();
    ma = matrixPow(ma, M);
    int res;
    for(int i = 0; i < N; i++){
        res = 0;
        for(int j = 0; j < N; j++) res += ma.d[i][j]*(L[j] - '0');
        printf("%d", mod(res));
    }
    printf("\n");
}
int main(){
    while(scanf("%d", &M) == 1) {
        cin >> L;
        N = L.size();
        solve();
    }
    return 0;
}

循环矩阵优化【46MS】
在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define MXM 110
#define mod(x) ((x)%2)
int N, M;
string L;
struct mat{ 
    LL d[MXM][MXM];
    mat operator*(const mat x){ // 循环矩阵乘法
        mat ret;
        for(int i = 0; i < N; i++){
            ret.d[0][i] = 0;
            for(int j = 0; j < N; j++){ // 0-1矩阵乘法、%2运算优化
                ret.d[0][i] ^= d[0][j] & x.d[j][i];
            }
        }
        for(int i = 1; i < N; i++) // 循环矩阵的乘积，相邻行错1列
            for(int j = 0; j < N; j++)
                ret.d[i][j] = ret.d[i-1][(j-1+N)%N];
        return ret;
    }
    void init_unit(){
        for(int i = 0; i < N; i++)
            for(int j = 0; j < N; j++) d[i][j] = i == j ? 1 : 0;
    }
    void init(){
        memset(d, 0, sizeof d);
        for(int i = 0; i < N; i++)
            for(int j = 0; j < N; j++){
                if(i == j) d[i][j] = 1, d[i][(N+j-1)%N] = 1;
            }
    }
}ma;
mat matrixPow(mat base, int pow){
    mat res;
    res.init_unit();
    while(pow){
        if(pow & 1) res = res * base;
        base = base * base;
        pow >>= 1;
    }
    return res;
}
void solve(){
    ma.init();
    ma = matrixPow(ma, M);
    int res;
    for(int i = 0; i < N; i++){
        res = 0;
        for(int j = 0; j < N; j++) res += ma.d[i][j]*(L[j] - '0');
        printf("%d", mod(res));
    }
    printf("\n");
}
int main(){
    while(scanf("%d", &M) == 1) {
        cin >> L;
        N = L.size();
        solve();
    }
    return 0;
}