题目描述
毛毛虫经过及时的变形，最终逃过的一劫，离开了菜妈的菜园。 毛毛虫经过千山万水，历尽千辛万苦，最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬~~毛毛虫爬到了一颗小小的“毛景树”下面，发现树上长着他最爱吃的毛毛果~~~ “毛景树”上有N个节点和N-1条树枝，但节点上是没有毛毛果的，毛毛果都是长在树枝上的。但是这棵“毛景树”有着神奇的魔力，他能改变树枝上毛毛果的个数： Change k w：将第k条树枝上毛毛果的个数改变为w个。 Cover u v w：将节点u与节点v之间的树枝上毛毛果的个数都改变为w个。 Add u v w：将节点u与节点v之间的树枝上毛毛果的个数都增加w个。 由于毛毛虫很贪，于是他会有如下询问： Max u v：询问节点u与节点v之间树枝上毛毛果个数最多有多少个。
输入
第一行一个正整数N。 接下来N-1行，每行三个正整数Ui，Vi和Wi，第i+1行描述第i条树枝。表示第i条树枝连接节点Ui和节点Vi，树枝上有Wi个毛毛果。 接下来是操作和询问，以“Stop”结束。
输出
对于毛毛虫的每个询问操作，输出一个答案。
样例输入
4
1 2 8
1 3 7
3 4 9
Max 2 4
Cover 2 4 5
Add 1 4 10
Change 1 16
Max 2 4
Stop
样例输出
9
16
题解
树剖+线段树傻X题
pushdown的时候顺序是先染色后加减，染色时把儿子的加减标记清空。
然后各种傻X操作就过了。
#include <cstdio>
#include <algorithm>
#define N 200010
#define lson l , mid , x << 1
#define rson mid + 1 , r , x << 1 | 1
using namespace std;
int head[N] , to[N << 1] , next[N << 1] , cnt , x[N] , y[N] , z[N] , r[N];
int fa[N] , deep[N] , si[N] , bl[N] , pos[N] , tot;
int w[N] , maxn[N << 2] , tag[N << 2] , cov[N << 2] , n;
char str[10];
void add(int x , int y)
{
to[++cnt] = y , next[cnt] = head[x] , head[x] = cnt;
}
void dfs1(int x)
{
int i;
si[x] = 1;
for(i = head[x] ; i ; i = next[i])
if(to[i] != fa[x])
fa[to[i]] = x , deep[to[i]] = deep[x] + 1 , dfs1(to[i]) , si[x] += si[to[i]];
}
void dfs2(int x , int c)
{
int i , k = 0;
bl[x] = c , pos[x] = ++tot;
for(i = head[x] ; i ; i = next[i])
if(to[i] != fa[x] && si[k] < si[to[i]])
k = to[i];
if(k)
{
dfs2(k , c);
for(i = head[x] ; i ; i = next[i])
if(to[i] != fa[x] && to[i] != k)
dfs2(to[i] , to[i]);
}
}
void pushup(int x)
{
maxn[x] = max(maxn[x << 1] , maxn[x << 1 | 1]);
}
void build(int l , int r , int x)
{
if(l == r)
{
maxn[x] = w[l];
return;
}
int mid = (l + r) >> 1;
build(lson) , build(rson);
pushup(x);
}
void pushdown(int x)
{
int ls = x << 1 , rs = x << 1 | 1;
if(cov[x]) maxn[ls] = maxn[rs] = cov[x] , tag[ls] = tag[rs] = 0 , cov[ls] = cov[rs] = cov[x] , cov[x] = 0;
if(tag[x]) maxn[ls] += tag[x] , maxn[rs] += tag[x] , tag[ls] += tag[x] , tag[rs] += tag[x] , tag[x] = 0;
}
void change(int b , int e , int c , int l , int r , int x)
{
if(b <= l && r <= e)
{
maxn[x] = c , tag[x] = 0 , cov[x] = c;
return;
}
pushdown(x);
int mid = (l + r) >> 1;
if(b <= mid) change(b , e , c , lson);
if(e > mid) change(b , e , c , rson);
pushup(x);
}
void update(int b , int e , int a , int l , int r , int x)
{
if(b <= l && r <= e)
{
maxn[x] += a , tag[x] += a;
return;
}
pushdown(x);
int mid = (l + r) >> 1;
if(b <= mid) update(b , e , a , lson);
if(e > mid) update(b , e , a , rson);
pushup(x);
}
int query(int b , int e , int l , int r , int x)
{
if(b <= l && r <= e) return maxn[x];
pushdown(x);
int mid = (l + r) >> 1 , ans = 0;
if(b <= mid) ans = max(ans , query(b , e , lson));
if(e > mid) ans = max(ans , query(b , e , rson));
return ans;
}
void solvechange(int x , int y , int c)
{
while(bl[x] != bl[y])
{
if(deep[bl[x]] < deep[bl[y]]) swap(x , y);
change(pos[bl[x]] , pos[x] , c , 1 , n , 1) , x = fa[bl[x]];
}
if(deep[x] > deep[y]) swap(x , y);
if(x != y) change(pos[x] + 1 , pos[y] , c , 1 , n , 1);
}
void solveupdate(int x , int y , int a)
{
while(bl[x] != bl[y])
{
if(deep[bl[x]] < deep[bl[y]]) swap(x , y);
update(pos[bl[x]] , pos[x] , a , 1 , n , 1) , x = fa[bl[x]];
}
if(deep[x] > deep[y]) swap(x , y);
if(x != y) update(pos[x] + 1 , pos[y] , a , 1 , n , 1);
}
int solvequery(int x , int y)
{
int ans = 0;
while(bl[x] != bl[y])
{
if(deep[bl[x]] < deep[bl[y]]) swap(x , y);
ans = max(ans , query(pos[bl[x]] , pos[x] , 1 , n , 1)) , x = fa[bl[x]];
}
if(deep[x] > deep[y]) swap(x , y);
if(x != y) ans = max(ans , query(pos[x] + 1 , pos[y] , 1 , n , 1));
return ans;
}
int main()
{
int i , a , b , c;
scanf("%d" , &n);
for(i = 1 ; i < n ; i ++ ) scanf("%d%d%d" , &x[i] , &y[i] , &z[i]) , add(x[i] , y[i]) , add(y[i] , x[i]);
dfs1(1) , dfs2(1 , 1);
for(i = 1 ; i < n ; i ++ ) r[i] = (deep[x[i]] > deep[y[i]] ? x[i] : y[i]) , w[pos[r[i]]] = z[i];
build(1 , n , 1);
while(scanf("%s" , str) != EOF && str[1] != 't')
{
scanf("%d%d" , &a , &b);
switch(str[1])
{
case 'h': change(pos[r[a]] , pos[r[a]] , b , 1 , n , 1); break;
case 'o': scanf("%d" , &c) , solvechange(a , b , c); break;
case 'd': scanf("%d" , &c) , solveupdate(a , b , c); break;
default: printf("%d
" , solvequery(a , b));
}
}
return 0;
}