码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj3730. 震波

时间:2019-02-06 15:47:17      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:nbsp   ace   code   getchar   代码   大于   int   continue   for   

动态点分治

考虑从每一个"块"里找到距离k范围内的点的和

为了去重,

每个x维护两个线段树:(都是关于自己分治树子树的点)

1.下标为距离x的距离,权值为val的

2.下标为距离x的分治树father的距离,权值为val

 

这样,统计的时候

计算分治树祖先块的时候,把从自己那里出来的块的东西再减去

 

注意:

1.点分治nowrt是全局变量,所以divi这一层用tmp额外记录一下,否则回溯回来再下去的时候,nowrt就不是这一层的了

2.不能某个祖先的dis大于k,就直接break了。因为到分治树祖先的距离没有单调性!!!!

 

代码:

#include<bits/stdc++.h>
#define il inline
#define reg register int
#define numb (ch^‘0‘)
#define mid ((l+r)>>1)
using namespace std;
typedef long long ll;
il void rd(int &x){
    char ch;bool fl=false;
    while(!isdigit(ch=getchar()))(ch==-)&&(fl=true);
    for(x=numb;isdigit(ch=getchar());x=x*10+numb);
    (fl==true)&&(x=-x);
}
namespace Miracle{
const int N=100000+5;
int n,m;
struct node{
    int nxt,to;
}e[2*N];
int hd[N],cnt;
void add(int x,int y){
    e[++cnt].nxt=hd[x];
    e[cnt].to=y;
    hd[x]=cnt;
}
int fa[N];
int sz[N];
int mx[N];
int nowsz;
int tmprt;
int rt[N][2];//0 is myself ; 1 is disfather
int nowrt;
struct tr{
    int ls,rs;
    int sum;
}t[N*50];
int tot;
int dist[N][20];
int dep[N];
int val[N];
int vis[N];
void pushup(int x){
    t[x].sum=t[t[x].ls].sum+t[t[x].rs].sum;
}
void upda(int &x,int l,int r,int p,int v){
    if(!x) x=++tot;
    if(l==r){
        t[x].sum+=v;
        return;
    }
    if(p<=mid) upda(t[x].ls,l,mid,p,v);
    else upda(t[x].rs,mid+1,r,p,v);
    pushup(x);
}
int query(int x,int l,int r,int L,int R){//no dis=0 (myself)
    if(L>R) return 0;
    if(L<=l&&r<=R){
        return t[x].sum;
    }
    int ret=0;
    if(L<=mid) ret+=query(t[x].ls,l,mid,L,R);
    if(mid<R) ret+=query(t[x].rs,mid+1,r,L,R);
    return ret;
}
void dfs1(int x,int f,int d,int dis){//find nowrt
    sz[x]=1;
    mx[x]=0;
    dep[x]=d;
    if(d!=1){
        upda(tmprt,0,n,dis,val[x]);
    }
    for(reg i=hd[x];i;i=e[i].nxt){
        int y=e[i].to;
        if(vis[y]||y==f) continue;
        dfs1(y,x,d,dis+1);
        sz[x]+=sz[y];
        mx[x]=max(mx[x],sz[y]);
    }
    mx[x]=max(mx[x],nowsz-sz[x]);
    if(mx[x]<=nowsz/2&&(nowsz-sz[x])<=nowsz/2){
        nowrt=x;
    }
}
void dfs2(int x,int f,int d,int dis){//find dis from nowrt
    sz[x]=1;
    dep[x]=d;
    dist[x][d]=dis;
    upda(rt[nowrt][0],0,n,dis,val[x]);
    for(reg i=hd[x];i;i=e[i].nxt){
        int y=e[i].to;
        if(vis[y]||y==f) continue;
        dfs2(y,x,d,dis+1);
        sz[x]+=sz[y];
    }
}
void divi(int x,int d,int f){
    //mxsz=0;
//    cout<<" x "<<x<<" : "<<d<<" "<<f<<endl;
    nowrt=0;
    tmprt=0;
    dfs1(x,0,d,1);
    fa[nowrt]=f;
    rt[nowrt][1]=tmprt;
//    cout<<" nowrt "<<nowrt<<endl;
    dfs2(nowrt,0,d,0);
    
    vis[nowrt]=1;
    int tmp=nowrt;
    for(reg i=hd[tmp];i;i=e[i].nxt){
        int y=e[i].to;
        if(!vis[y]){
            nowsz=sz[y];
            divi(y,d+1,tmp);
        }
    }
}
int wrk0(int x,int k){
    int ret=0;
    ret+=query(rt[x][0],0,n,0,k);
    int tmp=x;
    while(fa[x]){
        int y=fa[x];
    //    if(k-dist[tmp][dep[y]]>=0) 
        ret+=query(rt[y][0],0,n,0,k-dist[tmp][dep[y]]);
    //    else break;
        ret-=query(rt[x][1],0,n,0,k-dist[tmp][dep[y]]);
        
        x=y;
    }
    return ret;
}
void wrk1(int x,int v){
    int tmp=x;
    int c=v-val[x];
    val[x]=v;
    upda(rt[x][0],0,n,0,c);
    while(fa[x]){
        int y=fa[x];
        upda(rt[x][1],0,n,dist[tmp][dep[y]],c);
        upda(rt[y][0],0,n,dist[tmp][dep[y]],c);
        x=y;
    }
}
int main(){
    rd(n);rd(m);
    for(reg i=1;i<=n;++i) rd(val[i]);
    int x,y;
    for(reg i=1;i<n;++i){
        rd(x);rd(y);
        add(x,y);add(y,x);
    }
    nowsz=n;
    divi(1,1,0);
//    cout<<" toot "<<tot<<endl;
    int op,k;
    int lasans=0;
    while(m--){
        rd(op);
        rd(x);rd(k);
        x^=lasans;k^=lasans;
        if(op==0) printf("%d\n",lasans=wrk0(x,k));
        else wrk1(x,k);
    }
    return 0;
}
 
}
signed main(){
//    freopen("data.in","r",stdin);
//    freopen("my.out","w",stdout);
    Miracle::main();
    return 0;
}

/*
   Author: *Miracle*
   Date: 2019/2/6 10:11:29
*/

总结:

动态点分治往往为了去重,经常考虑x对fa的贡献

相当于在分治树的往fa走的边上造了一个线段树

(每个点一个,每个边一个,一共n+n-1个)

bzoj3730. 震波

标签:nbsp   ace   code   getchar   代码   大于   int   continue   for   

原文地址:https://www.cnblogs.com/Miracevin/p/10353703.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!