洛谷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！

还有，注意边权转点权。

附代码：

#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;
}