BZOJ2850 巧克力王国

本文介绍了一种使用K-D Tree解决特定二维平面问题的方法,即求满足线性不等式的点集的点权和。通过代码示例详细展示了K-D Tree的构建过程和查询算法,为解决类似问题提供了有效途径。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目就是让你求满足ax+by<c的点的点权和

这个……KDT可以过,不要问我为什么

以后看到不会的二维平面问题是不是都可以上KDT然后相信可以过啊-_-

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<bitset>
using namespace std;
#define MAXN 50010
#define MAXM 1010
#define ll long long
#define INF 1000000000
#define MOD 1000000007
#define eps 1e-8
struct pt{
	ll x;
	ll y;
	ll v;
	pt(){
		
	}
	pt(ll _x,ll _y,ll _v){
		x=_x;
		y=_y;
		v=_v;
	}
};
ll tot;
ll rt;
pt p[MAXN];
ll son[MAXN][2];
pt L[MAXN],R[MAXN],v[MAXN];
ll sum[MAXN];
ll n,m;
ll a,b,c;
bool cmpx(pt x,pt y){
	return x.x<y.x;
}
bool cmpy(pt x,pt y){
	return x.y<y.y;
}
bool (*cmp[])(pt x,pt y)={cmpx,cmpy};
inline void ud(ll x){
	L[x]=R[x]=v[x];
	sum[x]=v[x].v+sum[son[x][0]]+sum[son[x][1]];
	if(son[x][0]){
		L[x].x=min(L[x].x,L[son[x][0]].x);
		L[x].y=min(L[x].y,L[son[x][0]].y);
		R[x].x=max(R[x].x,R[son[x][0]].x);
		R[x].y=max(R[x].y,R[son[x][0]].y);
	}
	if(son[x][1]){
		L[x].x=min(L[x].x,L[son[x][1]].x);
		L[x].y=min(L[x].y,L[son[x][1]].y);
		R[x].x=max(R[x].x,R[son[x][1]].x);
		R[x].y=max(R[x].y,R[son[x][1]].y);
	}
}
void build(ll &x,ll y,ll z,bool f){
	if(y>z){
		return ;
	}
	if(!x){
		x=++tot;
	}
	ll mid=y+z>>1;
	nth_element(p+y,p+mid,p+z+1,cmp[f]);
	v[x]=p[mid];
	build(son[x][0],y,mid-1,f^1);
	build(son[x][1],mid+1,z,f^1);
	ud(x);
}
bool jud(ll x,ll y){
	return a*x+b*y<c;
}
ll ask(ll x){
	if(!x){
		return 0;
	}
	if(jud(L[x].x,L[x].y)&&jud(L[x].x,R[x].y)&&jud(R[x].x,L[x].y)&&jud(R[x].x,R[x].y)){
		return sum[x];
	}
	if((!jud(L[x].x,L[x].y))&&(!jud(L[x].x,R[x].y))&&(!jud(R[x].x,L[x].y))&&(!jud(R[x].x,R[x].y))){
		return 0;
	}
	if(jud(v[x].x,v[x].y)){
		return v[x].v+ask(son[x][0])+ask(son[x][1]);
	}
	return ask(son[x][0])+ask(son[x][1]);
}
int main(){
	ll i,x,y,z;
	scanf("%lld%lld",&n,&m);
	for(i=1;i<=n;i++){
		scanf("%lld%lld%lld",&x,&y,&z);
		p[i]=pt(x,y,z);
	}
	build(rt,1,n,0);
	while(m--){
		scanf("%lld%lld%lld",&a,&b,&c);
		printf("%lld\n",ask(rt));
	}
	return 0;
}

/*
3 3
1 2 5
3 1 4
2 2 1
2 1 6
1 3 5
1 3 7

*/


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值