[BZOJ3730][点分树][树状数组]震波

直接建出点分树,树上搞个线段树什么的就可以了……
然而我常数丑……就用树状数组了……
代码也很丑

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#define N 200010

using namespace std;

int n,m,cnt,Size,Max,tot;
int A[N],G[N],vis[N],rt[N],p[N],prt[N],dpt[N],s[N];
int fa[N][25];
int *B[N],*pB[N];
int por[N*100],por1[N*100];
struct edge{
  int t,nx;
}E[N<<1];

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void rea(int &x){
  char c=nc(); x=0;
  for(;c>'9'||c<'0';c=nc());for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc());
}

inline void Insert(int x,int y){
  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;
  E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt;
}

int explore(int x,int f,int d){
  int ret=1;
  for(int i=G[x];i;i=E[i].nx)
      if(E[i].t!=f&&!vis[E[i].t]) ret+=explore(E[i].t,x,d+1);
  return ret;
}

int Root(int x,int f,int &k){
  int s=1,iMax=0;
  for(int i=G[x];i;i=E[i].nx)
      if(E[i].t!=f&&!vis[E[i].t]){
          int w=Root(E[i].t,x,k);
          if(iMax<w) iMax=w;
          s+=w;
      }
  if(iMax<Size-s) iMax=Size-s;
  if(iMax<Max) Max=iMax,k=x;
  return s;
}

void divide(int x,int f){
  Size=explore(x,0,0);
  Max=1<<30; int k;Root(x,0,k); p[k]=f;
  vis[k]=1; s[k]=Size+2;
  for(int i=G[k];i;i=E[i].nx)
      if(!vis[E[i].t]) divide(E[i].t,k);
}

void dfs(int x,int f){
  dpt[x]=dpt[f]+1; fa[x][0]=f;
  for(int i=1;i<=16;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
  for(int i=G[x];i;i=E[i].nx)
      if(E[i].t!=f) dfs(E[i].t,x);
}

inline int LCA(int x,int y){
  if(dpt[x]<dpt[y]) return LCA(y,x);
  for(int i=16;~i;i--)
      if(dpt[fa[x][i]]>=dpt[y]) x=fa[x][i];
  if(x==y) return x;
  for(int i=16;~i;i--)
      if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
  return fa[x][0];
}

inline int dis(int x,int y){
  return dpt[x]+dpt[y]-2*dpt[LCA(x,y)];
}

inline int query(int *A,int x){
    int ret=0;
    for(;x>0;x-=x&-x) ret+=A[x];
    return ret;
}

inline int query(int* A,int r,int t){
    r=min(r+1,t);
    return query(A,r);
}

inline int Query(int x,int y){
  int ret=query(B[x],y,s[x]);
  for(int i=x;p[i];i=p[i]){
      int d=dis(x,p[i]);
      if(d>y) continue;
      ret+=query(B[p[i]],y-d,s[p[i]])-query(pB[i],y-d,s[i]);
  }
  return ret;
}

inline void Add(int *A,int x,int y,int t){
    x=min(x+1,t);
    if(x==0) return ;
    for(;x<=t;x+=x&-x) A[x]+=y;
}

inline void add(int x,int y){
  Add(B[x],0,y,s[x]);
  for(int i=x;p[i];i=p[i]){
      int d=dis(x,p[i]);
      Add(pB[i],d,y,s[i]);
      Add(B[p[i]],d,y,s[p[i]]);
  }
}

int main(){
  rea(n); rea(m);
  for(int i=1;i<=n;i++) rea(A[i]);
  for(int i=1;i<n;i++){
      int x,y; rea(x); rea(y);
      Insert(x,y);
  }
  divide(1,0); dfs(1,0); int Ans=0;
  int cnt0=0;
  for(int i=1;i<=n;i++){
    B[i]=por+cnt0; pB[i]=por1+cnt0; cnt0+=s[i]+2;
    //for(int j=0;j<s[i];j++) B[i][j]=pB[i][j]=0;
    }
  for(int i=1;i<=n;i++) add(i,A[i]);
  for(int i=1;i<=m;i++){
      int op,x,y; rea(op); rea(x); rea(y);
      x^=Ans; y^=Ans;
      if(op==0) printf("%d\n",Ans=Query(x,y));
      else add(x,y-A[x]),A[x]=y;
  }
  return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值