51nod 2014 小朋友的笑话(势能分析线段树)

版权声明:本文为博主原创文章,未经博主允许必须转载。 https://blog.csdn.net/qq_35950004/article/details/88780448

小O是一个很萌很萌的女孩子。
有一天小O叫了很多很多萌萌哒小朋友到家里来玩。
由于太无聊了,她们开始讲笑话。
总共有N个小朋友排成一排,编号1~N。
在某个时刻,会有编号为xi的小朋友看到了笑话li,然后她会把这个笑话讲出来,与她距离不超过ki的小朋友都会听到这个笑话。
当一个小朋友听到一个笑话时,如果她是第一次听得到这个笑话,那么她会觉得这个笑话非常好笑,笑的停不下来。如果她听到之前就是在笑的她会继续笑。
如果她之前听过这个笑话,那么她会觉得这个笑话非常无聊,她会立即停止笑。

现在小O想知道,在某些时刻一段区间内有多少小朋友在笑。

第一行两个整数N,M,表示小朋友的数量和事件数量,(1<=N,M<=100000)
接下来M行
第一行为一个正整数op(1<=op<=2),表示事件类型。
如果op=1,接下来三个整数xi li ki,表示该时间点小朋友xi(1<=xi<=N)讲了一个笑话li(1<=li<=100000),传播距离为ki(0<=ki<=N)。
如果op=2,接下来两个整数l r,表示该时间点小O想知道区间[l,r]中有多少小朋友在笑(1<=l<=r<=N)

对于一个op=1,可以发现就是听过笑话的人不笑,没听过的人笑,只要知道那些人听过,相当于区间赋值01,可以用一颗普通线段树维护。
那么我们开100000棵动态开点线段树,每颗存一个笑话有哪些人听过,如果一个线段上都没听过,或都听过,我们就可以在普通线段树上区间赋值。否则这个区间是不统一的,就下去继续线段树的操作。
发现我们只会把人从没变过变成听过。那么可以势能分析,我们让一个区间听过,最多可以使 log \log 个线段变成不统一的,势能增加 log n \log n 。复杂度 O ( n log n ) O(n\log n) ?
询问就很简单了,线段树基操。

PS:我以为动态开点线段树会MLE,因为需要标记下传。。。。。。所以我打了珂朵莉树。但是,你时间复杂度都是 O ( n log n ) O(n \log n) 的,动态开点的内存难道还会大于 O ( n log n ) O(n \log n) ?。。。

AC Code:

#include<bits/stdc++.h>
#define maxn 100005
#define F first
#define S second
#define lc now<<1
#define rc now<<1|1
using namespace std;

int n,m;
struct node{
	int l,r,col;
	node(int l=0,int r=0,int col=0):l(l),r(r),col(col){}
	bool operator <(const node &B)const{ return l < B.l; }
};
set<node >st[maxn];
set<node >::iterator it,it2;

int sum[maxn<<3],tag[maxn<<3],siz[maxn<<3];

void Build(int now,int l,int r){
	siz[now] = r - l + 1;
	tag[now] = -1;
	if(l == r) return;
	int mid = (l+r) >> 1;
	Build(lc,l,mid) , Build(rc,mid+1,r);
}

void dt(int now){
	if(tag[now] == 0){
		sum[now] = 0;
		tag[lc] = tag[rc] = 0;
	}
	if(tag[now] == 1){
		sum[now] = siz[now];
		tag[lc] = tag[rc] = 1;
	}
	tag[now] = -1;
}

void upd(int now){
	sum[now] = sum[lc] + sum[rc];
}

void Insert(int now,int l,int r,int ql,int qr,int col){
	dt(now);
	if(l > qr || ql > r) return;
	if(ql<=l && r<=qr){
		tag[now] = col;
		dt(now);
		return;
	}
	int mid = (l+r) >> 1;
	Insert(lc,l,mid,ql,qr,col);
	Insert(rc,mid+1,r,ql,qr,col);
	upd(now);
}

int Query(int now,int l,int r,int ql,int qr){
	dt(now);
	if(l > qr || ql > r) return 0;
	if(ql<=l && r<=qr) return sum[now];
	int mid = (l+r) >> 1;
	int ret = Query(lc,l,mid,ql,qr) + Query(rc,mid+1,r,ql,qr);
	upd(now);
	return ret;
}

int main(){
	scanf("%d%d",&n,&m);
	Build(1,1,n);
	for(int i=1;i<=100000;i++) st[i].insert(node(1,n,0));
	for(int i=1,op,l,r,x,k;i<=m;i++){
		scanf("%d",&op);
		if(op == 1){
			scanf("%d%d%d",&x,&l,&k);
			x-=k,k=2*k+x;
			x = max(x , 1) , k = min(k , n);
			it = st[l].upper_bound(x);
			it--;
			int mn = x , mx = k;
			for(;it!=st[l].end() && (*it).l<=k;){
				Insert(1,1,n,max((*it).l,x),min((*it).r,k),(*it).col!=1);
				if((*it).col == 1)
					mn = min(mn , (*it).l),
					mx = max(mx , (*it).r);
				it2 = it , it++;
				int L = (*it2).l , R = (*it2).r , C = (*it2).col;
				st[l].erase(it2);
				if(C == 0 && L < x)
					st[l].insert(node(L,x-1,0));
				if(C == 0 && R > k)
					st[l].insert(node(k+1,R,0));
			}
			st[l].insert(node(mn,mx,1));
		}
		else{
			scanf("%d%d",&l,&r);
			printf("%d\n",Query(1,1,n,l,r));
		}
	}
}

猜你喜欢

转载自blog.csdn.net/qq_35950004/article/details/88780448