求条玄学 CE

#include <iostream>
#include <cstring>
#include <array>
using namespace std;
typedef long long ll;
constexpr ll N=2e5+3;
struct Mat
{
    array<array<ll,3>,3> a{};
    ll sze=3;
    Mat operator*(const Mat &other)
    {
        Mat z;
        for (ll k=0;k<sze;k++)
        {
            for (ll i=0;i<sze;i++)
            {
                for (ll j=0;j<sze;j++)
                {
                    z.a[i][j]+=a[i][k]*other.a[k][j];
                }
            }
        }
        return z;
    }
    void eye()
    {
        a={};
        for (ll i=0;i<sze;i++)
        {
            a[i][i]=1;
        }
    }
} t[N<<2];
bool is_not_ava[N<<2];
inline ll ls(ll x)
{
    return x<<1;
}
inline ll rs(ll x)
{
    return x<<1|1;
}
inline void push_up(ll p)
{
    if (is_not_ava[ls(p)]&&is_not_ava[rs(p)])
    {
        t[p].eye();
    }
    else if (is_not_ava[ls(p)])
    {
        t[p]=t[rs(p)];
    }
    else if (is_not_ava[rs(p)])
    {
        t[p]=t[ls(p)];
    }
    else
    {
        t[p]=t[rs(p)]*t[ls(p)];
    }
}
void build(ll p,ll tl,ll tr)
{
    if (tl==tr)
    {
        t[p].eye();
        return;
    }
    ll tmid=(tl+tr)>>1;
    build(ls(p),tl,tmid);
    build(rs(p),tmid+1,tr);
    push_up(p);
}

void change_2(ll p,ll tl,ll tr,ll i)
{
    if (tl==tr)
    {
        t[p].a=array<array<ll,3>,3>{{{1,0,1},{0,0,1},{0,0,1}}};
        return;
    }
    ll tmid=(tl+tr)>>1;
    if (i<=tmid)
    {
        change_2(ls(p),tl,tmid,i);
    }
    else
    {
        change_2(rs(p),tmid+1,tr,i);
    }
    push_up(p);
}
void change_1(ll p,ll tl,ll tr,ll i)
{
    if (tl==tr)
    {
        t[p].a=array<array<ll,3>,3>{{{1,1,1},{0,1,1},{0,0,1}}};
        return;
    }
    ll tmid=(tl+tr)>>1;
    if (i<=tmid)
    {
        change_1(ls(p),tl,tmid,i);
    }
    else
    {
        change_1(rs(p),tmid+1,tr,i);
    }
    push_up(p);
}
void undo(ll p,ll tl,ll tr,ll i)
{
    if (tl==tr)
    {
        is_not_ava[p]=!is_not_ava[p];
        return;
    }
    ll tmid=(tl+tr)>>1;
    if (i<=tmid)
    {
        undo(ls(p),tl,tmid,i);
    }
    else
    {
        undo(rs(p),tmid+1,tr,i);
    }
    push_up(p);
}
ll related_i[N];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    ll n;
    cin>>n;
    build(1,1,n);
    for (ll i=1;i<=n;i++)
    {
        char op;
        cin>>op;
        related_i[i]=i;
        if (op=='1')
        {
            change_1(1,1,n,i);
        }
        else if (op== '2')
        {
            change_2(1,1,n,i);
        }
        else if (op=='3')
        {
            ll x;
            cin>>x;
            related_i[i]=related_i[x];
            undo(1,1,n,related_i[x]);
        }
        ll f=0;
        for (ll i=0;i<3;i++)
        {
            f+=t[1].a[0][i];
        }
        cout<<f<<'\n';
    }
}