bzoj 1984(线段树+树链剖分)

1984: 月下“毛景树”

Time Limit: 20 Sec   Memory Limit: 64 MB
Submit: 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

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个。

解题思路：线段树都不会打了。。。

将边权转到下面的点上，因为每个点对应的父亲只有一条，然后树链剖分，查询

的时候，要注意顶不要算进去。线段树，坑啊。

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