【BZOJ-1984】月下“毛景树” 树链剖分

时间:2023-03-08 18:35:28

1984: 月下“毛景树”

Time Limit: 20 Sec  Memory Limit: 64 MB
Submit: 1314  Solved: 416
[Submit][Status][Discuss]

Description

毛毛虫经过及时的变形,最终逃过的一劫,离开了菜妈的菜园。 毛毛虫经过千山万水,历尽千辛万苦,最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬~~毛毛虫爬到了一颗小小的“毛景树”下面,发现树上长着他最爱吃的毛毛果~~~ “毛景树”上有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之间树枝上毛毛果个数最多有多少个。

Input

第一行一个正整数N。 接下来N-1行,每行三个正整数Ui,Vi和Wi,第i+1行描述第i条树枝。表示第i条树枝连接节点Ui和节点Vi,树枝上有Wi个毛毛果。 接下来是操作和询问,以“Stop”结束。

Output

对于毛毛虫的每个询问操作,输出一个答案。

Sample Input

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

Sample Output

9
16

【Data Range】
1<=N<=100,000,操作+询问数目不超过100,000。
保证在任意时刻,所有树枝上毛毛果的个数都不会超过10^9个。

HINT

Source

Solution

很容易想到树链剖分,但此题权值在边上,所以思路应有所变化,即 边权下放至点 

具体实现起来并不是很难。 把边权下放到较深的点,然后进行正常的链剖即可。

维护两个标记,变换和增加,变换操作的时候,同时应该把当前的增加标记置成0,而增加操作时,则无需处理变换标记。

在区间修改的时候,需要多进行一次判断;这是不同于普通点权链剖的地方。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
inline int read()
{
int x=,f=;char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}
return x*f;
}
#define maxn 400010
int id[maxn<<],va[maxn<<],uu[maxn],vv[maxn],ww[maxn];
int n;struct data{int next,to,w;}edge[maxn<<];int head[maxn],cnt;
void add(int u,int v,int w){cnt++;edge[cnt].w=w;edge[cnt].to=v;edge[cnt].next=head[u];head[u]=cnt;}
void insert(int u,int v,int w){add(u,v,w); add(v,u,w);}
//--------------------------------------------------------------------------------------
int deep[maxn],fa[maxn],son[maxn],size[maxn],pl[maxn],sz,pr[maxn],top[maxn],pre[maxn];
void dfs_1(int now)
{
size[now]=;
for (int i=head[now]; i; i=edge[i].next)
if (edge[i].to!=fa[now])
{
fa[edge[i].to]=now;
deep[edge[i].to]=deep[now]+;
dfs_1(edge[i].to);
if (size[son[now]]<size[edge[i].to]) son[now]=edge[i].to;
size[now]+=size[edge[i].to];
}
}
void dfs_2(int now,int chain)
{
pl[now]=++sz; pre[sz]=now; top[now]=chain;
if (son[now]) dfs_2(son[now],chain);
for (int i=head[now]; i; i=edge[i].next)
if (edge[i].to!=son[now] && edge[i].to!=fa[now])
dfs_2(edge[i].to,edge[i].to);
pr[now]=sz;
}
//--------------------------------------------------------------------------------------
int tree[maxn<<],delc[maxn<<],dela[maxn<<];
inline void update(int now){tree[now]=max(tree[now<<],tree[now<<|]);}
inline void pushdown(int now)
{
if (delc[now]!=-)
{
int cc=delc[now]; delc[now]=-;
delc[now<<]=cc; dela[now<<]=; tree[now<<]=cc;
delc[now<<|]=cc; dela[now<<|]=; tree[now<<|]=cc;
}
if (dela[now])
{
int aa=dela[now]; dela[now]=;
dela[now<<]+=aa; tree[now<<]+=aa;
dela[now<<|]+=aa; tree[now<<|]+=aa;
}
}
void build(int now,int l,int r)
{
delc[now]=-; dela[now]=;
if (l==r) {tree[now]=va[l];return;}
int mid=(l+r)>>;
build(now<<,l,mid); build(now<<|,mid+,r);
update(now);
}
void point_change(int now,int l,int r,int loc,int val)
{
if (l==r) {tree[now]=val;return;}
int mid=(l+r)>>;pushdown(now);
if (loc<=mid) point_change(now<<,l,mid,loc,val);
else point_change(now<<|,mid+,r,loc,val);
update(now);
}
void segment_change(int now,int l,int r,int L,int R,int val)
{
if (L<=l && R>=r) {tree[now]=val;delc[now]=val;dela[now]=;return;}
int mid=(l+r)>>;pushdown(now);
if (L<=mid) segment_change(now<<,l,mid,L,R,val);
if (R>mid) segment_change(now<<|,mid+,r,L,R,val);
update(now);
}
void segment_add(int now,int l,int r,int L,int R,int val)
{
if (L<=l && R>=r) {tree[now]+=val;dela[now]+=val;return;}
int mid=(l+r)>>;pushdown(now);
if (L<=mid) segment_add(now<<,l,mid,L,R,val);
if (R>mid) segment_add(now<<|,mid+,r,L,R,val);
update(now);
}
int segment_ask(int now,int l,int r,int L,int R)
{
if (L<=l && R>=r) return tree[now];
int mid=(l+r)>>; int ans=;pushdown(now);
if (L<=mid) ans=max(segment_ask(now<<,l,mid,L,R),ans);
if (R>mid) ans=max(ans,segment_ask(now<<|,mid+,r,L,R));
return ans;
}
//--------------------------------------------------------------------------------------
void Change(int x,int val)
{
point_change(,,n,pl[id[x]],val);
}
void Cover(int x,int y,int val)
{
while (top[x]!=top[y])
{
if (deep[top[x]]<deep[top[y]]) swap(x,y);
segment_change(,,n,pl[top[x]],pl[x],val);
x=fa[top[x]];
}
if (deep[x]>deep[y]) swap(x,y);
if (x!=y) segment_change(,,n,pl[x]+,pl[y],val);//这是有别于平常的判断
}
void Add(int x,int y,int val)
{
while (top[x]!=top[y])
{
if (deep[top[x]]<deep[top[y]]) swap(x,y);
segment_add(,,n,pl[top[x]],pl[x],val);
x=fa[top[x]];
}
if (deep[x]>deep[y]) swap(x,y);
if (x!=y) segment_add(,,n,pl[x]+,pl[y],val);
}
void Max(int x,int y)
{
int ans=;
while (top[x]!=top[y])
{
if (deep[top[x]]<deep[top[y]]) swap(x,y);
ans=max(ans,segment_ask(,,n,pl[top[x]],pl[x]));
x=fa[top[x]];
}
if (deep[x]>deep[y]) swap(x,y);
if (x!=y) ans=max(ans,segment_ask(,,n,pl[x]+,pl[y]));
printf("%d\n",ans);
}
//--------------------------------------------------------------------------------------
int main()
{
// freopen("msn.in","r",stdin);
// freopen("msn.out","w",stdout);
n=read();
for (int i=; i<=n-; i++)
uu[i]=read(),vv[i]=read(),ww[i]=read(),insert(uu[i],vv[i],ww[i]);
dfs_1(); dfs_2(,);
for (int i=; i<=n-; i++)
if (deep[uu[i]]<deep[vv[i]]) id[i]=vv[i]; else id[i]=uu[i];
for (int i=; i<=n-; i++) va[pl[id[i]]]=ww[i];//边权下放至点,记录所下放的点的id
build(,,n);
while (true)
{
char opt[]; scanf("%s",opt);if(opt[]=='t')break;
int u=read(),v=read(),w;
switch (opt[])
{
case 'a': Max(u,v);break;
case 'o': w=read();Cover(u,v,w);break;
case 'd': w=read();Add(u,v,w);break;
case 'h': Change(u,v);break;
}
}
return ;
}

找DCrusher和Char哥帮我调过...无果,结果还得靠自己...然后发现又是二逼错误...