洛谷P2184 贪婪大陆

poi-bolg-poi 2021-08-27 原文

题目
思路
$Code$

题目

思路

修改区间时在开始位置$s+1$表示从则例开始埋一种雷，在结束位置$e+1$表示有一种雷埋到这里就结束了。
查询时输出$[1,r]$s的和减去$[1，l-1]$的e的和。用线段树来实现。

$Code$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
const int MAXN=1000001;
int n,m;
struct node{
	int l,r,s,e;
}tree[MAXN+1];
inline int read(){
	int x=0;bool f=0;char c=getchar();
	while(c<\'0\'||c>\'9\'){if(c==\'-\')f=!f;c=getchar();}
	while(c>=\'0\'&&c<=\'9\'){x=x*10+c-\'0\';c=getchar();}
	return f?-x:x;
}
void build(int l,int r,int now){
	tree[now].l=l,tree[now].r=r;
	if(l==r){
		tree[now].s=0;tree[now].e=0;
		return;
	}
	int mid=(l+r)>>1;
	build(l,mid,now<<1);
	build(mid+1,r,now<<1|1);
}
void a_s(int x,int now){
	if(tree[now].l==tree[now].r){
		tree[now].s+=1;
		return;
	}
	int mid=(tree[now].l+tree[now].r)>>1;
	if(x<=mid) a_s(x,now<<1);
	else a_s(x,now<<1|1);
	tree[now].s=tree[now<<1].s+tree[now<<1|1].s;
}
void a_e(int x,int now){
	if(tree[now].l==tree[now].r){
		tree[now].e+=1;
		return;
	}
	int mid=(tree[now].l+tree[now].r)>>1;
	if(x<=mid) a_e(x,now<<1);
	else a_e(x,now<<1|1);
	tree[now].e=tree[now<<1].e+tree[now<<1|1].e;
}
int ask_s(int x,int y,int now){
	if(tree[now].l>=x&&tree[now].r<=y){
		return tree[now].s;
	}
	int mid=(tree[now].l+tree[now].r)>>1,ans=0;
	if(x<=mid) ans+=ask_s(x,y,now<<1);
	if(y>mid) ans+=ask_s(x,y,now<<1|1);
	return ans;
}
int ask_e(int x,int y,int now){
	if(tree[now].l>=x&&tree[now].r<=y){
		return tree[now].e;
	}
	int mid=(tree[now].l+tree[now].r)>>1,ans=0;
	if(x<=mid) ans+=ask_e(x,y,now<<1);
	if(y>mid) ans+=ask_e(x,y,now<<1|1);
	return ans;
}

int main(){
	n=read(),m=read();
	build(1,n,1);
	int bz,x,y;
	while(m--){
		bz=read(),x=read(),y=read();
		if(bz==1) a_s(x,1),a_e(y,1);
		else printf("%d\n",ask_s(1,y,1)-ask_e(1,x-1,1));
	}
	return 0;
}