【BZOJ】4012: [HNOI2015]开店-点分树/树剖+主席树

最新推荐文章于 2020-10-04 20:42:09 发布
最新推荐文章于 2020-10-04 20:42:09 发布
阅读量346 收藏
点赞数 2
分类专栏： -------数据结构------- 主席树 树链剖分 点分树
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/corsica6/article/details/84795957
版权

传送门:bzoj4012

这题码+调试就花了一中午和一下午。。。最后发现问题都是有地方没开 l o n g   l o n g long \ long long longTAT。。。

法2的代码比较有技巧性(还不太熟。

题解

法1(点分树)：

首先点分治处理出 v : v: v:每个重心到其所管辖子树中每个点的距离，压进vector后按年龄排序，转成距离的前缀和, f v : fv: fv:该点点分树中的父亲结点到这颗子树每个点的距离，同样压进vector后排序求前缀和(容斥用)

暴力跳点分树，每层二分找到需要的区间处理即可。细节可以自己再想一下。

法2(树剖+主席树)：

注意到 d i s ( u , v ) = d e p ( u ) + d e p ( v ) − 2 × d e p ( L C A ( u , v ) ) dis(u,v)=dep(u)+dep(v)-2\times dep(LCA(u,v)) dis(u,v)=dep(u)+dep(v)−2×dep(LCA(u,v))，于是点之间的关系转成了分别处理每个点对于答案的贡献。

某个年龄区间的 d e p ( v ) dep(v) dep(v)可以排序+前缀和处理。关键问题在于如何快速求出 ∑ d e p ( L C A ( u , v ) ) \sum dep(LCA(u,v)) ∑dep(LCA(u,v))。

采用一种 O ( n l o g 2 n ) O(nlog^2n) O(nlog2n)的方法：把每个点到根路径上的所有边覆盖次数+1， ∑ d e p ( L C A ( u , v ) ) \sum dep(LCA(u,v)) ∑dep(LCA(u,v))就是 u u u到根路径上的每条边的边权 × \times ×覆盖次数。

由于每次查询的是一段年龄连续的区间，可以建立主席树，将点按年龄排序后逐个加入。

注意：点数太多，需要标记永久化。

代码

点分树：

#include<bits/stdc++.h>
#define pb push_back
#define mid ((L+R)>>1)
using namespace std;
const int N=15e4+10;
typedef long long ll;

int n,m,A,ag[N];ll ans;
int head[N],to[N<<1],nxt[N<<1],w[N<<1],tot;
int sz[N],f[N],son[N],top[N],dep[N],dis[N];
clock_t stt,edd;

char cp,OS[100];
template<class yyy>inline void rd(yyy &x)
{
    cp=getchar();x=0;
    for(;!isdigit(cp);cp=getchar());
    for(;isdigit(cp);cp=getchar()) x=(x<<3)+(x<<1)+(cp^48);
}

template<class yyy>inline void ot(yyy x)
{
    int re=0;for(;(!re)||(x);x/=10) OS[++re]='0'+x%10;
    for(OS[0]='\n';~re;--re) putchar(OS[re]);
}

inline void lk(int u,int v,int vv)
{to[++tot]=v;nxt[tot]=head[u];head[u]=tot;w[tot]=vv;}

void dfs(int x)
{
    sz[x]=1;
    for(int j,i=head[x];i;i=nxt[i]){
        j=to[i];if(j==f[x]) continue;
        f[j]=x;dep[j]=dep[x]+1;dis[j]=dis[x]+w[i];dfs(j);
        sz[x]+=sz[j];if(sz[j]>sz[son[x]]) son[x]=j; 
    }
}

void dfss(int x,int tpo)
{
    top[x]=tpo;if(!son[x]) return;dfss(son[x],tpo);
    for(int j,i=head[x];i;i=nxt[i]){
        j=to[i];if(j==f[x] || j==son[x]) continue;
        dfss(j,j);
    }
}

inline int LCA(int x,int y){for(;top[x]!=top[y];x=f[top[x]]) if(dep[top[x]]<dep[top[y]]) swap(x,y);
return dep[x]<dep[y]?x:y;}
inline int dist(int x,int y){
return dis[x]+dis[y]-(dis[LCA(x,y)]<<1);}

namespace ds{
     int f[N],mx[N],MN,S,rt,vs[N],b;
     struct P{ 
        int yr;ll d;
        P(int yr_=0,ll d_=0):yr(yr_),d(d_){};
        bool operator<(const P&ky)const{return yr<ky.yr;}
        bool operator <=(const P&ky)const{return yr<=ky.yr;}
     };
     vector<P>v[N],fv[N];

     void getrt(int x,int fr)
     {
          sz[x]=1;mx[x]=0;
          for(int j,i=head[x];i;i=nxt[i]){
             j=to[i];if(j==fr || vs[j]) continue;
             getrt(j,x);sz[x]+=sz[j];mx[x]=max(mx[x],sz[j]);
          }
          mx[x]=max(mx[x],S-sz[x]);
          if(mx[x]<MN) {MN=mx[x];rt=x;}
     }

     void dfs(int x,int fr,int ds)
     {
         v[b].pb(P(ag[x],ds));
         for(int j,i=head[x];i;i=nxt[i]){
            j=to[i];if(j==fr || vs[j]) continue;
            dfs(j,x,ds+w[i]);
         }
     }
     
     void dfss(int x,int fr,int ds)
     {
         fv[b].pb(P(ag[x],ds));
         for(int j,i=head[x];i;i=nxt[i]){
            j=to[i];if(j==fr || vs[j]) continue;
            dfss(j,x,ds+w[i]);
         }
     }

     void build(int x)
     {
         int i,j,k,sq=S;vs[x]=1;b=x;dfs(x,0,0);
         sort(v[x].begin(),v[x].end());j=v[x].size();
         for(i=1;i<j;++i) v[x][i].d+=v[x][i-1].d;
         for(int j,i=head[x];i;i=nxt[i]){
            j=to[i];if(vs[j]) continue;S=(sz[j]>sz[x])?(sq-sz[x]):sz[j];MN=n+1;
            getrt(j,x);f[rt]=x;b=rt;dfss(j,x,w[i]);
            sort(fv[rt].begin(),fv[rt].end());j=fv[rt].size();
            for(k=1;k<j;++k) fv[rt][k].d+=fv[rt][k-1].d;build(rt);
         }
     }

     inline ll cala(int l,int r,int dd)
     {
         int x=v[b].size(),y=-1,L=0,R=v[b].size()-1;
         for(;L<=R;) 
           v[b][mid].yr>=l?(R=(x=mid)-1):(L=mid+1);
         for(L=0,R=v[b].size()-1;L<=R;) 
           v[b][mid].yr<=r?(L=(y=mid)+1):(R=mid-1);
         if(x>y) return 0LL;
         return v[b][y].d-(x?v[b][x-1].d:0)+(ll)(y-x+1)*dd;
     }
     
     inline ll calb(int l,int r,int dd)
     {
     	 int x=fv[b].size(),y=-1,L=0,R=fv[b].size()-1;
     	 for(;L<=R;) 
            fv[b][mid].yr>=l?(R=(x=mid)-1):(L=mid+1);
         for(L=0,R=fv[b].size()-1;L<=R;) 
           fv[b][mid].yr<=r?(L=(y=mid)+1):(R=mid-1);
         if(x>y) return 0LL;
     	 return fv[b][y].d-(x?fv[b][x-1].d:0)+(ll)(y-x+1)*dd;
     }

     inline void query(int x,int l,int r)
     {
          b=x;ans=cala(l,r,0);int dd,ori=x;
          for(;f[x];x=b){
              dd=dist(ori,f[x]);
              b=x;ans-=calb(l,r,dd);
              b=f[x];ans+=cala(l,r,dd);
          }
     }
}

int main(){	
    using namespace ds;
    int i,j,k,x,y,z;
    rd(n);rd(m);rd(A);
    for(i=1;i<=n;++i) rd(ag[i]);
    for(i=1;i<n;++i){
        rd(x);rd(y);rd(z);lk(x,y,z);lk(y,x,z);
    }
    dep[1]=1;dfs(1);dfss(1,1);
    MN=n+1;S=n;getrt(1,0);
    build(rt);
    for(;m;--m){
        rd(x);rd(y);rd(z);y=(ans+y)%A;z=(ans+z)%A;
        query(x,min(y,z),max(y,z));ot(ans);
    }
    return 0;
}

树剖+主席树

#include<bits/stdc++.h>
#define mid ((l+r)>>1)
using namespace std;
const int N=15e4+10,M=2e7+10;
typedef long long ll;

int n,m,A,vl[N],bl[N],num,df[N],dfn;ll ans;
int head[N],to[N<<1],nxt[N<<1],w[N<<1],tot,dis[N];
int sz[N],f[N],son[N],top[N],dep[N],rt[N],cnt;
int ls[M],rs[M],st[M];ll ss[M],qz[N];

struct P{
    int id,v;
    bool operator<(const P&ky)const{return v<ky.v;}
}q[N];
clock_t stt,edd;

char cp,OS[100];
template<class yyy>inline void rd(yyy &x)
{
    cp=getchar();x=0;
    for(;!isdigit(cp);cp=getchar());
    for(;isdigit(cp);cp=getchar()) x=(x<<3)+(x<<1)+(cp^48);
}

template<class yyy>inline void ot(yyy x)
{
    int re=0;for(;(!re)||(x);x/=10) OS[++re]='0'+x%10;
    for(OS[0]='\n';~re;--re) putchar(OS[re]);
}

inline void lk(int u,int v,int vv)
{to[++tot]=v;nxt[tot]=head[u];head[u]=tot;w[tot]=vv;}

void dfs(int x)
{
    sz[x]=1;
    for(int j,i=head[x];i;i=nxt[i]){
        j=to[i];if(j==f[x]) continue;
        bl[j]=w[i];f[j]=x;dep[j]=dep[x]+1;dis[j]=dis[x]+w[i];dfs(j);
        sz[x]+=sz[j];if(sz[j]>sz[son[x]]) son[x]=j; 
    }
}

void dfss(int x,int tpo)
{
    top[x]=tpo;df[x]=++dfn;vl[dfn]=bl[x];
    if(!son[x]) return;dfss(son[x],tpo);
    for(int j,i=head[x];i;i=nxt[i]){
        j=to[i];if(j==f[x] || j==son[x]) continue;
        dfss(j,j);
    }
}

void ad(int pr,int &k,int l,int r,int L,int R)
{
    if(k==pr){k=++cnt;st[k]=st[pr];ss[k]=ss[pr];ls[k]=ls[pr];rs[k]=rs[pr];}	
    if(L==l && r==R) {st[k]++;return;}ss[k]+=(vl[R]-vl[L-1]);
    if(L<=mid) ad(ls[pr],ls[k],l,mid,L,min(R,mid));
    if(R>mid) ad(rs[pr],rs[k],mid+1,r,max(L,mid+1),R);
}

ll ask(int k,int l,int r,int L,int R)
{
    if(!k) return 0LL;
    if(L==l && r==R) return ss[k]+(ll)st[k]*(vl[r]-vl[l-1]);
    ll re=(ll)(vl[R]-vl[L-1])*st[k];
    if(L<=mid) re+=ask(ls[k],l,mid,L,min(R,mid));
    if(R>mid) re+=ask(rs[k],mid+1,r,max(L,mid+1),R);
    return re;
}

inline void query(int x,int L,int R)
{
    if(L>R) {ans=0LL;return;}ans=qz[R]-qz[L-1]+(ll)dis[x]*(R-L+1);
    L=rt[L-1];R=rt[R];
    for(;x;x=f[top[x]]) 
    ans-=((ask(R,1,n,df[top[x]],df[x])-ask(L,1,n,df[top[x]],df[x]))<<1);
}

inline int dn(int x)
{
    int l=1,r=n,re=n+1;
    for(;l<=r;) q[mid].v>=x?(r=(re=mid)-1):(l=mid+1);
    return re;
}

inline int up(int x)
{
    int l=1,r=n,re=-1;
    for(;l<=r;) q[mid].v<=x?(l=(re=mid)+1):(r=mid-1);
    return re;
}

int main(){	
    int i,j,k,x,y,z;
    rd(n);rd(m);rd(A);vl[0]=0;qz[0]=0LL;
    for(i=1;i<=n;++i) rd(q[i].v),q[i].id=i;sort(q+1,q+n+1);
    for(i=1;i<n;++i){rd(x);rd(y);rd(z);lk(x,y,z);lk(y,x,z);}
    dep[1]=1;dfs(1);dfss(1,1);for(i=1;i<=n;++i) vl[i]+=vl[i-1],qz[i]=qz[i-1]+dis[q[i].id];
    for(i=1;i<=n;++i)
    	for(rt[i]=rt[i-1],x=q[i].id;x;x=f[top[x]]) ad(rt[i-1],rt[i],1,n,df[top[x]],df[x]);
    for(;m;--m){
        rd(x);rd(y);rd(z);y=(ans+y)%A;z=(ans+z)%A;if(y>z) swap(y,z);
        query(x,dn(y),up(z));ot(ans);
    }
    return 0;
}
ccosi
关注
  • 2
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
[GDOI2016][树链剖分+主席树]疯狂动物城
EZ_TJY的博客
07-19 1594
题面： Description Nick 是只在动物城以坑蒙拐骗为生的狐狸，儿时受到偏见的伤害，放弃了自己的理想。他被兔子 Judy 设下圈套，被迫与她合作查案，而卷入意想不到的阴谋，历尽艰险后成为搭档。他们识破了绵羊副市长 Bellwether 的计划，发现是 Bellwether 陷害食肉动物，用毒药让食肉动物发狂。Bellwether 被抓到了监狱里面， Nick 和 Judy 过上了一...
bzoj 3730. 震波（动态点分治 + vector树状数组）
01-06
考虑点分树的做法，在点分树上每个点以距离为下标建一棵线段树，每次询问查询子树的贡献，再暴力向上跳合并父节点来自其它节点的贡献。 因为子树树形被破坏，在做减法时，子节点子树对父节点的贡献不能用子树维护的...
BZOJ4012 [HNOI2015]开店
weixin_30699831的博客
07-10 108
Description 风见幽香有一个好朋友叫八云紫，她们经常一起看星星看月亮从诗词歌赋谈到 人生哲学。最近她们灵机一动，打算在幻想乡开一家小店来做生意赚点钱。这样的 想法当然非常好啦，但是她们也发现她们面临着一个问题，那就是店开在哪里，面 向什么样的人群。很神奇的是，幻想乡的地图是一个树形结构，幻想乡一共有 n 个地方，编号为 1 到 n，被 n-1 条带权的边连接起来。每个地方...
[HNOI2015]开店【动态点分治/点分树】
既然弱小，就只顾变强就是了
04-14 200
题目链接 Q次强制在线查询，求N个点的固定树型结构的某点u到其余范围在[L, R]区间内的点的距离之和。 又是一道动态点分治的码量题，逐渐写出手感。 首先，由于这道题没有修改操作（可以省掉好多码量了呢），我们可以用前缀和、差分这样的方法来进行求解。所以，我们先看怎样求一点到树上各点的距离，这在[ZJOI2015]幻想乡战略游戏【动态点分治/点分树】中有了初略的涉及，这里再细致的再...
bzoj4012: [HNOI2015]开店
weixin_30797199的博客
12-18 115
传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=4012 思路：首先我们考虑一个简化的问题： 给定一棵树，每次询问所有点到一个点的距离和。 画个图就能知道：距离和=所有点到根的距离和+点数*u到根的距离-每个点与u的lca到根的距离*2 于是问题就成了求lca的dis和 那么我们先对每个点，把...
最近点 - 可持久化点分树 - 主席树
Mys_C_K的博客
03-19 515
题目大意：给一棵树，点有黑白，每次形如翻转一个点颜色，询问到某个点的最近黑色点距离，以及返回之前某个版本。 题解：显然将点分树可持久化一下，然后用一个可持久化堆（虽然写了个可持久化线段树）维护一下即可。点分树可持久化可以暴力拿个主席树维护点分树的点到堆的编号的映射，也可以将原树三度化之后直接可持久化，后者这部分复杂度一个log（但是并不很快）。 #include<bits/stdc++.h&...
wangjunrui666#bzoj-problem##5165. 树上倍增1
07-25
您需要写一个树上倍增算法，以实现如下操作：A x 新建一个节点，将它作为x节点的儿子，编号为当前节点总数+1。保证询问不超过1000次输出格式对于每一个Q操作，
Link-Cut-Tree 动态树经典讲稿附例题
10-11
适用ACM和OI等算法选手，Link-Cut-Tree 动态树经典讲稿，附BZOJ多道优秀练习题，原作者吉大附中实验学校 PoPoQQQ
数论题解：「BZOJ1053」反素数/「Violet5」樱花
01-22
「BZOJ1053」反素数详解 「BZOJ1053」反素数问题是数论领域中的一个经典问题，该问题要求我们计算不超过N的最大反质数。所谓反质数，即满足对所有0，g(x)>g(i)的自然数x，其中g(x)表示x的约数个数。 首先，我们...
wangjunrui666#bzoj-problem##4882. [Lydsy1705月赛]卡常生成树1
07-25
为了避免输入数据过于庞大，数据进行了压缩，解压方式可用下面这段代码解释：//在点i与点j之间添加一条边权为w的边第一行包含一个正整数T(1)，表示
动态点分治 / 点分树学习笔记 ---- BZOJ3730 震波 / [ZJOI2017] 幻想乡的战略游戏 / [HNOI2015]开店
FSYo 的博客
05-18 192
点分树就是将每一次的重心连起来， 形成一个深度不超过log的树 相当于将点分治的过程静态到一颗树上 实现如下， fa记录点分树上的父亲 void Divide(int u, int f){ fa[u] = f; vis[u] = 1; int sum = Siz; for(int i=first[u];i;i=nxt[i]){ int t = to[i]; if(vis[t]) c...
bzoj 4012: [HNOI2015]开店 （树链剖分+主席树）
clover_hxy的博客
04-19 615
题目描述传送门题解这道题维护和求解的方法和bzoj 3924: [Zjoi2015]幻想乡战略游戏是类似的。 但是这道题有一个[L,R]的区间限制，所以我们用主席树来维护，外层是按照离散化后的xix_i从小到大，内层是dfs序。代码#include<iostream> #include<cstring> #include<algorithm> #include<cstdio> #include<c
[BZOJ]4012: [HNOI2015]开店 树链剖分+主席树（线段树合并）
TYB的博客
11-23 286
树链剖分+主席树（线段树合并）
[HNOI2015]开店（树剖+主席树+标记永久化）
weixin_30679823的博客
01-05 89
听说正解点分树？我不会就对了 此题是 \([LNOI2014]LCA\) 强化版，也是差分一下，转化为区间加区间和 不过权值有大小要求，那么我们按照权值排序，依次加入主席树，询问的时候 \(lower\_bound\) 一下找到区间 \([l,r]\) 在主席树上实现区间加，肯定要标记永久化。每次最多修改 \(2\lfloor \log^n\rfloor\) 个区间，所以一次最多会开出 \(4\l...
训练周记（树剖+主席树）
C_Dreamy的博客
10-04 266
这周给平衡树开了一个头，这周可能是自打寒假以来最用功的一周了，树链剖分和线段树不说掌握的很透彻，但是模板题打的还是很顺手，还有主席树，比较难理解，虽然感觉平衡树更难理解一些，但是时间剩的不大允许了， 贴一下这后的模板： 树链剖分： vector<int> G[N<<1]; int sz[N],son[N],fa[N],dep[N]; //函数调用 s 为源点：dfs(s,0) void dfs1(int u,int f) //寻找重儿子 { sz[
【HNOI2015】开店
samjia2000的博客
07-03 677
题目大意给出一棵有n个节点的树，每条边有个边长，每个节点有个权值a[i]，有q个询问形如”u l r”询问点u到所有权值在l到r之间的点的路径长度和一眼解法一个点与多个点之间的距离和，显然我们可以用点分治来做，对于每个分治中心，我们将其所有该层的点都记录下来，那么每层共有n个，最多nlog2nnlog_2n个，同时，在我们计算答案时要计算每个分治中心对其父亲的影响。 询问时我们就直接在重心树上跳，
[HNOI2015]开店
Cyhlnj
01-06 292
如果没有年龄区间 画图发现ans=∑ni=1disi+n∗disu−2∗∑ni=1dislca(i,u)\sum_{i=1}^n dis_i + n * dis_u - 2 * \sum_{i=1}^{n} dis_{lca(i, u)} 对∑ni=1dislca(i,u)\sum_{i=1}^{n} dis_{lca(i, u)}用树链剖分，对于每个点，都向上走到根，记录每条路经过次数，询问
Distance on the tree （树链剖分+主席树）
A_Thinking_Reed_的博客
08-20 224
题目：https://nanti.jisuanke.com/t/38229 题意：树上有边权，每次查询树上两点路径上边权小于等于k的数量。 思路：边权下推成点权，树剖化树为区间，区间查找小于k的数的个数。 代码： #include <bits/stdc++.h> #define LL long long using namespace std; typedef pair&lt...
大学生创业计划书(30)-两份资料.doc
最新发布
08-20
大学生创业计划书(30)-两份资料.doc
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值