BZOJ2850: 巧克力王国
Description
巧克力王国里的巧克力都是由牛奶和可可做成的。但是并不是每一块巧克力都受王国人民的欢迎,因为大家都不喜
欢过于甜的巧克力。对于每一块巧克力,我们设x和y为其牛奶和可可的含量。由于每个人对于甜的程度都有自己的
评判标准,所以每个人都有两个参数a和b,分别为他自己为牛奶和可可定义的权重,因此牛奶和可可含量分别为x
和y的巧克力对于他的甜味程度即为ax + by。而每个人又有一个甜味限度c,所有甜味程度大于等于c的巧克力他都
无法接受。每块巧克力都有一个美味值h。现在我们想知道对于每个人,他所能接受的巧克力的美味值之和为多少
Input
第一行两个正整数n和m,分别表示巧克力个数和询问个数。接下来n行,每行三个整数x,y,h,含义如题目所示。再
接下来m行,每行三个整数a,b,c,含义如题目所示。
Output
输出m行,其中第i行表示第i个人所能接受的巧克力的美味值之和。
Sample Input
3 3
1 2 5
3 1 4
2 2 1
2 1 6
1 3 5
1 3 7
1 2 5
3 1 4
2 2 1
2 1 6
1 3 5
1 3 7
Sample Output
5
0
4
0
4
HINT
1 <= n, m <= 50000,1 <= h <= 10^9,-10^9 <= a, b, x, y <= 10^9。
k-d树模板题,据说时间复杂度为每次询问O(sqrt(N))。
#include<cstdio> #include<cctype> #include<queue> #include<cstring> #include<algorithm> #define L T[o].lc #define R T[o].rc #define rep(i,s,t) for(int i=s;i<=t;i++) #define dwn(i,s,t) for(int i=s;i>=t;i--) #define ren for(int i=first[x];i;i=next[i]) using namespace std; const int BufferSize=1<<16; char buffer[BufferSize],*head,*tail; inline char Getchar() { if(head==tail) { int l=fread(buffer,1,BufferSize,stdin); tail=(head=buffer)+l; } return *head++; } inline int read() { int x=0,f=1;char c=Getchar(); for(;!isdigit(c);c=Getchar()) if(c=='-') f=-1; for(;isdigit(c);c=Getchar()) x=x*10+c-'0'; return x*f; } typedef long long ll; const int maxn=50010; const int inf=2e9; int D; struct Node { int mx[2],mn[2],x[2],lc,rc,v; ll s; bool operator < (const Node& ths) const {return x[D]<ths.x[D];} }T[maxn],x; void maintain(int o) { T[o].s=T[L].s+T[R].s+T[o].v; rep(c,0,1) { T[o].mn[c]=min(min(T[L].mn[c],T[R].mn[c]),T[o].x[c]); T[o].mx[c]=max(max(T[L].mx[c],T[R].mx[c]),T[o].x[c]); } } void build(int& o,int l,int r,int cur) { o=0;if(l>r) return; int mid=l+r>>1;D=cur;nth_element(T+l,T+mid,T+r+1); o=mid;build(L,l,mid-1,cur^1);build(R,mid+1,r,cur^1); maintain(o); } int a,b,c; int in(int x,int y) {return (ll)a*x+(ll)b*y<c;} int check(int o) { int cnt=0; if(in(T[o].mn[0],T[o].mn[1])) cnt++; if(in(T[o].mn[0],T[o].mx[1])) cnt++; if(in(T[o].mx[0],T[o].mn[1])) cnt++; if(in(T[o].mx[0],T[o].mx[1])) cnt++; if(!cnt) return 0; if(cnt==4) return 1; return -1; } ll query(int o) { if(!o) return 0; int f=check(o); if(!f) return 0; if(f==1) return T[o].s; return query(L)+query(R)+T[o].v*in(T[o].x[0],T[o].x[1]); } int n,m,rt; int main() { T[0].mn[0]=T[0].mn[1]=inf; T[0].mx[0]=T[0].mx[1]=-inf; n=read();m=read(); rep(i,1,n) T[i].x[0]=read(),T[i].x[1]=read(),T[i].v=read(); build(rt,1,n,0); rep(i,1,m) { a=read();b=read();c=read(); printf("%lld\n",query(rt)); } return 0; }