题目链接: 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;
}