lugou P2184 贪婪大陆(线段树)

题目描述
小 FF 最后一道防线是一条长度为 nn 的战壕,小 FF 拥有无数多种地雷,而 SCV 每次可以在 [L, R]区间埋放同一种不同于之前已经埋放的地雷。由于情况已经十万火急,小 FF 在某些时候可能会询问你在 [L,R]区间内有多少种不同的地雷,他希望你能尽快的给予答复。

输入格式
第一行为两个整数n和m,n表示防线长度,m 表示 SCV 布雷次数及小 FF 询问的次数总和。

接下来有m 行,每行三个整数 q,l,r:

若 q=1,则表示 SCV 在 [l, r]这段区间布上一种地雷;
若 q=2,则表示小 FF 询问当前 [l, r] 区间总共有多少种地雷。

第一眼看到就可以发现是线段树,但如何对它进行维护呢?很显然,直接维护是做不到的,因为区间统计时可能会有重复。其实不难看出两个端点是重中之重,我们应该从两个端点下手,怎么操作呢,可以维护两个东西,一个suml(起点),sumr(终点),每次操作就在起点(suml)和终点(sumr)++,让其次查询的时候求出1~r的suml - 1~l的sumr就可以了。
至于为什么这么操作,嘿嘿,我们可以将1~l分解成 l~r , 1~l 区间的suml与sumr可以互相抵消,那么剩下的就是l~r的suml。

#include<bits/stdc++.h>
using namespace std;
int n,m;
int len=0;
int root;
struct node
{
	int l,r,lc,rc,suml,sumr;
};
node e[500000];
int bt(int x,int y)
{
	len++;
	int now=len,lc=-1,rc=-1;
	if(x==y) ;
	else
	{
		int mid=(x+y)/2;
		lc=bt(x,mid);
		rc=bt(mid+1,y);
	}
	e[now]={x,y,lc,rc,0,0};
	return now;
}
void cg1(int now,int x)
{
	int l=e[now].l,r=e[now].r;
	if(l==r)
	{
		e[now].suml++;
		return ;
	}
	else 
	{
		int mid=(l+r)/2;
		int lc=e[now].lc,rc=e[now].rc;
		if(x>=mid+1) cg1(rc,x);
		else cg1(lc,x);
		e[now].suml=e[lc].suml+e[rc].suml;
		return ;
	}
}
void cg2(int now,int x)
{
	int l=e[now].l,r=e[now].r;
	if(l==r)
	{
		e[now].sumr++;
		return ;
	}
	else 
	{
		int mid=(l+r)/2;
		int lc=e[now].lc,rc=e[now].rc;
		if(x>=mid+1) cg2(rc,x);
		else cg2(lc,x);
		e[now].sumr=e[lc].sumr+e[rc].sumr;
		return ;
	}
}
int find1(int now,int x,int y)
{
	int l=e[now].l,r=e[now].r;
	if(l==x&&y==r) 
	{
		return e[now].suml;
	}
	else
	{
		int lc=e[now].lc,rc=e[now].rc;
		int mid=(l+r)/2;
		if(x>=mid+1) return find1(rc,x,y);
		else if(y<=mid) return find1(lc,x,y);
		else return find1(lc,x,mid)+find1(rc,mid+1,y);
	}
}
int find2(int now,int x,int y)
{
	int l=e[now].l,r=e[now].r;
	if(l==x&&y==r) 
	{
		return e[now].sumr;
	}
	else
	{
		int lc=e[now].lc,rc=e[now].rc;
		int mid=(l+r)/2;
		if(x>=mid+1) return find2(rc,x,y);
		else if(y<=mid) return find2(lc,x,y);
		else return find2(lc,x,mid)+find2(rc,mid+1,y);
	}
}
int find(int x,int y)
{
	return find1(root,1,y)-find2(root,1,x-1);
}
int main()
{
	scanf("%d%d",&n,&m);
	root=bt(1,n);
	for(int i=1;i<=m;i++)
	{
		int p,x,y;
		scanf("%d",&p);
		if(p==1)
		{
			scanf("%d%d",&x,&y);
//			printf("-");
			cg1(root,x);
			cg2(root,y);
		}
		else 
		{
			scanf("%d%d",&x,&y);
			if(x>y) swap(x,y);
			printf("%d\n",find(x,y));	
		}
	}
	return 0;
 } 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值