动态点分治练习1——P3345 [ZJOI2015]幻想乡战略游戏

蒟蒻的垂死挣扎

动态点分治好难。。。。

动态点分治支持在原树中修改，并且可以多组询问。它基于每次找重心时，将这些重心重新构成一颗点分树，全局的重心为点分树的根，它的子树的重心是他的儿子，以此类推。这样我们就得到了一个树高为log级别的优秀的树，它就可以支持一些例如暴跳父亲的操作，从而达到优秀的复杂度。

一般解决的问题是 求全局对一个点的影响 的题。还是需要对具体题目进行分析。

对于这一题来说，构出点分树之后我们直接在这棵树上进行修改和统计答案，维护自己子树的贡献 自己子树对父亲的贡献 和 路径长度，这样就可以在每次修改时保证答案的正确性。而对于答案的查找，我们可以从根节点出发，计算子树的答案值，如果子树更优则跳到子树上继续寻找，可以证明更优的子树最多只有一个，所以这样就可以优秀地寻找到正确答案。

我算出来时间有三只log ，但是极其不满。

 

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<set>
#define ll long long
#define MOD 998244353
#define N 100100
#define M 1001000
#define RG register

using namespace std;

inline ll read(){
    RG ll x=0,o=1; RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
    if(ch=='-') o=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=((x<<3)+(x<<1))+ch-'0',ch=getchar();
    return x*o;
}

vector<int> V[N],VR[N];
int n,siz[N],son[N],first[N],Fa[N],fa[N],dep[N],top[N],tot,root,RT,Max,Siz,vis[N];
ll w[N],tw[N],sum[N],dis[N];
struct mona { int nxt,en,w; } s[N<<1];
inline void Insert(int x,int y,int w) { s[++tot]=(mona) { first[x],y,w },first[x]=tot; }
inline void Dfs1(int k,int f,ll Dis){
    dis[k]=Dis,dep[k]=dep[f]+1,fa[k]=f,siz[k]=1; int num=0;
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(en==f) continue ;
        Dfs1(en,k,Dis+s[i].w),siz[k]+=siz[en];
        if(siz[en]>num) num=siz[en],son[k]=en;
    }
}
inline void Dfs2(int k,int to){
    top[k]=to; if(son[k]) Dfs2(son[k],to);
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(en==fa[k]||en==son[k]) continue ;
        Dfs2(en,en);
    }
}
inline int Lca(int x,int y){
    while(top[x]!=top[y]){
        if(dep[top[y]]>dep[top[x]]) swap(x,y);
        x=fa[top[x]];
    }   if(dep[x]<dep[y]) return x; else return y;
}
inline ll Dis(int x,int y){
    return dis[x]+dis[y]-(dis[Lca(x,y)]<<1);
}

inline void Getroot(int k,int f){
    int mmax=0; siz[k]=1;
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(en==f||vis[en]) continue ;
        Getroot(en,k),siz[k]+=siz[en],mmax=max(mmax,siz[en]);
    }   mmax=max(mmax,Siz-siz[k]);
    if(mmax<Max) Max=mmax,root=k;
}
inline void Divide(int k,int f){
    Fa[k]=f,vis[k]=1; int Copy=Siz;
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(en==f||vis[en]) continue ;
        Siz= siz[k]>siz[en]?siz[en]:Copy-siz[k] ;
        Max=1e9,root=0,Getroot(en,0),
            VR[k].push_back(en),V[k].push_back(root),Divide(root,k);
    }
}

inline void Modify(int k,int val){
    sum[k]+=val;
    for(RG int i=k,f=Fa[i];Fa[i];i=Fa[i],f=Fa[i]){
        RG int Ds=Dis(f,k);
        sum[f]+=val,tw[i]+=1LL*val*Ds,w[f]+=1LL*val*Ds;
    }   return ;
}
inline ll Count(int k){
    RG ll ans=w[k];
    for(RG int i=k,f=Fa[i];Fa[i];i=Fa[i],f=Fa[i]){
        ans+=w[f]-tw[i]+(sum[f]-sum[i])*Dis(k,f);
    }   return ans;
}
inline ll Query(int k){
    RG int zzq=1; RG ll ans;
    while(zzq){ zzq=0,ans=Count(k); RG int sz=VR[k].size();
        for(RG int i=0;i<sz;++i){
            ll now=Count(VR[k][i]);
            if(now<ans) { zzq=1,k=V[k][i]; break ; }
        }
    }   return ans;
}

int main(){
    n=read(); RG int Q=read();
    for(RG int i=1;i<n;++i){
        int x=read(),y=read(),w=read();
        Insert(x,y,w),Insert(y,x,w);
    }   Dfs1(1,0,0),Dfs2(1,1);
    Max=1e9,Siz=n,Getroot(1,0),RT=root,Divide(root,0);
    while(Q--){
        int x=read(),y=read();
        Modify(x,y); printf("%lld\n",Query(RT));
    }
}