洛谷P4315 月下“毛景树”

题目描述

毛毛虫经过及时的变形，最终逃过的一劫，离开了菜妈的菜园。 毛毛虫经过千山万水，历尽千辛万苦，最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬毛毛虫爬到了一颗小小的“毛景树”下面，发现树上长着他最爱吃的毛毛果~ “毛景树”上有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”结束。
输出格式：对于毛毛虫的每个询问操作，输出一个答案。

输入输出样例

输入样例#1： 

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

输出样例#1： 

9
16

说明

1<=N<=100,000，操作+询问数目不超过100,000。保证在任意时刻，所有树枝上毛毛果的个数都不会超过10^9个。树链剖分+线段树。同时有 区间加 与 区间修改。pushdown 时先区间修改，并把左右儿子的区间加标记改为0，再下传区间加。线段树区间修改时，还要将区间加的标记改为0！还有，注意
4000
边权转点权。附代码：#include<iostream>
#include<algorithm>
#include<cstdio>
#define LSON rt<<1
#define RSON rt<<1|1
#define DATA(x) b[x].data
#define SIGN1(x) b[x].c
#define SIGN2(x) b[x].d
#define LSIDE(x) b[x].l
#define RSIDE(x) b[x].r
#define MAXN 100010
using namespace std;
int n,c=1,d=1;
int head[MAXN],deep[MAXN],size[MAXN],son[MAXN],fa[MAXN],top[MAXN],id[MAXN];
struct node1{
int next,to,w;
}a[MAXN<<1];
struct node2{
int data,c,d,l,r;
}b[MAXN<<2];
struct node3{
int u,v,w;
}g[MAXN];
inline int read(){
int date=0,w=1;char c=0;
while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
return date*w;
}
inline int max(const int &x,const int &y){return x>y?x:y;}
inline void add(int u,int v,int w){
a[c].to=v,a[c].w=w;a[c].next=head[u];head[u]=c++;
a[c].to=u;a[c].w=w;a[c].next=head[v];head[v]=c++;
}
void dfs1(int rt){
son[rt]=0;size[rt]=1;
for(int i=head[rt];i;i=a[i].next){
int will=a[i].to;
if(!deep[will]){
deep[will]=deep[rt]+1;
fa[will]=rt;
dfs1(will);
size[rt]+=size[will];
if(size[son[rt]]<size[will])son[rt]=will;
}
}
}
void dfs2(int rt,int f){
id[rt]=d++;top[rt]=f;
if(son[rt])dfs2(son[rt],f);
for(int i=head[rt];i;i=a[i].next){
int will=a[i].to;
if(will!=fa[rt]&&will!=son[rt])
dfs2(will,will);
}
}
inline void pushup(int rt){
DATA(rt)=max(DATA(LSON),DATA(RSON));
}
inline void pushdown(int rt){
if(LSIDE(rt)==RSIDE(rt))return;
if(SIGN1(rt)!=-1){
SIGN1(LSON)=DATA(LSON)=SIGN1(rt);
SIGN1(RSON)=DATA(RSON)=SIGN1(rt);
SIGN1(rt)=-1;
SIGN2(LSON)=SIGN2(RSON)=0;
}
if(SIGN2(rt)){
SIGN2(LSON)+=SIGN2(rt);DATA(LSON)+=SIGN2(rt);
SIGN2(RSON)+=SIGN2(rt);DATA(RSON)+=SIGN2(rt);
SIGN2(rt)=0;
}
}
void buildtree(int l,int r,int rt){
int mid;
LSIDE(rt)=l;
RSIDE(rt)=r;
SIGN1(rt)=-1;
if(l==r){
DATA(rt)=0;
return;
}
mid=l+r>>1;
buildtree(l,mid,LSON);
buildtree(mid+1,r,RSON);
pushup(rt);
}
void update_one(int l,int r,int c,int rt){
int mid;
if(l<=LSIDE(rt)&&RSIDE(rt)<=r){
SIGN1(rt)=DATA(rt)=c;
SIGN2(rt)=0;
return;
}
pushdown(rt);
mid=LSIDE(rt)+RSIDE(rt)>>1;
if(l<=mid)update_one(l,r,c,LSON);
if(mid<r)update_one(l,r,c,RSON);
pushup(rt);
}
void update_two(int l,int r,int c,int rt){
int mid;
if(l<=LSIDE(rt)&&RSIDE(rt)<=r){
SIGN2(rt)+=c;DATA(rt)+=c;
return;
}
pushdown(rt);
mid=LSIDE(rt)+RSIDE(rt)>>1;
if(l<=mid)update_two(l,r,c,LSON);
if(mid<r)update_two(l,r,c,RSON);
pushup(rt);
}
int query(int l,int r,int rt){
int mid,ans=0;
if(l<=LSIDE(rt)&&RSIDE(rt)<=r)return DATA(rt);
pushdown(rt);
mid=LSIDE(rt)+RSIDE(rt)>>1;
if(l<=mid)ans=max(ans,query(l,r,LSON));
if(mid<r)ans=max(ans,query(l,r,RSON));
return ans;
}
void work1(int x,int y,int k){
while(top[x]!=top[y]){
if(deep[top[x]]<deep[top[y]])swap(x,y);
update_one(id[top[x]],id[x],k,1);
x=fa[top[x]];
}
if(deep[x]>deep[y])swap(x,y);
if(x!=y)update_one(id[x]+1,id[y],k,1);
return;
}
void work2(int x,int y,int k){
while(top[x]!=top[y]){
if(deep[top[x]]<deep[top[y]])swap(x,y);
update_two(id[top[x]],id[x],k,1);
x=fa[top[x]];
}
if(deep[x]>deep[y])swap(x,y);
if(x!=y)update_two(id[x]+1,id[y],k,1);
return;
}
void work3(int x,int y){
int s=0;
while(top[x]!=top[y]){
if(deep[top[x]]<deep[top[y]])swap(x,y);
s=max(s,query(id[top[x]],id[x],1));
x=fa[top[x]];
}
if(deep[x]>deep[y])swap(x,y);
if(x!=y)s=max(s,query(id[x]+1,id[y],1));
printf("%d\n",s);
return;
}
void work(){
char ch[10];
int x,y,k;
while(1){
scanf("%s",ch);
if(ch[0]=='S')break;
x=read();y=read();
switch(ch[0]){
case 'C':{
if(ch[1]=='h'){
g[x].w=y;
update_one(id[g[x].v],id[g[x].v],y,1);
}
if(ch[1]=='o'){
k=read();
work1(x,y,k);
}
break;
}
case 'A':{
k=read();
work2(x,y,k);
break;
}
case 'M':{
work3(x,y);
break;
}
}
}
}
void init(){
n=read();
for(int i=1;i<n;i++){
g[i].u=read();g[i].v=read();g[i].w=read();
add(g[i].u,g[i].v,g[i].w);
}
deep[1]=1;
dfs1(1);
dfs2(1,1);
buildtree(1,n,1);
for(int i=1;i<n;i++){
if(deep[g[i].u]>deep[g[i].v])swap(g[i].u,g[i].v);
update_one(id[g[i].v],id[g[i].v],g[i].w,1);
}
}
int main(){
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
init();
work();
return 0;
}