Live2d Test Env

luoguU60884 【模板】动态点分治套线段树

题目连接:https://www.luogu.org/problemnew/show/U60884

题意:N个点,标号为1∼N,用N1条双向带权通道连接,保证任意两个点能互相到达。

Q次询问,问从编号为x的点到达标号L∼R的点其中一个点的最小距离是多少。

说明 :N,Q<1e5,边权<1e4;

 

思路:不难想到点分树,保存每个点到其“负责”的点的距离,这样的话可以套线段树,线段树保存其他点到点的距离。

但是,点分树上有个需要解决的问题是:如果x顺着点分树向父亲走,那么在父亲保存的线段树中要除去从儿子上来的那一部分(否则的话,不是简单路径,求出来的可能会错)。 想了一下这里很难实现,因为是取min操作,所以线段树上二分估计也不行。 所以卡住了。

然后求问群友,群友提到了虚树,我觉得此题的数据需要没法实现。  后面猛地想通,我们不需要考虑“回走”这种情况,因为我们求的是最小距离,而非简单路径肯定是没有简单路径优的。   那么就是果然如标题所说,是个板子题了。

代码实现:点分树+线段树+树剖求LCA。

点分树:点分治的过程中新建的树,树根是第一次找到是重心,每一层的重心与上一层的重心连边得到点分树。

点分树里保存的是自己作为重心时,会“负责”的点,即此时还没有vis过的,且与自己连通的点。 

所有经过“x”到达的点,保存在了两部分里:一是x在点分树里保存的信息。二是x在点分树的祖先里保存的信息(这一部分保留了x向上传递是信息,所以大部分题要把这里除去,此题由于是取min,可以不考虑)。 

求LCA:开始用了ST表,但是感觉空间耗费太大,就改为了树剖,跑起来还挺快。

#include<bits/stdc++.h>
#define rep(i,w,v) for(int i=w;i<=v;i++)
#define FOR() for(int i=Laxt[u];i;i=Next[i])
using namespace std;
#define ll long long
#define RG register
#define maxn 100010
#define maxm 200010
#define inf 1e9
int fa[maxn],n,cnt;
int dep[maxn];bool vis[maxn];
int Laxt[maxn],Next[maxm],To[maxm],Len[maxm];
int siz[maxn],fcy[maxn],hson[maxn],Top[maxn];
void add(int u,int v,int len)
{
    Next[++cnt]=Laxt[u]; Laxt[u]=cnt; To[cnt]=v; Len[cnt]=len;
}
void dfs1(int u,int ff)
{
    fa[u]=ff; siz[u]=1;
    dep[u]=dep[ff]+1;
    FOR(){
        int v=To[i];if(v==ff)continue;
        fcy[v]=fcy[u]+Len[i];
        dfs1(v,u); siz[u]+=siz[v];
        if(siz[v]>siz[hson[u]]) hson[u]=v;
    }
}
void dfs2(int u,int tp)
{
    Top[u]=tp;
    if(hson[u])dfs2(hson[u],tp);
    FOR()
        if(To[i]!=fa[u]&&To[i]!=hson[u])
            dfs2(To[i],To[i]);
}
int LCA(int u,int v)
{
    while(Top[u]^Top[v]) dep[Top[u]]<dep[Top[v]]?v=fa[Top[v]]:u=fa[Top[u]];
    return dep[u]<dep[v]?u:v;
}
int Dis(int u,int v){return fcy[u]+fcy[v]-2*fcy[LCA(u,v)];}
int Fa[maxn],Size,root,mx,rt[maxn];
void Getroot(int u,int ff)
{
    siz[u]=1;int ret=0;
    FOR(){
        int v=To[i];if(v==ff||vis[v])continue;
        Getroot(v,u);siz[u]+=siz[v];
        ret=max(ret,siz[v]);
    }
    ret=max(ret,Size-siz[u]);
    if(ret<mx) mx=ret,root=u;
}
void DFS(int u,int ff)
{
    vis[u]=true;Fa[u]=ff;
    FOR(){
        int v=To[i];if(vis[v])continue;
        mx=Size=siz[v];
        Getroot(v,u);
        DFS(root,u);
    }
}
struct in{ int l,r,mn; }s[maxn<<6]; int tot;
void modify(int &Now,int L,int R,int pos,int val)
{
    if(!Now){Now=++tot; s[tot].mn=inf;}
    s[Now].mn=min(s[Now].mn,val);
    if(L==R) return; int Mid=(L+R)>>1;
    if(pos<=Mid) modify(s[Now].l,L,Mid,pos,val);
    else modify(s[Now].r,Mid+1,R,pos,val);
}
int query(int Now,int L,int R,int l,int r)
{
    if(!Now) return inf;
    if(l<=L&&r>=R) return s[Now].mn;
    int Mid=(L+R)>>1,res=inf;
    if(l<=Mid) res=min(res,query(s[Now].l,L,Mid,l,r));
    if(r>Mid) res=min(res,query(s[Now].r,Mid+1,R,l,r));
    return res;
}
void Modify(int x)
{
    modify(rt[x],1,n,x,0);
    for(int i=x;Fa[i];i=Fa[i]){
        int dis=Dis(x,Fa[i]);
        modify(rt[Fa[i]],1,n,x,dis);
    }
}
int Query(int x,int L,int R)
{
    int res=query(rt[x],1,n,L,R);
    for(int i=x;Fa[i];i=Fa[i])
    {
        int dis=Dis(x,Fa[i]);
        res=min(res,dis+query(rt[Fa[i]],1,n,L,R));
    }
    return res;
}
int main()
{
    int Q,u,v,len;
    scanf("%d",&n);
    rep(i,1,n-1){
        scanf("%d%d%d",&u,&v,&len);
        add(u,v,len); add(v,u,len);
    }
    dfs1(1,0); dfs2(1,1);
    Size=mx=n;
    Getroot(1,0); DFS(root,0);
    rep(i,1,n) Modify(i);
    int ans=0,l,r,x;
    scanf("%d",&Q);
    while(Q--){
        scanf("%d%d%d",&l,&r,&x);
        printf("%d\n",Query(x,l,r));
    }
    return 0;
}

 

posted @ 2019-07-16 15:21  nimphy  阅读(347)  评论(0编辑  收藏  举报