上周看Lytning(鄙校一队某大佬)开了一个LCT的专题,里面有这个题,正好我也没怎么写过LCT维护子树的题,就决定拿这个题开刀。
先来说说LCT维护子树的大致思路,LCT里的轻边在Splay里都是只有儿子指向父亲但是父亲不指向儿子的虚边,如果维护的是树链,只用维护Splay里的实边,但是维护子树的话只要把每个节点的虚边也维护一下就行了。
我之前只写过不带修改的子树维护,不带修改只要把虚子树信息向上更新即可,但是要修改的话,就要有标记下放,标记下方的时候要对所有虚子树下放标记。如果用一个线性结构维护虚子树,就会被菊花图卡死,于是就想到用一个平衡树来维护所有的虚子树。这个就是传说中的Toptree。
但是我开了这个题之后,发现我进了一个无底洞,写题3小时,debug一星期,本来以为只是要维护的变量多了点而已,但是写的时候发现有一些细节不是很好处理。说一下主要遇到的几个问题,最开始遇到的问题是函数调用,前面的函数要调后面的函数,虽然我觉得这样不是很优雅,但是想了想也没啥好的解决办法。遇到的第二个问题,而是蒙蔽我时间最长的一个问题,我平衡树用的是Treap,但是我没写过标记下放的Treap,于是我在删节点的时候写残了,这个真是坑死我了。还有一个就是Splay和LCT的access(expose)过程中的标记下放,我之前Splay的标记下放都是边转边下放,网上其他大佬们的代码都是从根一路下放下来再转上去,可能我的写法是邪教吧,之前没出过问题,但是这个题就是不对,这个问题其实我现在还没明白为什么不对,希望有大佬知道的可以告诉我……
一些闲话,比如说STL的stack真的血慢,还有就是我的数据结构都是指针写的,虽然经常被学长喷卡内存啥的,但是BZOJ是32位机不怕嘿嘿嘿。
最后奉上我又臭又长的代码,基本上是照着mhy的代码改的
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 100005, INF = 0x7fffffff;
struct Treap
{
struct Node
{
int idx, pr, siz, maxv, minv, sumv, add, val;
bool lazy;
Node *child[2];
inline int cmp(int) const;
inline void maintain();
inline void change_val(int);
inline void add_val(int);
inline void push_down();
}pool[MAXN], *nil, *alloc;
Node* new_Node(int);
void init();
void rotate(Node*&, int);
void insert(Node*, Node*&);
void erase(int, Node*&);
void debug(Node*);
}treap;
struct Splay_tree
{
struct Node
{
int idx;
Treap::Node *root;
Node *child[2], *parent;
inline bool isroot() const;
inline void maintain();
inline void tree_change(int);
inline void chain_change(int);
inline void tree_add(int);
inline void chain_add(int);
inline void reverse();
inline void push_down();
inline void insert(Node*);
inline void erase(Node*);
}*nil, pool[MAXN], *alloc;
Node* new_Node(int);
void init();
Node* find(Node*);
void update(Node*);
void rotate(Node*);
void splay(Node*);
Node* prec(Node*);
};
struct Record
{
int key, cadd, tadd, cval, tval, csiz, tsiz,
cmax, tmax, cmin, tmin, csum, tsum;
bool flip, clazy, tlazy;
Treap::Node* tr;
Splay_tree::Node* sp;
}rec[MAXN], *alloc = rec;
int new_Record(int key)
{
Record* ret = alloc++;
ret->csiz = 1;
ret->key = ret->cmax = ret->cmin = ret->csum = key;
ret->tmax = -INF;
ret->tmin = INF;
ret->tsum = ret->tsiz = ret->cadd = ret->tadd = 0;
ret->flip = ret->clazy = ret->tlazy = false;
return ret - rec;
}
inline int Treap::Node::cmp(int v) const
{
if (v == idx) return -1;
else return v < idx ? 0 : 1;
}
inline void Treap::Node::maintain()
{
siz = rec[idx].csiz + rec[idx].tsiz;
maxv = max(rec[idx].cmax, rec[idx].tmax);
minv = min(rec[idx].cmin, rec[idx].tmin);
sumv = rec[idx].csum + rec[idx].tsum;
for (int i = 0; i < 2; i++