蓝桥杯算法双周赛-魔术师-线段树维护矩阵乘法

题目

刚开始做这题的时候我连矩阵乘法都不会，现在有机会便来直面这道题。

分析

对于每一个盒子，虽然可以有多个小球，但它一定只存在一种颜色，因为无论是颜色互换还是染色，由于初始时小球只有一个，盒子内的小球颜色要么是一起不变，要么是一起变，无论它是否增多，盒子内小球颜色一定都一样。

首先我们可以考虑用 1  3 的矩阵维护每一个盒子，用线段树维护每一个矩阵，子节点信息以矩阵的加法上传至父节点，这样根节点矩阵的三列值一定表示所有盒子红、黄、绿球的个数。

我们可以考虑用每个盒子的 1  3 的矩阵 和一个3  3 的矩阵的 矩阵乘法 模拟魔术操作，由于矩阵满足乘法结合律和乘法分配律，所以整个线段树维护的矩阵也满足，每一次对区间[l, r]的操作相当于对[l,r]的每一个矩阵右乘一个3  3的矩阵，而懒标记的下传不影响矩阵右乘的顺序，维护了矩阵乘法，所以可以使用懒标记来维护线段树的每一个节点。

对于操作颜色互换：

                                         =           

对于操作染色：

                                          =         

对于操作分裂：

                                          =        

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 100050, P = 998244353;
struct Matrix {
    int a[3][3];
    void init(){
        memset(a, 0, sizeof a);
        for (int i = 0; i < 3; ++ i) a[i][i] = 1;
    }
    int *operator [](int i){
        return a[i];
    }
    Matrix operator * (Matrix tmp){
        Matrix t = {0};
        for (int i = 0; i < 3; ++ i)
            for (int j = 0; j < 3; ++ j)
                for (int k = 0; k < 3; ++ k)
                    t[i][j] = (t[i][j] + 1ll*a[i][k]*tmp[k][j]) % P;
        return t;
    }
    Matrix operator + (Matrix tmp){
        Matrix t = {0};
        for (int i = 0; i < 3; ++ i)
            for (int j = 0; j < 3; ++ j)
                t[i][j] = (a[i][j] + tmp[i][j]) % P;
        return t;
    }
    void show(int row){
        for (int i = 0; i < row; ++ i){
            for (int j = 0; j < 3; ++ j) cout << a[i][j] << " ";
            cout << "\n";
        }
    }
    friend ostream &operator << (ostream &out, Matrix tmp){ //写这个单纯为了练熟练度
        out << tmp[0][0] <<  " " << tmp[0][1] << " " << tmp[0][2] << "\n";
        return out;
    }
};
struct Tree {
    Matrix c, tag;
    int l, r;
    bool st;
}tr[4*N];

int head[N], n, m; 
void pushup(int id){
    tr[id].c = tr[id<<1].c + tr[id<<1|1].c;
}
void pushtag(int id, Matrix x){
    tr[id].c = tr[id].c * x, tr[id].tag = tr[id].tag * x, tr[id].st = true;
}
void pushdown(int id){
    pushtag(id<<1, tr[id].tag), pushtag(id<<1|1, tr[id].tag);
    tr[id].tag.init(), tr[id].st = false;
}
void build(int id, int l, int r){
    tr[id].l = l, tr[id].r = r, tr[id].tag.init();
    if (l == r){
        tr[id].c[0][head[l]] = 1;
        return;
    }
    int mid = l+r >> 1;
    build(id<<1, l, mid), build(id<<1|1, mid+1, r);
    pushup(id);
}
void modify(int id, int l, int r, Matrix x){
    if (tr[id].l >= l && tr[id].r <= r){
        tr[id].c = tr[id].c * x, tr[id].tag = tr[id].tag * x, tr[id].st = true;
        return;
    }
    if (tr[id].st) pushdown(id);
    int mid = tr[id].l + tr[id].r >> 1;
    if (l <= mid) modify(id<<1, l, r, x);
    if (r > mid) modify(id<<1|1, l, r, x);
    pushup(id);
}

int main(){
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; ++ i) cin >> head[i], -- head[i];
    build(1, 1, n);
    while (m--){
        int l, r, opt, a, b;
        cin >> l >> r >> opt >> a;
        Matrix tmp = {0};
        if (opt == 1){
            cin >> b, -- a, -- b;
            int c = 3-a-b;
            tmp[c][c] = tmp[a][b] = tmp[b][a] = 1;
        }
        else if (opt == 2){
            cin >> b, -- a, -- b;
            int c = 3-a-b;
            tmp[c][c] = tmp[a][b] = tmp[b][b] = 1;
        }else tmp.init(), -- a, ++ tmp[a][a];

        modify(1, l, r, tmp);
        tr[1].c.show(1);
    }
    return 0;
}