动态树LCT小结

      最开始看动态树不知道找了多少资料,总感觉不能完全理解。但其实理解了就是那么一回事。。。动态树在某种意思上来说跟树链剖分很相似,都是为了解决序列问题,树链剖分由于树的形态是不变的,所以可以通过预处理节点间的关系,将树转化成连续的区间,再加以其它的数据结构,便能以较快的速度处理序列的修改和查询。

      而动态树的问题,是包括了树的合并和拆分操作。这个时候,通过预处理实现的静态树的序列算法不能满足我们的要求,于是我们需要一颗‘动态’的树,能在O(logN)的时间复杂度,处理所有操作。

 

Splay实现的Link/cut tree

      Splay能够维护一颗树的信息,我们将多颗Splay树,通过从下自上的单向边连成一颗树。我们将这些边称为“虚边”

     这个时候,Splay树只能维护它本身的节点,而不能照顾到由虚边连成的树。

     由于要处理一段序列,我们就要得到一段序列。

     下面是杨哲的论文的一段原话:

称一个点被访问过, 如果刚刚执行了对这个点的 ACCESS 操作.
如果结点 v 的子树中, 最后被访问的结点在子树 w 中, 这里 w 是 v 的儿子, 那么就称 w 是 v 的 Pre-
ferred Child. 如果最后被访问过的结点就是 v 本身, 那么它没有 Preferred Child. 每个点到它的 Preferred
Child 的边称作 Preferred Edge. 由 Preferred Edge 连接成的不可再延伸的路径称为 Preferred Path.
这样, 整棵树就被划分成了若干条 Preferred Path. 对每条 Preferred Path, 用这条路上的点的深度作
为关键字, 用一棵平衡树来维护它(在这棵平衡树中, 每个点的左子树中的点, 都在 Preferred Path 中这个点
的上方; 右子树中的点, 都在 Preferred Path 中这个点的下方). 需要注意的是, 这种平衡树必须支持分离与
合并. 这里, 我们选择 Splay Tree 作为这个平衡树的数据结构. 我们把这棵平衡树称为一棵 Auxiliary Tree.
知道了树 T 分解成的这若干条 Preferred Path, 我们只需要再知道这些路径之间的连接关系, 就可以表
示出这棵树 T. 用 Path Parent 来记录每棵 Auxiliary Tree 对应的 Preferred Path 中的最高点的父亲结点,
如果这个 Preferred Path 的最高点就是根结点, 那么令这棵 Auxiliary Tree 的 Path Parent 为 null.
Link-Cut Trees 就是将要维护的森林中的每棵树 T 表示为若干个 Auxiliary Tree, 并通过 Path
Parent 将这些 Auxiliary Tree 连接起来的数据结构.

    通过上述的Access操作,我们就可以得到一段从任意点到根的序列,而通过Splay我们又可以将任意点变成根!

    同时使用Splay我们可以很轻松地维护点的信息。

    从这里似乎看到了LCT的核心思路了。

考虑核心的操作Access:

node *Access (node *u) {

	node *v = NIL;

	for (; u != NIL; u = u->par) {

		Splay (u);

		u->Ch[1] = v;

		update (v = u);

	}

	return v;

}

  

这样每次将当前Splay树连接到虚边连接的上一颗Splay树的根的右子树,就保证了Splay树的二叉树性质,同时对于所有指向儿子的点中,在u到根这个序列中,左儿子的深度总是小于父亲,右儿子的深度总是大于父亲。

正是这个关键的性质可以让我们实现我们需要的功能。

 

首先第一个想到的自然是找到一个点的根,只需要不断往左子树找就好了

node *getroot (node *x) {

	for (x = Access (x); clear (x), x->Ch[0] != NIL; x = x->Ch[0]);

	return x;

}

  

 

第二个是要让一个结点x变成新的根,显然做过Access后,在序列 根->x中根在Splay树的最左,x在Splay树的最右,此时只要将Splay树的根(注意区分这两个根)的左右子树交换位置,便让x成为了新的根,于是我们只要打上一个交换标记就好了

inline void evert (node *x) {

	Access (x)->rev ^= 1;

	Splay (x);

}

  

在打完标记后要让x旋转至根更新Splay树

 

下面要实现树的合并。要在不同的两颗树的两个结点u,v间连接一条边,那么先要让其中一个点成为根,用虚边连另外一个点,再用Access将虚边变成实边就好了 

 

inline void link (node *x, node *y) {

	evert (x);

	x->par = y;

	Access (x);

}

 

  

 

  

同样树的分离也是一样,先然其中一个结点x成为根,选取y到x的序列成为一颗Splay树,这个时候再将y旋转至根,那么显然它的左子树包含了除了y的其它点,将它们分离即可

 

inline void cut (node *x, node *y) {

	evert (x);

	Access (y);

	Splay (y);

	y->Ch[0]->par = NIL;

	y->Ch[0] = NIL;

	update (y);

}

 

  

 

  

如果要对一段序列进行操作,例如对树上x到y的路径上的点进行操作。先让x成为根,选取y到x的路径上的点和边做一颗Splay树,将y旋转至根(更新Splay树),将标记传给y就好了

 

inline void modify (node *x, node *y, int w) {

	evert (x);

	Access (y), Splay (y);

	_inc (y, w);

}

 

  

 

 

查询只要同修改一样,只要直接返回我们需要的值就好了。

 

 


下面是一些例题:

1.HDU 4010

只需要将上述操作按要求调用就行了,模板题 ------  题解

 

 

2.BZOJ 2002

装置从0开始

第i个装置能到达第i + ki个装置,意味着i的父亲是i + ki,如果i + ki 大于等于N,它的父亲就是N,这样即询问树上某个点到N 的距离,即由这点到n的Splay树的节点个数-1;

#include <iostream> #include <cstdio>

using namespace std; const int MAXN = 200009; struct node { int sum; bool rev; node *par, *ch[2]; node() {sum = 0, rev = 0, par = ch[0] = ch[1] = 0;} node (int a) : sum (a) {rev = 0, par = ch[0] = ch[1] = 0;} } dt[MAXN], nil (0), *NIL = &nil; struct LinkcutTree { inline void update (node * x) { x->sum = x->ch[0]->sum + x->ch[1]->sum + 1; } void Rotate (node *x) { node *p = x->par, *g = p->par; int c = p->ch[0] == x; //0左旋,1右旋

        p->ch[c ^ 1] = x->ch[c]; if (x->ch[c] != NIL) x->ch[c]->par = p; x->par = g; if (g->ch[0] == p) g->ch[0] = x; else if (g->ch[1] == p) g->ch[1] = x; x->ch[c] = p; p->par = x; update (p); } //将x旋转至x所在Splay树的根

    void Splay (node *x) { node *p, *g; while ( (p = x->par) != NIL && (p->ch[0] == x || p->ch[1] == x) ) { if ( (g = x->par) != NIL && (p->ch[0] == x || p->ch[1] == x) ) { Rotate (x); } else { if ( (g->ch[1] == p) == (p->ch[1] == x) ) Rotate (p), Rotate (x); else Rotate (x), Rotate (x); } } update (x); } //获取从u到根的一段

    node *Access (node *u) { node *v = NIL; for (; u != NIL; u = u->par) { Splay (u); u->ch[1] = v; update (v = u); } return v; } } LCT; int n, m; int f[MAXN], vis[MAXN]; int main() { scanf ("%d", &n); for (int i = 0; i < n; i++) scanf ("%d", &f[i]); for (int i = 0; i <= n; i++) { dt[i].sum = 1, dt[i].rev = 0; dt[i].par = dt[i].ch[0] = dt[i].ch[1] = NIL; int t = i + f[i] < n ? i + f[i] : n; if(i!=n) dt[i].par = dt + t; } scanf ("%d", &m); for (int i = 1, cmd, x, k; i <= m; i++) { scanf ("%d %d", &cmd, &x); node * const tem = dt + x; if (cmd == 1) { LCT.Access (tem); LCT.Splay(tem); printf ("%d\n", tem->sum-1); } else { scanf ("%d", &k); LCT.Splay (tem); tem->ch[0]->par = tem->par; tem->ch[0] = NIL; tem->par = dt + (x + k < n ? x + k : n); } } }
View Code

 

 

3.BZOJ 2243

维护cl,cr,sum分别表示最左边的颜色,最右边的颜色,和颜色段数。

每个节点x的相邻的两个节点的颜色就可以由 x->ch[0]->cr 和x->ch[1]->cl 得到

颜色段数也可以由左右儿子得到

要注意的是,在打上翻转标记后cl和cr也要交换

#include <iostream> #include <cstdio>

using namespace std; const int MAXN = 100009; struct node { //本身的颜色,最左边节点的颜色,最右边节点的颜色,颜色段数

    int color, cl, cr, sum, cover; bool rev; node *par, *ch[2]; inline void cov (int x) { cover = cr = cl = color = x, sum = 1; } inline void re() { swap (cl, cr); rev ^= 1; } } dt[MAXN], *NIL = dt; struct LinkcutTree { inline void clear (node *const x) { if (x == NIL) return ; if (x->rev) { swap (x->ch[0], x->ch[1]); x->ch[0]->re(); x->ch[1]->re(); x->rev = 0; } if (x->cover) { if (x->ch[0] != NIL) x->ch[0]->cov (x->cover); if (x->ch[1] != NIL) x->ch[1]->cov (x->cover); x->cover = 0; } } inline void update (node * x) { if (x->ch[0] != NIL) x->cl = x->ch[0]->cl; else x->cl = x->color; if (x->ch[1] != NIL) x->cr = x->ch[1]->cr; else x->cr = x->color; x->sum = 1; if (x->ch[0] != NIL) { x->sum += x->ch[0]->sum; if (x->ch[0]->cr == x->color) --x->sum; } if (x->ch[1] != NIL) { x->sum += x->ch[1]->sum; if (x->ch[1]->cl == x->color) --x->sum; } } void Rotate (node *x) { node *p = x->par, *g = p->par; int c = p->ch[0] == x; //0左旋,1右旋

        p->ch[c ^ 1] = x->ch[c]; if (x->ch[c] != NIL) x->ch[c]->par = p; x->par = g; if (g->ch[0] == p) g->ch[0] = x; else if (g->ch[1] == p) g->ch[1] = x; x->ch[c] = p; p->par = x; update (p); } void Splay (node *x) { node *p, *g; clear (x); while ( (p = x->par) != NIL && (p->ch[0] == x || p->ch[1] == x) ) { if ( (g = p->par) != NIL && (g->ch[0] == p || g->ch[1] == p) ) { clear (g), clear (p), clear (x); if ( (g->ch[1] == p) == (p->ch[1] == x) ) Rotate (p), Rotate (x); else Rotate (x), Rotate (x); } else { clear (p), clear (x); Rotate (x); } } update (x); } node *Access (node *u) { node *v = NIL; for (; u != NIL; u = u->par) { Splay (u); u->ch[1] = v; update (v = u); } return v; } inline void evert (node *x) { Access (x)->re(); Splay (x); } inline void link (node *x, node *y) { evert (x); x->par = y; Access (x); } inline int query (node *x, node *y) { evert (x); Access (y), Splay (y); return y->sum; } inline void modify (node *x, node *y, int w) { evert (x); Access (y), Splay (y); y->cov (w); } } LCT; int n, m; int main() { scanf ("%d %d", &n, &m); for (int i = 1, x; i <= n; i++) { scanf ("%d", &x); dt[i].par = dt[i].ch[0] = dt[i].ch[1] = NIL; dt[i].cover =0, dt[i].sum = 1; dt[i].cl = dt[i].cr = dt[i].color = x+1; } for (int i = 1, x, y; i < n; ++i) { scanf ("%d %d", &x, &y); LCT.link (dt + x, dt + y); } char cmd; for (int i = 1, u, v, k; i <= m; i++) { scanf ("\n%c %d %d", &cmd, &u, &v); if (cmd == 'Q') printf ("%d\n", LCT.query (dt + u, dt + v) ); else if (cmd == 'C') { scanf ("%d", &k); LCT.modify (dt + v, dt + u, k+1); } } return 0; }
View Code

 

4.BZOJ 2631

 

重点在于处理加和乘的共存问题

乘的时候所有值都要乘,加的时候sum要算上Splay树所有节点,中间值会爆INT

/*

       BZOJ 2631 LCT

       需要的操作 路径权值 + *

       处理+和*的共存

*/

#include <iostream>

#include <cstdio>

#define ll long long

using namespace std;



const int MAXN = 100009, mod = 51061;



struct node {

    int val, sum, inc, mtp,cnt;

    bool rev;

    node *par, *ch[2];

} dt[MAXN], *NIL = dt;



struct LinkcutTree {

    inline void _inc (node * x,  int inc) {

        if (x == NIL) return;

        x->inc=(x->inc + inc)%mod;

        x->val=(x->val + inc)%mod;

        x->sum=(x->sum + ((ll)inc*x->cnt)%mod)%mod;

    }

    inline void _mtp (node *x, int mtp) {

        if (x == NIL) return;

        x->inc=((ll)x->inc * mtp)%mod;

        x->val=((ll)x->val * mtp)%mod;

        x->sum=((ll)x->sum*mtp)%mod;

        x->mtp=((ll)x->mtp*mtp)%mod;

    }

    inline void clear (node *const x) {

        if (x == NIL) return ;

        if (x->mtp!=1) {

            _mtp (x->ch[0], x->mtp);

            _mtp (x->ch[1], x->mtp);

            x->mtp = 1;

        }

        if (x->inc) {

            _inc (x->ch[0], x->inc);

            _inc (x->ch[1], x->inc);

            x->inc = 0;

        }

        if (x->rev) {

            swap (x->ch[0], x->ch[1]);

            x->ch[0]->rev ^= 1;

            x->ch[1]->rev ^= 1;

            x->rev = 0;

        }

    }

    inline void update (node * x) {

        x->sum=x->val,x->cnt=1;

        if(x->ch[0]!=NIL) {

                            x->sum=(x->sum+ x->ch[0]->sum);

                            x->cnt=(x->cnt+ x->ch[0]->cnt);

        }

        if(x->ch[1]!=NIL) {

                            x->sum=(x->sum+ x->ch[1]->sum);

                            x->cnt=(x->cnt+ x->ch[1]->cnt);

        }

        while(x->sum>=mod) x->sum-=mod;

    }

    void Rotate (node *x) {

        node *p = x->par, *g = p->par;

        int c = p->ch[0] == x; //0左旋,1右旋

        p->ch[c ^ 1] = x->ch[c];

        if (x->ch[c] != NIL) x->ch[c]->par = p;

        x->par = g;

        if (g->ch[0] == p) g->ch[0] = x;

        else if (g->ch[1] == p) g->ch[1] = x;

        x->ch[c] = p;

        p->par = x;

        update (p);

    }

    void Splay (node *x) {

        node *p, *g;

        clear (x);

        while ( (p = x->par) != NIL && (p->ch[0] == x || p->ch[1] == x) ) {

            if ( (g = p->par) != NIL && (g->ch[0] == p || g->ch[1] == p) ) {

                clear (g), clear (p), clear (x);

                if ( (g->ch[1] == p) == (p->ch[1] == x) )

                    Rotate (p), Rotate (x);

                else

                    Rotate (x), Rotate (x);

            }

            else {

                clear (p), clear (x);

                Rotate (x);

            }

        }

        update (x);

    }

    node *Access (node *u) {

        node *v = NIL;

        for (; u != NIL; u = u->par) {

            Splay (u);

            u->ch[1] = v;

            update (v = u);

        }

        return v;

    }

    inline void evert (node *x) {

        Access (x)->rev ^= 1;Splay (x);

    }

    inline void link (node *x, node *y) {

        evert (x);x->par = y;Access (x);

    }

    inline void cut (node *x, node *y) {

        evert (x);Access (y);Splay (y);

        x=y->ch[0]->par = NIL;

        update (y);

    }

    inline int query (node *x, node *y) {

        evert (x);Access (y), Splay (y);

        return y->sum;

    }

    inline void modifyadd (node *x, node *y, int w) {

        evert (x);Access (y), Splay (y);

        _inc (y, w);

    }

    inline void modifymtp (node *x, node *y, int w) {

        evert (x);Access (y), Splay (y);

        _mtp (y, w);

    }

} LCT;

int n, q;

int main() {

    scanf("%d %d",&n,&q);

    for (int i = 0; i <= n; i++) {

        dt[i].inc = 0;

        dt[i].mtp = dt[i].sum = dt[i].val = dt[i].cnt=1;

        dt[i].par = dt[i].ch[0] = dt[i].ch[1] = NIL;

    }

    for (int i = 1, x, y; i < n; i++) {

        scanf("%d %d",&x,&y);

        LCT.link (dt + x, dt + y);

    }

    char cmd;

    for (int i = 1, x, y, k, u, v; i <= q; i++) {

        scanf ("\n%c", &cmd);

        switch (cmd) {

        case '+': {

            scanf ("%d %d %d", &x, &y, &k);

            LCT.modifyadd (dt + x, dt + y, k);

            break;

        };

        case '-': {

            scanf ("%d %d %d %d", &x, &y, &u, &v);

            LCT.cut (dt + x, dt + y);

            LCT.link (dt + u, dt + v);

            break;

        }

        case '*': {

            scanf ("%d %d %d", &x, &y, &k);

            LCT.modifymtp (dt + x, dt + y,k);

            break;

        }

        case '/': {

            scanf ("%d %d", &x, &y);

            printf ("%d\n", LCT.query (dt + x, dt + y) );

            break;

        }

        }

    }

}
View Code

 

你可能感兴趣的:(动态)