洛谷P2184 贪婪大陆

题目

思路

修改区间时在开始位置(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
",ask_s(1,y,1)-ask_e(1,x-1,1));
	}
	return 0;
}
原文地址:https://www.cnblogs.com/poi-bolg-poi/p/11122954.html