[bzoj2850]2850: 巧克力王国

2850: 巧克力王国

Time Limit: 60 Sec  Memory Limit: 512 MB
[Submit][Status][Discuss]

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

Sample Output

5
0
4

HINT

1 <= n, m <= 50000，1 <= 10^9，-10^9 <= a, b, x, y <= 10^9。

 

Source

我对kdtree不到位，只能看模板

此题据说不是kdtree，但是能过

输入不是lld竟然WA了

#include<algorithm>#include<iostream>#include<cstdio>#include<cmath>using namespace std;const int N = 50000 + 5;typedef long long ll;ll A,B,C,ans;int F,n,m,root;struct node{int d[2],mx[2],mn[2],v,l,r; ll sum;int& operator[]( int x ){return d[x];}friend bool operator<(node a,node b){ return a[F] < b[F]; }}p[N];bool check( int x, int y ){ return A*x+B*y < C; }int cal( node x ){int res=0;res += check(x.mn[0],x.mn[1]);res += check(x.mx[0],x.mn[1]);res += check(x.mn[0],x.mx[1]);res += check(x.mx[0],x.mx[1]);return res;}struct kdtree{node t[N];void update( int k ){int l = t[k].l, r = t[k].r;for( int i = 0; i < 2; i++ ){t[k].mn[i] = t[k].mx[i] = t[k][i];if(l) t[k].mn[i] = min(t[k].mn[i],t[l].mn[i]);if(l) t[k].mx[i] = max(t[k].mx[i],t[l].mx[i]);if(r) t[k].mn[i] = min(t[k].mn[i],t[r].mn[i]);if(r) t[k].mx[i] = max(t[k].mx[i],t[r].mx[i]);}t[k].sum = t[l].sum + t[r].sum + t[k].v;}int build( int l, int r, int now ){F = now; int mid = (l+r)>>1;nth_element(p+l,p+mid,p+r+1);t[mid] = p[mid];if( l < mid ) t[mid].l = build( l, mid-1, now^1 );if( r > mid ) t[mid].r = build( mid+1, r, now^1 );update(mid); return mid;}void query( int k ){int l = t[k].l, r = t[k].r;if( check(t[k][0],t[k][1]) ) ans += t[k].v;int tl = 0, tr = 0;if(l) tl = cal(t[l]);if(r) tr = cal(t[r]);if( tl == 4 ) ans += t[l].sum;else if( tl ) query(l);if( tr == 4 ) ans += t[r].sum;else if( tr ) query(r);}}kd;int main(){scanf("%d%d", &n, &m);for( int i = 1; i <= n; i++ ) scanf("%lld%lld%lld", &p[i][0], &p[i][1], &p[i].v);root = kd.build( 1, n, 0 );while( m-- ){scanf("%lld%lld%lld", &A, &B, &C);ans = 0; kd.query(root);printf("%lld\n", ans);}return 0;}