bzoj2850巧克力王国 K-Dtree

2850: 巧克力王国

Time Limit: 60 Sec   Memory Limit: 512 MB
Submit: 800   Solved: 296
[ 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

Violet 0

昨天晚上KDtree调到11:30，几近崩溃的时候终于A了
嗯，裸的KDtree，以xy两个维度建一颗KDtree，仍然保存最大最小xy信息，还有美味度之和，判断的时候如果所有最大最小都符合小于C，那么整块的巧克力都可以吃掉。还有，若没有一块可行，整块巧克力都不能吃。
写评估函数的时候只判断可以吃掉的情况，没有排除不能吃的情况，结果TLE到死，T了1个小时。GG看来KDtree评估函数真的很重要，写得好写得差差了个十万八千里

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<cmath>
#define maxn 55000
using namespace std;
int read()
{
    char ch = getchar(); int x = 0, f = 1;
    while(ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
    while(ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}

int D;
struct tree {
	int d[2], mn[2], mx[2], l, r, val;
	long long sum;
	bool operator < (const tree &b) const {
		return d[D] < b.d[D];
	}
}t[maxn], a[maxn];
int n, m, sz, root;
long long A, B, C, ans;

void update(int p) {
	int ls = t[p].l, rs = t[p].r;
	t[p].sum = t[ls].sum + t[rs].sum + t[p].val;
	for(int i = 0;i <= 1; ++i) {
		t[p].mn[i] = t[p].mx[i] = t[p].d[i];
		if(ls) {
			t[p].mn[i] = min(t[p].mn[i], t[ls].mn[i]);
			t[p].mx[i] = max(t[p].mx[i], t[ls].mx[i]);
		}
		if(rs) {
			t[p].mn[i] = min(t[p].mn[i], t[rs].mn[i]);
			t[p].mx[i] = max(t[p].mx[i], t[rs].mx[i]);
		}
	}
}

void KD_build(int &p, int L, int R, int d) {
	D = d; int mid = L + R >> 1;
	p = mid;
	nth_element(a + L, a + mid, a + R + 1);
	t[mid] = a[mid];
	if(L < mid) KD_build(t[mid].l, L, mid - 1, d ^ 1);
	if(R > mid) KD_build(t[mid].r, mid + 1, R, d ^ 1);
	update(mid);
}

bool judge(int x, int y) {return A * x + B * y < C;}
int eval(int p) {
	int ret = 0;
	ret += judge(t[p].mn[0], t[p].mn[1]);
	ret += judge(t[p].mn[0], t[p].mx[1]);
	ret += judge(t[p].mx[0], t[p].mn[1]);
	ret += judge(t[p].mx[0], t[p].mx[1]);
	return ret;
}

void KD_query(int p) {
	int ls = t[p].l, rs = t[p].r;
	if(judge(t[p].d[0], t[p].d[1])) ans += t[p].val;
	if(ls) {
		int g = eval(ls);
		if(g == 4) ans += t[ls].sum;
		else if(g) KD_query(ls);
	}
	if(rs) {
		int g = eval(rs);
		if(g == 4) ans += t[rs].sum;
		else if(g) KD_query(rs);
	}
}

void init() {
	n = read(); m = read();
	for(int i = 1;i <= n; ++i) {
		a[i].d[0] = read(); a[i].d[1] = read();
		a[i].val = read();
	}
	KD_build(root, 1, n, 0);
}

void solve() {
	while(m--) {
		A = read(); B = read(); C = read();
		ans = 0; KD_query(root);
		printf("%lld\n", ans);
	}
}

int main()
{
	init();
	solve(); 
	return 0;
}