[NOI2015]软件包管理器

题目地址


注意点:

  • 动态开点线段树如果写区间覆盖问题一定要保证子结点存在.
  • 数据点里结点从0开始编号,平移加一即可.
  • 区间覆盖线段树需要写pushdown和pushup,query和insert也有一定的改动,但整体上难度降低.

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
const int MAXN=1000010,MAXM=1000010;
struct Node{
	int ls,rs;
	int tag,sum;
}tr[MAXN];
int root=0,nodeCnt=0;
void init(int now){
	if(!tr[now].ls)tr[now].ls=++nodeCnt;
	if(!tr[now].rs)tr[now].rs=++nodeCnt;
}
void pushup(int now){
	init(now);
	tr[now].sum=tr[tr[now].ls].sum+tr[tr[now].rs].sum;
} 
void pushdown(int now,int l,int r){
	init(now);
	int mid=(l+r)>>1;
	tr[tr[now].ls].tag=tr[now].tag;
	tr[tr[now].rs].tag=tr[now].tag;
    if(tr[now].tag==1)tr[tr[now].ls].sum=tr[tr[now].rs].sum=0;
    else if(tr[now].tag==2)tr[tr[now].ls].sum=mid-l+1,tr[tr[now].rs].sum=r-mid;
    tr[now].tag=0;
}
void insert(int &now,int l,int r,int ll,int rr,int x){
	if(!now)now=++nodeCnt;
	init(now);
	if(ll<=l&&rr>=r){
		tr[now].tag=x;
		if(x==1)tr[now].sum=0;
		else if(x==2)tr[now].sum=r-l+1;
		return;
	}
	int mid=(l+r)>>1;
	if(tr[now].tag)pushdown(now,l,r);
	if(ll<=mid)insert(tr[now].ls,l,mid,ll,rr,x);
	if(rr>mid)insert(tr[now].rs,mid+1,r,ll,rr,x);
	pushup(now);
}
int query(int now,int l,int r,int ll,int rr){
	if(ll<=l&&rr>=r)return tr[now].sum;
	int mid=(l+r)>>1;
	init(now);
	if(tr[now].tag){
		pushdown(now,l,r);
	}
	int ans=0;
	if(ll<=mid)ans+=query(tr[now].ls,l,mid,ll,rr);
	if(rr>mid)ans+=query(tr[now].rs,mid+1,r,ll,rr);
	return ans;
}
struct Edge{
	int from,to,nxt;
}e[MAXM];
int head[MAXN],edgeCnt=1;
void addEdge(int u,int v){
	e[++edgeCnt].from=u;
	e[edgeCnt].to=v;
	e[edgeCnt].nxt=head[u];
	head[u]=edgeCnt;
}
int dep[MAXN],fa[MAXN],son[MAXN],siz[MAXN];
void dfs1(int x){
	dep[x]=dep[fa[x]]+1;
	siz[x]=1;
	for(int i=head[x];i;i=e[i].nxt){
		int nowV=e[i].to;
		if(!dep[nowV]){
			fa[nowV]=x;
			dfs1(nowV);
			siz[x]+=siz[nowV];
			if(siz[nowV]>siz[son[x]])son[x]=nowV;
		}
	}
}
int id[MAXN],idCnt=0;
int initValue[MAXN],top[MAXN];
void dfs2(int x){
	id[x]=++idCnt;
	if(x==son[fa[x]])top[x]=top[fa[x]];
	else top[x]=x;
	if(son[x])dfs2(son[x]);
	for(int i=head[x];i;i=e[i].nxt){
		int nowV=e[i].to;
		if(!id[nowV])
			dfs2(nowV);
	}
}
void modify(int u,int v,int x){
	while(top[u]!=top[v]){
		if(dep[top[u]]>dep[top[v]])swap(u,v);
		insert(root,1,MAXN,id[top[v]],id[v],x);
		v=fa[top[v]];
	}
	insert(root,1,MAXN,min(id[u],id[v]),max(id[u],id[v]),x);
}
int sum(int u,int v){
	int ans=0;
	while(top[u]!=top[v]){
		if(dep[top[u]]>dep[top[v]])swap(u,v);
		ans+=id[v]-id[top[v]]+1-query(root,1,MAXN,id[top[v]],id[v]);
		v=fa[top[v]];
	}
	if(dep[u]<dep[v])swap(u,v);
	ans+=id[u]-id[v]+1-query(root,1,MAXN,id[v],id[u]);
	return ans;
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i=2;i<=n;i++){
		int v;
		scanf("%d",&v);
		v=v+1;
		addEdge(v,i);
		addEdge(i,v);
	}
	dfs1(1);
	dfs2(1);
	int q;
	scanf("%d",&q);
	string str;
	int x;
	for(int i=1;i<=q;i++){
		cin>>str>>x;
		x=x+1;
		if(str[0]=='i'){
			cout<<sum(1,x)<<endl;
			modify(1,x,2);
		}else if(str[0]=='u'){
			cout<<query(root,1,MAXN,id[x],id[x]+siz[x]-1)<<endl;
			insert(root,1,MAXN,id[x],id[x]+siz[x]-1,1);
		}
	}
	return 0;
}
原文地址:https://www.cnblogs.com/zbsy-wwx/p/11680621.html