[HDU4918]点分治+树状数组

Query on the subtree

叉姐的题 看了看聊天记录发现自己光调就是至少两个多小时...不过慢慢来吧 严格意义上讲是自从上学期final week以来复健的第一道大数据结构题...其实寒假因为突击驾照的缘故也荒废掉很多时间...

题意其实很简明扼要: 两种操作 改变树上某个点的权值 或者 询问在某个点d距离范围内的所有权值和

其实不论考不考虑权值 这道题对我来说都是不好写的 毕竟权值在这里就是要求一个树状数组而已

最大的困难还是来自于 要层层统计root贡献的时候 去重以及正确记录当前点与rt的距离!

MARK几个我wa的地方(在拍std的情况下改的还算顺利):

add&sum头一昏双双写错

内存超限/不够 据说subrt是要开nlogn的 不过100w将将也能过吧

建立一套新边表的时候时刻注意变量名...

是divide(root)而不是divide(v)

给vector init深度的时候保险起见还是要resize sum...毕竟重心只由个数决定...与深度没有绝对关系...

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#define mo 1000000007
#define N 202020
#define M 4004040
#define ll long long
using namespace std;
int edgenum,head[N],jie[N],maxx[M],vet[N],size[N];
int edgenew,headnew[N],jienew[M],vetnew[M],pri[M],id[M],maxd[N],disrt[N];
int n,Q,nowd,sum,root,f[N],flag[N],a[N],nows;
struct Tree{
    int n;
    vector<int>t;
    void init(int size){
        t.clear();
        n=size;
        for(int i=0;i<=n;i++)t.push_back(0);
    }
    void add(int k,int x){
        for(int i=k;i<=n;i+=i&(-i))t[i]+=x;
    }
    int sum(int k){
        int ans=0;
        if(k<=0)return 0;
        for(int i=k;i>0;i-=i&(-i))ans+=t[i];
        //printf("%d %d\n",k,ans);
        return ans;
    }
};
Tree t[N];Tree subt[1001010];
void getroot(int u,int ff,int dep){
    int e=head[u];size[u]=1;f[u]=0;
    nowd=max(nowd,dep);
    while(e>0){
        int v=vet[e];
        if(v!=ff && flag[v]==0)
        {getroot(v,u,dep+1);size[u]+=size[v];f[u]=max(f[u],size[v]);}
        e=jie[e];
    }
    f[u]=max(f[u],sum-size[u]);
    if(f[u]<f[root])root=u;
}
void add(int u,int v){edgenum++;vet[edgenum]=v;jie[edgenum]=head[u];head[u]=edgenum;}
void addnew(int u,int anc,int subanc,int w,int ww){
    //printf("=======%d %d %d %d\n",anc,subanc,u,ww);
    edgenew++;vetnew[edgenew]=anc;pri[edgenew]=ww;
    jienew[edgenew]=headnew[u];
    headnew[u]=edgenew;
    id[edgenew]=w;
}
void ins(int anc,int u,int ff,int dep){
    int e=head[u];disrt[u]=dep;
    t[anc].add(dep,a[u]);
    //printf("`````````````%d %d %d\n",anc,u,dep);
    while(e>0){
        int v=vet[e];
        if(v!=ff && flag[v]==0){
            ins(anc,v,u,dep+1);
        }
        e=jie[e];
    }
}
void insnew(int anc,int u,int subroot,int id,int ff,int dep){
    int e=head[u];
    subt[id].add(dep,a[u]);
    addnew(u,anc,subroot,id,disrt[u]);
    //printf("aaaaaaaaa====%d %d %d %d \n",anc,u,id,dep);
    while(e>0){
        int v=vet[e];
        if(v!=ff && flag[v]==0){
            insnew(anc,v,subroot,id,u,dep+1);
        }
        e=jie[e];
    }
}
void divide(int u){
    int e=head[u];flag[u]=1;
    t[u].init(sum+1);
    maxd[u]=sum;
    while(e>0){
        int v=vet[e];
        if(flag[v]==0){
            ins(u,v,0,1);    
        }
        e=jie[e];
    }
    e=head[u];
    while(e>0){
        int v=vet[e];
        if(flag[v]==0){
            sum=size[v];
            f[root]=n+1;getroot(v,0,0);
           //printf("..........%d %d %d\n",u,root,nowd);
            nows++;
            maxx[nows]=sum;
            subt[nows].init(sum+1);
            //addnew(root,u,nows,disrt[root]);
            insnew(u,v,root,nows,0,1);
            divide(root);
        }
        e=jie[e];
    }
    return;
}
int main(){
    
    while(scanf("%d%d",&n,&Q)!=EOF){
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        edgenum=0;edgenew=0;nows=0;
        memset(flag,0,sizeof(flag));
        memset(head,0,sizeof(head));
        memset(headnew,0,sizeof(headnew));
        memset(maxd,0,sizeof(maxd));
        memset(maxx,0,sizeof(maxx));
        for(int i=1;i<n;i++){
            int u,v;
            scanf("%d%d",&u,&v);
            add(u,v);add(v,u);
        }
        root=0;f[root]=n+1;sum=n;
        nowd=0;
        getroot(1,0,0);int alr=root;
        divide(root);
        while(Q--){
            char str[10];int u,d;
            scanf("%s%d%d",str,&u,&d);
            if(str[0]=='!'){
                int e=headnew[u],dis1=0;
                while(e>0){
                    int v=vetnew[e],ii=id[e];dis1=pri[e];
                    t[v].add(dis1,d-a[u]);
                    subt[ii].add(dis1,d-a[u]);
                    e=jienew[e];
                }
                a[u]=d;
            }else{
                int e=headnew[u],ee=e,dis1=0,ans=a[u]+t[u].sum(min(d,maxd[u]));
                //printf("%d %d ???????%d\n",u,maxd[u],t[3].sum(1));
                //printf("??????/%d\n",ans);
                while(e>0){
                    int v=vetnew[e],ii=id[e];dis1=pri[e];
                    if(d>dis1)ans+=t[v].sum(min(maxd[v],d-dis1));
                    if(d>=dis1)ans+=a[v];
                    if(d>dis1)ans-=subt[ii].sum(min(maxx[ii],d-dis1));
                    //printf("====%d %d %d\n",v,maxx[ii],dis1);
                    e=jienew[e];
                }
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}

过段时间要去找claris的题刷 明天先把dp题补完然后把Legilimens去年的模拟题更掉