PoPoQQQ 再一次orz…没看得特别明白的可以回来看看蒟蒻的补充口胡

我这里提一下关于 $splay$ 维护的子树信息…
在原树上考虑,对于每一个点 $i$ 都有这样一个信息 $x_i=k_i*x_{fa[i]}+b_i$ .
特别的,对于根节点 $r$ ,设它的父亲为 $sf(special\ father)$ ,有 $x_r=k_r*x_{sf}+b_r$ .
那么我们要维护的就是对于这个联通块中的每一个点 $i$ , $x_i=k'*x_{sf}+b'$ 的 $(k',b')$ , 如果知道了每个点的这个信息, 那么就可以得到 $x_{sf}=k'*x_{sf}+b'$ ,就能解出 $x_{sf}$ .进而解出所有的 $x_i$ .
回到 $LCT$ 上,我们要求一个点 $i$ 的 $(k',b')$ ,首先 $access(i)$ .那么从根到 $i$ 这条路径就成了一棵 $splay$ .因为一个点在 $splay$ 中前驱是自己在原树上的父亲,自己是后继在原树上的父亲,那么我们定义一个结构体 $(k,b)$ 用来表示线性关系.因为根节点(深度最小)肯定是 $splay$ 中最左边的节点(一直往左儿子走),没有前驱,那么它的 $(k,b)$ 值就是本身的 $x_r=k_r*x_{sf}+b_r$ 中的 $(k_r,b_r)$ .那么我们在 $splay$ 的每个节点上维护一个线性关系 $(k,b)$ ,表示这个点 $x_i=k*x_{sf}+b_i$ .
如何在 $splay$ 上维护呢?我们定义一个运算来合并两个线性关系.比如有两个式子①和②.我们定义①*②表示把①合并到②上.也就是这样 $①\ y=k_1x+b_1\\②\ z=k_2y+b_2$ 合并之后就是 $z=k_2(k1x+b1)+b2$
合并代码就是

inline line operator *(const line &o)const {
line res;
res.k = o.k * k % mod;
res.b = (o.k * b + o.b) % mod;
return res;
}

( 表示把*this合并到o上 )
在 $splay$ 上因为左子树是深度小于当前点的(原树上就是这个点的上方),右子树是深度大于当前点的(原树上这个点的下方),因为是每个点的值取决于父亲,那么就要从上往下计算.体现在 $splay$ 上就是把左儿子的线性关系合并到自己上,再把自己的线性关系合并到右二子的线性关系上.对于点 $x$ 上传信息的代码就是

sum[x] = sum[ls] * w[x] * sum[rs];

( $w[x]$ 表示本点的二元关系,也就是 $(k_x,b_x)$ )
这样一来,要查一个点与 $x_{sf}$ 的线性关系就直接 $splay$ 到根,然后根处储存的线性关系就是要求的.
…(没看懂?多YY下)
那个…还有…这道题可以不用exgcd,直接预处理逆元或者快速幂就行了.

CODE

这题不用换根,不用断边,就没写…

#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
char cb[1<<15],*cs=cb,*ct=cb;
#define getc() (cs==ct&&(ct=(cs=cb)+fread(cb,1,1<<15,stdin),cs==ct)?0:*cs++)
template<class T>inline void read(T &res) {
    char ch; int flg = 1; for(;!isdigit(ch=getchar());)if(ch=='-')flg=-flg;
    for(res=ch-'0';isdigit(ch=getchar());res=res*10+ch-'0'); res*=flg;
}
const int MAXN = 30005;
const int mod = 10007;
int n, q, FA[MAXN], vis[MAXN], inv[mod];
struct line {
	int k, b;
	line() { k = 1, b = 0; }
	inline line operator *(const line &o)const {
		line res;
		res.k = o.k * k % mod;
		res.b = (o.k * b + o.b) % mod;
		return res;
	}
	inline int f(int x) { return (k * x + b) % mod; }
};
namespace LCT {
	#define ls ch[x][0]
	#define rs ch[x][1]
	int ch[MAXN][2], fa[MAXN], sf[MAXN];
	line sum[MAXN], w[MAXN];
	inline bool isr(int x) { return ch[fa[x]][0] != x && ch[fa[x]][1] != x; }
	inline bool get(int x) { return ch[fa[x]][1] == x; }
	inline void upd(int x) {
		sum[x] = sum[ls] * w[x] * sum[rs];
	}
	inline void rot(int x) {
		int y = fa[x], z = fa[y];
		bool l = get(x), r = l^1;
		if(!isr(y)) ch[z][get(y)] = x;
		fa[ch[x][r]] = y, fa[y] = x, fa[x] = z;
		ch[y][l] = ch[x][r], ch[x][r] = y;
		upd(y), upd(x);
	}
	inline void splay(int x) {
		for(; !isr(x); rot(x))
			if(!isr(fa[x])) rot(get(x) == get(fa[x]) ? fa[x] : x);
	}
	inline void access(int x) { int y = 0;
		for(; x; x=fa[y=x]) splay(x), ch[x][1] = y, upd(x);
	}
	inline int sert(int x) { //search_root
		access(x), splay(x);
		while(ch[x][0]) x=ch[x][0];
		return x;
	}
	inline int query(int x) {
		int rt = sert(x);
		access(sf[rt]), splay(sf[rt]);
		int k = sum[sf[rt]].k, b = sum[sf[rt]].b;
		bool flg = 0;
		if(k == 1) {
			if(b) return -1; //无解
			flg = 1; //多解的情况还要继续讨论
		}
		int val_sf = k ? (mod-b) * inv[k-1] % mod : b; //求出了x_sf的值
		access(x), splay(x);
		if(flg) {
			if(!sum[x].k) return sum[x].b; //系数为0就不是多解
			else return -2; //否则多解
		}
		return sum[x].f(val_sf);
	}
	inline void modify(int x, int y, int K, int B) {
		int rt = sert(x);
		w[x].k = K, w[x].b = B, upd(x);
	
		if(x == rt) sf[x] = 0;
		else {
			access(x), splay(x);
			ch[x][0] = fa[ch[x][0]] = 0, upd(x);
			if(sert(sf[rt]) != rt) {
				access(rt), splay(rt);
				fa[rt] = sf[rt], sf[rt] = 0;
			}
		}
		
		access(x), splay(x);
		if(sert(y) == x) sf[x] = y;
		else fa[x] = y;
	}
}
using namespace LCT;
int cur;
void DFS(int x) {
	vis[x] = cur;
	if(vis[FA[x]] == cur) { sf[x] = FA[x]; return; }
	fa[x] = FA[x];
	if(!vis[FA[x]]) DFS(FA[x]);
}

inline void pre() {
	inv[0] = inv[1] = 1;
	for(int i = 2; i < mod; ++i)
		inv[i] = (mod - mod/i) * inv[mod%i] % mod;
}

int main () {
	pre(); read(n);
	for(int i = 1, K, B; i <= n; ++i) {
		read(K), read(FA[i]), read(B);
		w[i].k = K, w[i].b = B;
	}
	for(int i = 1; i <= n; ++i)
		if(!vis[i]) ++cur, DFS(i);
	read(q);
	char s; int x, y, K, B;
	while(q--) {
		while(!isalpha(s=getchar()));
		read(x);
		if(s == 'A') printf("%d\n", query(x));
		else {
			read(K), read(y), read(B);
			modify(x, y, K, B);
		}
	}
}

BZOJ 2759 一个动态树好题 (LCT)

CODE

猜你喜欢