2019牛客暑期多校训练营(第八场)I Inner World 线段树

题目链接: https://ac.nowcoder.com/acm/contest/888/I

题意:

你开始的时候有n个只有点1的树,现在你要进行m次给树加点的操作,每次操作给你u v l r,表示你要在第l到第r棵树上的点u后加一个v,即加一个v变成u的儿子,并保证u这个点一定都存在。

做法:

因为它题目保证所有的v都是不一样的,即把所有树当成一棵看的话,加进去的最后应该是一棵没有重复点的树,所以我们可以直接当成是一棵树来做,但是每棵树在过程中加的区间是不一样的。

用的是一种类似括号线段树的方法,但是这里有点不一样,每个点要先用dfs序来确确定一个左界和一个右界,左界表示的是到达点 i i i 这个点的时候已经是第几个点了,右界表示的是当统计到第几个点的时候不算是我的子树了(其实操作起来很简单,就是在dfs离开这个点的时候记录一下就好了)。那么我只要遍历树的时候加上从我的开始到我的结束的部分就可以了,但是从我开始的地方还要加上我自己的结束比较麻烦,直接就我的结束减去我的开始的前一个序列就好了。

代码

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
#define lson rt<<1
#define rson rt<<1|1
#define mid (l+r)/2
using namespace std;
typedef long long ll;
const int maxn=300005;
int L[maxn],R[maxn],id[maxn],cnt,n,m;
ll laz[maxn<<2],sum[maxn<<2],ans[maxn];
int st[maxn],en[maxn],q;
struct node{
    int x,l,r,v;
    node(){}
    node(int x,int l,int r,int v):x(x),l(l),r(r),v(v){}

};
vector<node> Q[maxn];
vector<int> E[maxn];
void dfs(int u){
    st[u]=++cnt;
    id[cnt]=u;
    for(auto v:E[u]) dfs(v);
    en[u]=cnt;
}
void push_down(int rt,int l,int r){
    if(laz[rt]){
        sum[lson]+=(mid-l+1)*laz[rt];
        sum[rson]+=(r-mid)*laz[rt];
        laz[lson]+=laz[rt];
        laz[rson]+=laz[rt];
        laz[rt]=0;
    }
}
void update(int l,int r,int rt,int ql,int qr){
    if(ql<=l&&r<=qr){
        sum[rt]+=(ll)r-l+1;
        laz[rt]+=1;
        return ;
    }
    push_down(rt,l,r);
    if(ql<=mid) update(l,mid,lson,ql,qr);
    if(qr>mid) update(mid+1,r,rson,ql,qr);
    sum[rt]=sum[lson]+sum[rson];
}
ll query(int l,int r,int rt,int ql,int qr){
    if(ql<=l&&r<=qr){
        return sum[rt];
    }
    ll ret=0;
    push_down(rt,l,r);
    if(ql<=mid) ret+=query(l,mid,lson,ql,qr);
    if(qr>mid) ret+=query(mid+1,r,rson,ql,qr);
    return ret;
}
int main(){
    scanf("%d%d",&n,&m);

    rep(i,1,m) {
        int u,v,ll,rr;
        scanf("%d%d%d%d",&u,&v,&ll,&rr);
        E[u].push_back(v);
        L[v]=ll,R[v]=rr;
    }
    L[1]=1,R[1]=n;
    dfs(1);
    scanf("%d",&q);
    rep(i,1,q){
        int x,l,r;
        scanf("%d%d%d",&x,&l,&r);
        Q[st[x]-1].push_back(node(i,l,r,-1));
        Q[en[x]].push_back(node(i,l,r,1));
    }
    for(int i=1;i<=cnt;i++){
        update(1,n,1,L[id[i]],R[id[i]]);
        for(auto e:Q[i]){
            ans[e.x]+=query(1,n,1,e.l,e.r)*e.v;
        }
    }
    rep(i,1,q) printf("%lld\n",ans[i]);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值