1984: 月下“毛景树”
Time Limit: 20 Sec Memory Limit: 64 MBSubmit: 1433 Solved: 449
[ 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
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
16
【Data Range】
1<=N<=100,000,操作+询问数目不超过100,000。
保证在任意时刻,所有树枝上毛毛果的个数都不会超过10^9个。
解题思路:线段树都不会打了。。。
将边权转到下面的点上,因为每个点对应的父亲只有一条,然后树链剖分,查询
的时候,要注意顶不要算进去。线段树,坑啊。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
int n,len,cnt,ans,lx,rx,change,adg;
int to[210000],next[210000],h[210000],zhi[210000],bian[210000],zong[210000];
int deep[110000],f[110000],size[110000],son[110000],xvier[110000];
int messi[500000],dui[110000],top[110000],set[500000],ma[500000],add[500000];
char c[100];
inline int read()
{
char y;int x=0,f=1; y=getchar();
while (y<'0' || y>'9') {if (y=='-') f=-1; y=getchar();}
while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
return x*f;
}
void insert(int x,int y,int z,int d)
{
++len; to[len]=y; next[len]=h[x]; h[x]=len; zhi[len]=z; bian[len]=d;
}
void dfs(int now,int fa,int dep)
{
deep[now]=dep; f[now]=fa; size[now]=1; son[now]=0; int mx=-1;
int u=h[now];
while (u!=0)
{
if (to[u]!=fa)
{
xvier[to[u]]=zhi[u]; zong[bian[u]]=to[u];
dfs(to[u],now,dep+1);
size[now]+=size[to[u]];
if (size[to[u]]>mx) mx=size[to[u]],son[now]=to[u];
}
u=next[u];
}
}
void dfs1(int now,int fa,int tl)
{
++cnt; messi[cnt]=xvier[now]; dui[now]=cnt;
if (tl==-1) tl=now; top[now]=tl;
if (son[now]==0) return;
dfs1(son[now],now,tl);
int u=h[now];
while (u!=0)
{
if (to[u]!=fa && to[u]!=son[now])
{
dfs1(to[u],now,-1);
}
u=next[u];
}
}
void build(int now,int l,int r)
{
set[now]=-1; add[now]=0;
if (l==r)
{
ma[now]=messi[l]; return;
}
int mid=(l+r)/2;
build(now*2,l,mid);
build(now*2+1,mid+1,r);
ma[now]=max(ma[now*2],ma[now*2+1]);
}
void pushdown(int now,int l,int r)
{
if (set[now]!=-1 && l!=r)
{
set[now*2]=set[now*2+1]=ma[now*2]=ma[now*2+1]=set[now];
add[now*2]=add[now*2+1]=0;
}
set[now]=-1;
if (l!=r && add[now]!=0)
{
add[now*2]+=add[now]; add[now*2+1]+=add[now];
ma[now*2]+=add[now]; ma[now*2+1]+=add[now];
}
add[now]=0;
}
void query(int now,int l,int r)
{
pushdown(now,l,r);
if (set[now]!=-1)
{
ans=max(ans,set[now]); return;
}
if (lx<=l && r<=rx)
{
ans=max(ans,ma[now]);
return;
}
int mid=(l+r)/2;
if (lx<=mid) query(now*2,l,mid);
if (mid+1<=rx) query(now*2+1,mid+1,r);
}
void maindown(int now,int l,int r)
{
if (l!=r) ma[now]=max(ma[now*2],ma[now*2+1]);
}
void add_(int now,int l,int r)
{
pushdown(now,l,r);
if (lx<=l && r<=rx)
{
add[now]+=adg; ma[now]+=adg; return;
}
int mid=(l+r)/2;
if (lx<=mid) add_(now*2,l,mid);
if (mid+1<=rx) add_(now*2+1,mid+1,r);
maindown(now,l,r);
}
void change_(int now,int l,int r)
{
pushdown(now,l,r);
if (lx<=l && r<=rx)
{
set[now]=change; ma[now]=change; add[now]=0; return ;
}
int mid=(l+r)/2;
if (lx<=mid) change_(now*2,l,mid);
if (mid+1<=rx) change_(now*2+1,mid+1,r);
maindown(now,l,r);
}
int main()
{
n=read();
for (int i=1;i<=n-1;++i)
{
int u,v,z;
u=read(); v=read(); z=read();
insert(u,v,z,i); insert(v,u,z,i);
}
dfs(1,0,1);
cnt=0;
dfs1(1,0,-1);
build(1,1,n);
scanf("%s",c);
while (c[0]!='S')
{
if (c[0]=='M')
{
int u,v; u=read(); v=read(); ans=-1;
while (u!=v)
{
if (top[u]!=top[v])
{
if (deep[top[u]]>deep[top[v]]) lx=dui[top[u]],rx=dui[u],u=f[top[u]];else
lx=dui[top[v]],rx=dui[v],v=f[top[v]];
}else
{
if (deep[u]>deep[v]) lx=dui[v]+1,rx=dui[u];else
lx=dui[u]+1,rx=dui[v];
u=v;
}
query(1,1,n);
}
printf("%d\n",ans);
}else
if (c[0]=='A')
{
int u,v; u=read(); v=read(); adg=read();
while (u!=v)
{
if (top[u]!=top[v])
{
if (deep[top[u]]>deep[top[v]]) lx=dui[top[u]],rx=dui[u],u=f[top[u]];else
lx=dui[top[v]],rx=dui[v],v=f[top[v]];
}else
{
if (deep[u]>deep[v]) lx=dui[v]+1,rx=dui[u];else
lx=dui[u]+1,rx=dui[v];
u=v;
}
add_(1,1,n);
}
}else
if (c[0]=='C')
{
if (c[1]=='h')
{
int k=read(); change=read(); lx=dui[zong[k]]; rx=lx;
change_(1,1,n);
}else
{
int u,v; u=read(); v=read(); change=read();
while (u!=v)
{
if (top[u]!=top[v])
{
if (deep[top[u]]>deep[top[v]]) lx=dui[top[u]],rx=dui[u],u=f[top[u]];else
lx=dui[top[v]],rx=dui[v],v=f[top[v]];
}else
{
if (deep[u]>deep[v]) lx=dui[v]+1,rx=dui[u];else
lx=dui[u]+1,rx=dui[v];
u=v;
}
change_(1,1,n);
}
}
}
scanf("%s",c);
}
}