SPOJ GSS7 Can you answer these queries VII

题意:

维护一棵树, 需要满足修改路径上的值和询问路径上最大连续子段和。


思路:

用lct维护。 修改直接搞个lazy就好, 询问的时候如果是求出lca的话比较麻烦, 但是可以先把u弄成树的跟, 然后access(v), 然后在splay(u), 这时候u节点所在的splay树里面就是u到v的所有节点, 所以输出最大值就可以了。。

重点还是在push_up和push_down。。。 其他都是模板。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define N 100020
#define M 200020
#define inf 0x3f3f3f3f


int pre[N], ch[N][2], key[N], sz[N];
int sum[N], lx[N], rx[N], mx[N];
int c[N];
bool rev[N], rt[N];
int n;

int fst[N], nxt[M], vv[M], e;
void init() {
	for(int i = 0; i <= n; ++i) {
		fst[i] = -1;
		rev[i] = false;
		rt[i] = true;
		ch[i][0] = ch[i][1] = 0;
		pre[i] = 0;
		if(i) sz[i] = 1;
		c[i] = -inf;
	}
	e = 0;
}
void add(int u, int v) {
	vv[e] = v, nxt[e] = fst[u], fst[u] = e++;
}

void debug() {
	for(int i = 1; i <= n; ++i) {
		printf("node[%d]:pre %d ch %d %d sz %d rev %s\n", i, pre[i], ch[i][0], ch[i][1], sz[i], rev[i]? "true": "false");
	}
}
void dfs(int u, int p) {
	pre[u] = p;
	for(int i = fst[u]; ~i; i = nxt[i]) {
		int v = vv[i];
		if(v == p) continue;
		dfs(v, u);
	}
}
void push_up(int x) {
	int lc = ch[x][0], rc = ch[x][1];
	sum[x] = sum[lc] + sum[rc] + key[x];
	sz[x] = sz[lc] + sz[rc] + 1;
	if(lc == 0 && rc == 0) {
		mx[x] = lx[x] = rx[x] = key[x];
		return;
	}
	if(rc == 0) {
		mx[x] = max(key[x], mx[lc]);
		mx[x] = max(mx[x], key[x] + rx[lc]);
		lx[x] = max(lx[lc], sum[lc] + key[x]);
		rx[x] = max(key[x], key[x] + rx[lc]);
		return ;
	}
	if(lc == 0) {
		mx[x] = max(key[x], mx[rc]);
		mx[x] = max(mx[x], key[x] + lx[rc]);
		lx[x] = max(key[x], key[x] + lx[rc]);
		rx[x] = max(rx[rc], sum[rc] + key[x]);
		return;
	}
	mx[x] = max(mx[lc], mx[rc]);
	mx[x] = max(mx[x], key[x]);
	mx[x] = max(mx[x], key[x] + rx[lc]);
	mx[x] = max(mx[x], key[x] + lx[rc]);
	mx[x] = max(mx[x], key[x] + rx[lc] + lx[rc]);
	lx[x] = max(lx[lc], sum[lc] + key[x]);
	lx[x] = max(lx[x], sum[lc] + key[x] + lx[rc]);
	rx[x] = max(rx[rc], sum[rc] + key[x]);
	rx[x] = max(rx[x], sum[rc] + key[x] + rx[lc]);
}
void update_c(int x, int val) {
	if(!x) return;
	key[x] = val;
	sum[x] = val * sz[x];
	c[x] = val;
	if(val > 0) {
		mx[x] = lx[x] = rx[x] = sum[x];
	}
	else
		mx[x] = lx[x] = rx[x] = val;
}
void update_rev(int x) {
	if(!x) return;
	swap(ch[x][0], ch[x][1]);
	rev[x] ^= 1;
	swap(lx[x], rx[x]);
}
void push_down(int x) {
	if(c[x] != -inf) {
		update_c(ch[x][0], c[x]);
		update_c(ch[x][1], c[x]);
		c[x] = -inf;
	}
	if(rev[x]) {
		update_rev(ch[x][0]);
		update_rev(ch[x][1]);
		rev[x] = false;
	}
}
void rot(int x) {
	int y = pre[x], d = ch[y][1] == x;
	ch[y][d] = ch[x][!d];
	pre[ch[x][!d]] = y;
	ch[x][!d] = y;
	pre[x] = pre[y];
	pre[y] = x;
	if(rt[y]) rt[y] = false, rt[x] = true;
	else {
		ch[pre[x]][ch[pre[x]][1]==y] = x;
	}
	push_up(y);
}
void P(int x) {
	if(!rt[x]) P(pre[x]);
	push_down(x);
}
void splay(int x) {
	P(x);
	while(!rt[x]) {
		int f = pre[x], ff = pre[f];
		if(rt[f]) rot(x);
		else if((ch[ff][1] == f) == (ch[f][1] == x))
			rot(f), rot(x);
		else rot(x), rot(x);
	}
	push_up(x);
}
int access(int x) {
	int y = 0;
	for(; x; y = x, x = pre[x]) {
		splay(x);
		rt[ch[x][1]] = true;
		ch[x][1] = y;
		rt[y] = false;
		push_up(x);
	}
	return y;
}

void lca(int &u, int &v) {
	access(v), v = 0;
	for(splay(u); pre[u]; v = u, u = pre[u], splay(u)) {
		rt[ch[u][1]] = true;
		ch[u][1] = v;
		rt[v] = false;
		push_up(u);
	}
}
void make_root(int x) {
	access(x);
	splay(x);
	update_rev(x);
}
int main() {
	//freopen("tt.txt", "r", stdin);
	while(scanf("%d", &n) != EOF) {
		init();
		for(int i = 1; i <= n; ++i) {
			scanf("%d", &key[i]);
			sum[i] = mx[i] = lx[i] = rx[i] = key[i];
		}
		for(int i = 1; i < n; ++i) {
			int u, v;
			scanf("%d%d", &u, &v);
			add(u, v), add(v, u);
		}
		dfs(1, 0);
		int m;
		scanf("%d", &m);
		while(m--) {
			int op, u, v, c;
			scanf("%d%d%d", &op, &u, &v);
			if(op == 2) scanf("%d", &c);
			if(op == 2) {
				lca(u, v);
				key[u] = c;
				update_c(ch[u][1], c);
				update_c(v, c);
				push_up(u);
			}
			else {
				make_root(u);
				access(v);
				splay(u);
				//debug();
				printf("%d\n", max(0, mx[u]));
			}
		}
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值