Submission #622976

# Submission time Handle Problem Language Result Execution time Memory
622976 2022-08-04T22:24:11 Z izanbf Examination (JOI19_examination) C++14
100 / 100
2600 ms 62248 KB
#include <bits/stdc++.h>
using namespace std;
 
struct Point {
	int x, y, z;
};
 
struct Query {
	int a, b, c, id;
};
 
const int B = 350;
const int MAXSZ = 6e5+9;

int bit[MAXSZ]; // binary/fenwick indexed tree
int sz; // bit size

int bit_sum(int r) {
    int ret = 0;
    for (; r >= 0; r = (r & (r + 1)) - 1)
        ret += bit[r];
    return ret;
}

int bit_sum(int l, int r) {
    return bit_sum(r) - bit_sum(l - 1);
}

void bit_add(int idx, int delta) {
    for (; idx < sz; idx = idx | (idx + 1))
        bit[idx] += delta;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
 
	int N, Q;
	cin >> N >> Q;
 
	set<int> values;
 
	vector<Point> pts(N);
	for (int i = 0; i < N; ++i) {
		Point& p = pts[i];
		cin >> p.x >> p.y;
		p.z = p.x + p.y;
		values.insert(p.x);
		values.insert(p.y);
		values.insert(p.z);
	}
 
	vector<Query> qs(Q);
	for (int i = 0; i < Q; ++i) {
		Query& q = qs[i];
		cin >> q.a >> q.b >> q.c;
		q.id = i;
		values.insert(q.a);
		values.insert(q.b);
		values.insert(q.c);
	}
 
 
	unordered_map<int,int> comp;
	for (int v : values) {
		comp[v] = comp.size();
	}
 
	for (Query& q : qs) {
		q.a = comp[q.a];
		q.b = comp[q.b];
		q.c = comp[q.c];
	}
 
	for (Point& p : pts) {
		p.x = comp[p.x];
		p.y = comp[p.y];
		p.z = comp[p.z];
	}
 
	sz = comp.size() + 2;
 
	sort(qs.begin(), qs.end(), [](const Query& l, const Query& r) {
		return l.c < r.c;
	});
 
	sort(pts.begin(), pts.end(), [](const Point& l, const Point& r) {
		return l.z < r.z;
	});
 
	int blocks = (N+B-1)/B;
	vector<vector<Query>> block_queries(blocks);
	int j = 0;
 
	for (int block = 1; block < blocks; ++block) {
		int i = B*block;
		while (j < Q and qs[j].c <= pts[i].z) {
			block_queries[block-1].push_back(qs[j]);
			++j;
		}
	}
	while (j < Q) {
		block_queries.back().push_back(qs[j]);
		++j;
	}
 
	vector<int> ans(Q);
 
	for (int block = 0; block < blocks; ++block) {
		if (block_queries[block].empty()) continue;
 
		memset(bit, 0, sizeof(int)*sz);
 
		vector<Point> in_block;
		in_block.reserve(B);
		for (int i = B*block; i < N and i < B*(block+1); ++i) {
			in_block.push_back(pts[i]);
		}
 
		vector<Point> over_block;
		over_block.reserve(N-B*block);
		for (int i = B*(block+1); i < N; ++i) {
			over_block.push_back(pts[i]);
			bit_add(pts[i].y, 1);
		}
 
		sort(over_block.begin(), over_block.end(), [&](const Point& l, const Point& r) {
			return l.x > r.x;
		});
 
		sort(block_queries[block].begin(), block_queries[block].end(), [&](const Query& l, const Query& r) {
			return l.a < r.a;
		});
 
		for (const Query& q : block_queries[block]) {
			int cnt_in = 0;
			int cnt_over = 0;
 
			// points in block
			for (const Point& p : in_block) {
				if (p.x >= q.a and p.y >= q.b and p.z >= q.c) ++cnt_in;
			}
 
			// points over block
			while ((not over_block.empty()) and over_block.back().x < q.a) {
				const Point& p = over_block.back();
				bit_add(p.y, -1);
				over_block.pop_back();
			}

			cnt_over = bit_sum(q.b, sz-1);
 
			ans[q.id] = cnt_in + cnt_over;
		}
	}
 
	for (int i = 0; i < Q; ++i) {
		cout << ans[i] << '\n';
	}
}
# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 12 ms 2188 KB Output is correct
8 Correct 11 ms 2132 KB Output is correct
9 Correct 11 ms 2200 KB Output is correct
10 Correct 8 ms 1680 KB Output is correct
11 Correct 9 ms 1620 KB Output is correct
12 Correct 5 ms 468 KB Output is correct
13 Correct 8 ms 1948 KB Output is correct
14 Correct 9 ms 1876 KB Output is correct
15 Correct 10 ms 1876 KB Output is correct
16 Correct 5 ms 1108 KB Output is correct
17 Correct 7 ms 1328 KB Output is correct
18 Correct 2 ms 468 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 221 ms 19336 KB Output is correct
2 Correct 228 ms 19276 KB Output is correct
3 Correct 238 ms 19360 KB Output is correct
4 Correct 218 ms 15968 KB Output is correct
5 Correct 184 ms 15892 KB Output is correct
6 Correct 86 ms 6200 KB Output is correct
7 Correct 240 ms 19204 KB Output is correct
8 Correct 234 ms 19140 KB Output is correct
9 Correct 240 ms 18840 KB Output is correct
10 Correct 152 ms 15244 KB Output is correct
11 Correct 149 ms 15164 KB Output is correct
12 Correct 91 ms 6084 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 221 ms 19336 KB Output is correct
2 Correct 228 ms 19276 KB Output is correct
3 Correct 238 ms 19360 KB Output is correct
4 Correct 218 ms 15968 KB Output is correct
5 Correct 184 ms 15892 KB Output is correct
6 Correct 86 ms 6200 KB Output is correct
7 Correct 240 ms 19204 KB Output is correct
8 Correct 234 ms 19140 KB Output is correct
9 Correct 240 ms 18840 KB Output is correct
10 Correct 152 ms 15244 KB Output is correct
11 Correct 149 ms 15164 KB Output is correct
12 Correct 91 ms 6084 KB Output is correct
13 Correct 1753 ms 21944 KB Output is correct
14 Correct 1373 ms 19616 KB Output is correct
15 Correct 300 ms 19288 KB Output is correct
16 Correct 1154 ms 16868 KB Output is correct
17 Correct 1022 ms 16856 KB Output is correct
18 Correct 139 ms 6736 KB Output is correct
19 Correct 1762 ms 21888 KB Output is correct
20 Correct 1799 ms 20856 KB Output is correct
21 Correct 1554 ms 21784 KB Output is correct
22 Correct 145 ms 15296 KB Output is correct
23 Correct 154 ms 15300 KB Output is correct
24 Correct 75 ms 6080 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 12 ms 2188 KB Output is correct
8 Correct 11 ms 2132 KB Output is correct
9 Correct 11 ms 2200 KB Output is correct
10 Correct 8 ms 1680 KB Output is correct
11 Correct 9 ms 1620 KB Output is correct
12 Correct 5 ms 468 KB Output is correct
13 Correct 8 ms 1948 KB Output is correct
14 Correct 9 ms 1876 KB Output is correct
15 Correct 10 ms 1876 KB Output is correct
16 Correct 5 ms 1108 KB Output is correct
17 Correct 7 ms 1328 KB Output is correct
18 Correct 2 ms 468 KB Output is correct
19 Correct 221 ms 19336 KB Output is correct
20 Correct 228 ms 19276 KB Output is correct
21 Correct 238 ms 19360 KB Output is correct
22 Correct 218 ms 15968 KB Output is correct
23 Correct 184 ms 15892 KB Output is correct
24 Correct 86 ms 6200 KB Output is correct
25 Correct 240 ms 19204 KB Output is correct
26 Correct 234 ms 19140 KB Output is correct
27 Correct 240 ms 18840 KB Output is correct
28 Correct 152 ms 15244 KB Output is correct
29 Correct 149 ms 15164 KB Output is correct
30 Correct 91 ms 6084 KB Output is correct
31 Correct 1753 ms 21944 KB Output is correct
32 Correct 1373 ms 19616 KB Output is correct
33 Correct 300 ms 19288 KB Output is correct
34 Correct 1154 ms 16868 KB Output is correct
35 Correct 1022 ms 16856 KB Output is correct
36 Correct 139 ms 6736 KB Output is correct
37 Correct 1762 ms 21888 KB Output is correct
38 Correct 1799 ms 20856 KB Output is correct
39 Correct 1554 ms 21784 KB Output is correct
40 Correct 145 ms 15296 KB Output is correct
41 Correct 154 ms 15300 KB Output is correct
42 Correct 75 ms 6080 KB Output is correct
43 Correct 2600 ms 62168 KB Output is correct
44 Correct 2560 ms 62248 KB Output is correct
45 Correct 2189 ms 62092 KB Output is correct
46 Correct 1739 ms 45160 KB Output is correct
47 Correct 1404 ms 45156 KB Output is correct
48 Correct 140 ms 6512 KB Output is correct
49 Correct 1923 ms 61960 KB Output is correct
50 Correct 2488 ms 62176 KB Output is correct
51 Correct 1836 ms 61888 KB Output is correct
52 Correct 1028 ms 34684 KB Output is correct
53 Correct 273 ms 26260 KB Output is correct