bzoj4418 [Shoi2013]扇形面积并 扫描线+二分+树状数组

Description

---

给定N个同心的扇形，求有多少面积，被至少K个扇形所覆盖。

对于100%的数据，1≤n≤10⁵, 1≤m≤10⁶,1≤k≤5000,1≤ri≤10⁵,-m≤a1,a2≤m

Solution

---

我们把圆心拉成直线，然后就变成了矩形覆盖。扫描线做就可以了
由于一定是扇形，因此覆盖次数一定是单调的。线段树维护差分数组可以在线段树上二分做到一个log，二分+树状数组也可以跑得很快

一开始写线段树被卡了。。

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define lowbit(x) (x&-x)

typedef long long LL;
const int N=200005;

struct Q {
	int x,r;
} q[N*4];

LL s[N<<2];
int n,m,k,maxR;

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void add(int x,int v) {
	for (;x<=maxR;x+=lowbit(x)) s[x]+=v;
}

int get(int x) {
	int res=0;
	for (;x;x-=lowbit(x)) res+=s[x];
	return res;
}

bool cmp(Q a,Q b) {
	return a.x<b.x;
}

int calc() {
	int l=1,r=maxR;
	for (;l<=r;) {
		int mid=(l+r)>>1;
		if (get(mid)>=k) l=mid+1;
		else r=mid-1;
	}
	return l-1;
}

int main(void) {
	freopen("xiaoqiao.in","r",stdin);
	freopen("xiaoqiao.out","w",stdout);
	n=read(),m=read(),k=read();
	int cnt=0;
	rep(i,1,n) {
		int r=read(),st=read(),ed=read();
		maxR=std:: max(maxR,r);
		st+=m; ed+=m;
		if (st>ed) {
			q[++cnt]=(Q) {st,r};
			q[++cnt]=(Q) {m*2,-r};
			q[++cnt]=(Q) {0,r};
			q[++cnt]=(Q) {ed,-r};
		} else {
			q[++cnt]=(Q) {st,r};
			q[++cnt]=(Q) {ed,-r};
		}
	}
	std:: sort(q+1,q+cnt+1,cmp);
	LL ans=0;
	for (int i=0,j=1;i<=m*2;i++) {
		for (int res,r;j<=cnt&&q[j].x<=i;j++) {
			res=calc();
			ans+=1LL*res*res*(q[j].x-q[j-1].x);
			r=abs(q[j].r);
			add(1,q[j].r/r); add(r+1,-q[j].r/r);
		}
	}
	printf("%lld\n", ans);
	return 0;
}