前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >CF GYM102759 I. Query On A Tree 17

CF GYM102759 I. Query On A Tree 17

作者头像
yzxoi
发布2022-09-19 13:27:53
5520
发布2022-09-19 13:27:53
举报
文章被收录于专栏:OI

CF GYM102759 I. Query On A Tree 17

Description

题目链接:GYM102759 I

给定一棵以 1​ 为根节点的拥有 N​ 个节点的带点权树,初始时每个点的点权均为 0​,有 Q​ 次操作,每次操作有两种类型:

  • 1 u,以 u 为根节点的子树内每个点点权均加 1
  • 2 u v,将节点 u 到节点 v 的简单路径上的点的点权均加 1

每次操作后询问使 \sum_{y=1}^N A_y\times dist(x,y)​ 最小的 x,其中 dist(x,y) 表示点 x 到点 y 的路径上的边数,如果有多个,请输出离根节点 1 最近的那个。

N,Q\leq 10^5

Solution

显然最后的答案就是求带权重心。

那么我们相当于求深度最浅的重心。

对于一个点 v,如果它是答案,把它作为根,那么 root 所在的子树大小 \leq 权值和的一半,也就是说剩下的大小(以 root 为根时子树 v 的大小)\ge 权值和的一半,也就是说中位数一定包含在 v 所包含的子树里。

直接上轻重链剖分,可以线段树上二分求出中位数,每次倍增向上跳,判一下是否符合条件即可。

Code

代码语言:javascript
复制
#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define W while
#define I inline
#define RI register int
#define LL long long
#define Cn const
#define CI Cn int&
#define gc getchar
#define D isdigit(c=gc())
#define pc(c) putchar((c))
#define min(x,y) ((x)<(y)?(x):(y))
#define max(x,y) ((x)>(y)?(x):(y))
using namespace std;
namespace Debug{
    Tp I void _debug(Cn char* f,Ty t){cerr<<f<<'='<<t<<endl;}
    Ts I void _debug(Cn char* f,Ty x,Ar... y){W(*f!=',') cerr<<*f++;cerr<<'='<<x<<",";_debug(f+1,y...);}
    Tp ostream& operator<<(ostream& os,Cn vector<Ty>& V){os<<"[";for(Cn auto& vv:V) os<<vv<<",";os<<"]";return os;}
    #define gdb(...) _debug(#__VA_ARGS__,__VA_ARGS__)
}using namespace Debug;
namespace FastIO{
    Tp I void read(Ty& x){char c;int f=1;x=0;W(!D) f=c^'-'?1:-1;W(x=(x<<3)+(x<<1)+(c&15),D);x*=f;}
    Ts I void read(Ty& x,Ar&... y){read(x),read(y...);}
    Tp I void write(Ty x){x<0&&(pc('-'),x=-x,0),x<10?(pc(x+'0'),0):(write(x/10),pc(x%10+'0'),0);}
    Tp I void writeln(Cn Ty& x){write(x),pc('\n');}
}using namespace FastIO;
Cn int N=1e5+10;
int n,m,fir[N],nxt[N<<1],son[N<<1],tot,dep[N],f[N][21],sz[N],mx[N],dfn[N],top[N],bk[N],cnt;
I void Add(CI x,CI y){nxt[++tot]=fir[x],fir[x]=tot,son[tot]=y;}
#define to son[i]
I void Dfs1(CI x,CI fa){RI i;for(dep[x]=dep[f[x][0]=fa]+(sz[x]=1),i=fir[x];i;i=nxt[i]) to^fa&&(Dfs1(to,x),sz[x]+=sz[to],sz[mx[x]]<sz[to]&&(mx[x]=to));}
I void Dfs2(CI x,CI Top){
    if(top[x]=Top,bk[dfn[x]=++cnt]=x,mx[x]) Dfs2(mx[x],Top);
    RI i;for(i=fir[x];i;i=nxt[i]) to^f[x][0]&&to^mx[x]&&(Dfs2(to,to),0);
}
class SegmentTree{
    private:
        LL T[N<<2],tg[N<<2];
        #define mid (l+r>>1)
        #define PT CI x=1,CI l=1,CI r=n
        #define LT x<<1,l,mid
        #define RT x<<11,mid+1,r
        #define PU(x) (T[x]=T[x<<1]+T[x<<11])
        I void PD(CI x,CI l,CI r){tg[x]&&(T[x<<1]+=tg[x]*(mid-l+1),tg[x<<1]+=tg[x],T[x<<11]+=tg[x]*(r-mid),tg[x<<11]+=tg[x],tg[x]=0);}
    public:
        I void U(CI L,CI R,CI v,PT){
            if(L<=l&&r<=R) return T[x]+=v*(r-l+1),tg[x]+=v,void();
            PD(x,l,r),L<=mid&&(U(L,R,v,LT),0),R>mid&&(U(L,R,v,RT),0),PU(x);
        }
        I LL Q(CI L,CI R,PT){
            if(L<=l&&r<=R) return T[x];
            LL S=0;return PD(x,l,r),L<=mid&&(S+=Q(L,R,LT)),R>mid&&(S+=Q(L,R,RT)),S;
        }
        I int G(LL k,PT){
            if(l==r) return bk[l];
            return PD(x,l,r),T[x<<1]>=k?G(k,LT):G(k-T[x<<1],RT);
        }
        I LL S(){return T[1];}
}T;
I void U(RI x,RI y){
    W(top[x]^top[y]) dep[top[x]]<dep[top[y]]&&(swap(x,y),0),T.U(dfn[top[x]],dfn[x],1),x=f[top[x]][0];
    dfn[x]>dfn[y]&&(swap(x,y),0),T.U(dfn[x],dfn[y],1);
}
I int Q(){
    LL k=T.S()/2+1;RI j,i=T.G(k);
    if(T.Q(dfn[i],dfn[i]+sz[i]-1)>=k) return i;
    for(j=20;~j;j--) if(f[i][j]&&T.Q(dfn[f[i][j]],dfn[f[i][j]]+sz[f[i][j]]-1)<k) i=f[i][j];
    return f[i][0];
}
int main(){
    RI i,j,o,x,y;for(read(n),i=1;i<n;i++) read(x,y),Add(x,y),Add(y,x);for(Dfs1(1,0),Dfs2(1,1),j=1;j<=20;j++) for(i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1];
    for(read(m);m--;) read(o,x),o&1?T.U(dfn[x],dfn[x]+sz[x]-1,1):(read(y),U(x,y)),writeln(Q());return 0;
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021-07-24 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • CF GYM102759 I. Query On A Tree 17
    • Description
      • Solution
        • Code
        领券
        问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档