[BZOJ Contest-2017省队十连测推广赛1·T1][BZOJ4765][分块][dfs序]普通计算姬

考虑对1~n的序列分块

令每个点在序列中的贡献 ai i 为根的子树的权值和,那么分块后可以求出块j的贡献和 sumj ,那么对于一个询问 [l,r] ,这个区间里完整的块的答案可以扫一遍 sum 数组求出来,这样复杂度是 O(n) ,剩下的点暴力求出这个点单个的贡献。

怎么求单个点的贡献呢
可以用dfs序来维护子树的权值和,那么只用在dfs序上用树状数组来维护就行了,这样是 O(nlogn)

还有就是修改,先在dfs序上的树状数组中进行修改(因为询问时要用树状数组求单个点的贡献),然后就要对每个块进行修改。
因为是单点修改,所以每次修改只会对它的祖先(包括他自己)的贡献造成影响,用 f[i][j] 表示第 j 个块中是 i 的祖先的个数, f[i][j] 可以用dfs加个桶在 O(nn) 的时间内算出来。
那么修改点 i 的权值为 x ,对于块 j 的影响就是 f[i][j](xA[i]) ,其中 A[i] 为点 i 修改前的权值。
这样修改操作也可以在 O(n+logn) 中完成啦

注意要开unsigned long long

#include 
#include 
#include 
#include 
#define N 100010
#define M 320

using namespace std;

typedef unsigned long long ll;

int n,m,cnt,rt,block,blockm,g,op,u,v;
int t[M],G[N],L[N],R[N],b[N];
int f[N][M];
ll bit[N<<1],sum[M],A[N];
ll Ans;
struct edge{
  int t,nx;
}E[N<<2];

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

inline void reaD(ll &x){
  char c=nc(); x=0;
  for(;c>57||c<48;c=nc());for(;c>=48&&c<=57;x=x*10+c-48,c=nc());
}

inline void reaD(int &x){
  char c=nc(); x=0;
  for(;c>57||c<48;c=nc());for(;c>=48&&c<=57;x=x*10+c-48,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;
}

inline void add(int x,int w){
  for(;x;x-=x&-x) bit[x]+=w;
}

inline ll query(int x){
  ll r=0;
  for(;x<=g;x+=x&-x) r+=bit[x];
  return r;
}

void dfs(int x,int p){
  t[b[x]]++;
  for(int i=1;i<=blockm;i++) f[x][i]=t[i];
  L[x]=++g; add(L[x],A[x]);
  for(int i=G[x];i;i=E[i].nx)
    if(E[i].t!=p) dfs(E[i].t,x);
  R[x]=++g; t[b[x]]--;
}

int w[30],wt;

inline void PutAns(ll x){
  if(x==0){putchar(48);putchar('\n');return ;}
  while(x) w[++wt]=x%10,x/=10;
  for(;wt;wt--) putchar(w[wt]+48); putchar('\n');
}

int main(){
  reaD(n); reaD(m); block=(int)sqrt(n); blockm=(n-1)/block+1;
  for(int i=1;i<=n;i++) reaD(A[i]);
  for(int i=1;i<=n;i++) b[i]=(i-1)/block+1;
  for(int i=1;i<=n;i++){
    reaD(u); reaD(v);
    if(u==0) rt=v;
    else Insert(u,v);
  }
  dfs(rt,0);
  for(int i=1;i<=n;i++)
    sum[b[i]]+=query(L[i])-query(R[i]);
  for(int i=1;i<=m;i++){
    reaD(op); reaD(u); reaD(v);
    if(op==1){
      add(L[u],v-A[u]);
      for(int j=1;j<=blockm;j++) sum[j]+=1ll*f[u][j]*(v-A[u]);
      A[u]=v;
    }
    else{
      Ans=0;
      if(b[u]==b[v])
    for(int j=u;j<=v;j++) Ans+=query(L[j])-query(R[j]);
      else{
    for(int j=b[u]+1;j<=b[v]-1;j++) Ans+=sum[j];
    for(int j=u;b[j]==b[u]&&j<=n;j++) Ans+=query(L[j])-query(R[j]);
    for(int j=v;b[j]==b[v]&&j;j--) Ans+=query(L[j])-query(R[j]);
      }
      //printf("%llu\n",Ans);
      PutAns(Ans);
    }
  }
  return 0;
}

你可能感兴趣的:(分块,&,阈值,dfs序)