• [Noi2015]软件包管理器 BZOJ4196


    分析:水题

    每次安装的时候和根节点求lca的过程中区间覆盖+区间查询

    每次删除的时候查询子树中安装的数量+区间覆盖

    附上代码:

    #include <cstdio>
    #include <algorithm>
    #include <cmath>
    #include <cstring>
    #include <queue>
    #include <iostream>
    #include <cstdlib>
    using namespace std;
    #define N 100005
    #define lson l,m,rt<<1
    #define rson m+1,r,rt<<1|1
    struct node
    {
    	 int to,next;	
    }e[N<<1];
    int head[N],cnt,dep[N],fa[N],anc[N],siz[N],son[N],sum[N<<2],cov[N<<2],n,Q,idx[N],tims;
    char s[20];
    void add(int x,int y)
    {
    	e[cnt].to=y;
    	e[cnt].next=head[x];
    	head[x]=cnt++;
    	return ;
    }
    void dfs1(int x,int from)
    {
    	dep[x]=dep[from]+1,siz[x]=1,fa[x]=from;
    	for(int i=head[x];i!=-1;i=e[i].next)
    	{
    		int to1=e[i].to;
    		if(to1!=from)
    		{
    			dfs1(to1,x);
    			siz[x]+=siz[to1];
    			if(siz[son[x]]<siz[to1])son[x]=to1;
    		}
    	}
    }
    void dfs2(int x,int top)
    {
    	idx[x]=++tims;
    	anc[x]=top;
    	if(son[x])dfs2(son[x],top);
    	for(int i=head[x];i!=-1;i=e[i].next)
    	{
    		int to1=e[i].to;
    		if(to1!=son[x]&&to1!=fa[x])dfs2(to1,to1);
    	}
    }
    void PushUp(int rt)
    {
    	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    }
    void PushDown(int rt,int m)
    {
    	if(cov[rt]!=-1)
    	{
    		cov[rt<<1]=cov[rt<<1|1]=cov[rt];
    		sum[rt<<1]=(m-(m>>1))*cov[rt];
    		sum[rt<<1|1]=(m>>1)*cov[rt];
    		cov[rt]=-1;
    		return ;
    	}
    }
    void build(int l,int r,int rt)
    {
    	cov[rt]=-1;
    	if(l==r)
    	{
    		sum[rt]=1;return ;
    	}
    	int m=(l+r)>>1;
    	build(lson);
    	build(rson);
    	PushUp(rt);
    }
    void Update(int L,int R,int c,int l,int r,int rt)
    {
    	if(L<=l&&r<=R)
    	{
    		cov[rt]=c;
    		sum[rt]=c*(r-l+1);
    		return ;
    	}
    	PushDown(rt,r-l+1);
    	int m=(l+r)>>1;
    	if(m>=L)Update(L,R,c,lson);
    	if(m<R)Update(L,R,c,rson);
    	PushUp(rt);
    }
    int query(int L,int R,int l,int r,int rt)
    {
    	if(L<=l&&r<=R)return sum[rt];
    	PushDown(rt,r-l+1);
    	int m=(l+r)>>1,ret=0;
    	if(m>=L)ret+=query(L,R,lson);
    	if(m<R)ret+=query(L,R,rson);
    	return ret;
    }
    int Install(int x)
    {
    	int ret=0;
    	while(x)
    	{
    		ret+=query(idx[anc[x]],idx[x],1,n,1);
    		//printf("%d %d %d
    ",idx[anc[x]],idx[x],ret);
    		Update(idx[anc[x]],idx[x],0,1,n,1);
    		x=fa[anc[x]];
    	}
    	return ret;
    }
    int Uninstall(int x)
    {
    	int ret=siz[x]-query(idx[x],idx[x]+siz[x]-1,1,n,1);
    	//printf("%d %d
    ",idx[x],idx[x]+siz[x]-1);
    	Update(idx[x],idx[x]+siz[x]-1,1,1,n,1);
    	return ret;
    }
    int main()
    {
    	memset(head,-1,sizeof(head));
    	scanf("%d",&n);
    	build(1,n,1);
    	for(int i=2;i<=n;i++)
    	{
    		int x;
    		scanf("%d",&x);
    		x++;
    		add(x,i);
    		add(i,x);
    	}
    	dfs1(1,0);
    	dfs2(1,1);
    	scanf("%d",&Q);
    	while(Q--)
    	{
    		int x;
    		scanf("%s%d",s,&x);x++;
    		if(s[0]=='i')
    		{
    			printf("%d
    ",Install(x));
    		}else
    		{
    			printf("%d
    ",Uninstall(x));
    		}
    	}
    	return 0;
    }
    

      

  • 相关阅读:
    Python语言基础03-分支和循环结构
    Python语言基础01-初识Python
    李宏毅深度学习笔记08---Bert、ELMO、GPT
    李宏毅深度学习笔记07---Transformer
    李宏毅深度学习笔记06---word embedding(词嵌入也称词向量)
    李宏毅深度学习笔记05---RNN 循环神经网络02
    李宏毅深度学习笔记04---RNN 循环神经网络01
    李宏毅深度学习笔记03---CNN 卷积神经网络
    李宏毅深度学习笔记02---深度学习的发展历程、3大研究步骤、
    李宏毅深度学习笔记---课程介绍
  • 原文地址:https://www.cnblogs.com/Winniechen/p/9042830.html
Copyright © 2020-2023  润新知