답안 #267163

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
267163 2020-08-15T22:43:24 Z johutha Rectangles (IOI19_rect) C++17
72 / 100
5000 ms 704704 KB
#pragma GCC optimize("Ofast")

#include "rect.h"
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

struct fenw
{
	int n;
	vector<int> table;

	int query(int k)
	{
		int res = 0;
		for (; k; k -= (k & (-k))) res += table[k];
		return res;
	}

	void update(int k, int v)
	{
		k++;
		for (; k <= n; k += (k & (-k))) table[k] += v;
	}

	fenw(int in) : n(in), table(vector<int>(n + 1)) {}
};

struct bord
{
	int a, b;
	int st, en;

	bool operator<(const bord& ot)
	const {
		if (a != ot.a) return a < ot.a;
		if (b != ot.b) return b < ot.b;
		if (st != ot.st) return st < ot.st;
		return en < ot.en;
	}
};

vector<bord> compress(vector<bord> ip)
{
	vector<bord> res;
	sort(ip.begin(), ip.end());

	for (auto gp : ip)
	{
		if (res.empty() || res.back().a != gp.a || res.back().b != gp.b || res.back().en + 1 != gp.st) res.push_back(gp);
		else res.back().en = gp.en;
	}
	return res;
}

long long count_rectangles(vector<vector<signed>> ip)
{
	int r = ip.size();
	int c = ip[0].size();

	vector<bord> hor;
	vector<bord> ver;

	for (int y = 0; y < r; y++)
	{
		vector<pair<int,int>> st;
		for (int x = 0; x < c; x++)
		{
			int sts = st.size();

			for (int i = sts - 1; i >= 0; i--)
			{
				if (st[i].first + 1 < x) hor.push_back(bord{st[i].first, x, y, y});
				if (st[i].second >= ip[y][x]) break;
			}

			while (!st.empty() && st.back().second <= ip[y][x]) st.pop_back();
			st.emplace_back(x, ip[y][x]);
		}
	}

	for (int x = 0; x < c; x++)
	{
		vector<pair<int,int>> st;
		for (int y = 0; y < r; y++)
		{
			int sts = st.size();

			for (int i = sts - 1; i >= 0; i--)
			{
				if (st[i].first + 1 < y) ver.push_back(bord{st[i].first, y, x, x});
				if (st[i].second >= ip[y][x]) break;
			}

			while (!st.empty() && st.back().second <= ip[y][x]) st.pop_back();
			st.emplace_back(y, ip[y][x]);
		}
	}

	hor = compress(hor);
	ver = compress(ver);

	vector<int> lw[2500][2500];
	vector<int> rg[2500][2500];

	for (int i = 0; i < (int)hor.size(); i++)
	{
		for (int j = hor[i].st; j <= hor[i].en; j++)
		{
			rg[j][hor[i].b - 1].push_back(i);
		}
	}

	for (int i = 0; i < (int)ver.size(); i++)
	{
		for (int j = ver[i].st; j <= ver[i].en; j++)
		{
			lw[ver[i].b - 1][j].push_back(i);
		}
	}

	long long res = 0;

	for (int i = 0; i < r; i++)
	{
		for (int j = 0; j < c; j++)
		{
			vector<pair<int,int>> a;
			vector<pair<int,int>> b;
			for (auto k : lw[i][j]) a.emplace_back(j - ver[k].st + 1, ver[k].b - ver[k].a - 1);
			for (auto k : rg[i][j]) b.emplace_back(i - hor[k].st + 1, hor[k].b - hor[k].a - 1);
			
			sort(a.begin(), a.end());
			sort(b.begin(), b.end());

			auto b2 = b;
			for (auto &p : b2) swap(p.first, p.second);
			sort(b2.begin(), b2.end());
			for (auto &p : b2) swap(p.first, p.second);

			fenw f(b.size());

			int s = 0;
			for (int i = 0; i < (int)a.size(); i++)
			{
				while (s < (int)b2.size() && b2[s].second <= a[i].first)
				{
					int id = lower_bound(b.begin(), b.end(), b2[s]) - b.begin();
					f.update(id, 1);
					s++;
				}
				int lb = lower_bound(b.begin(), b.end(), make_pair(a[i].second, -1000)) - b.begin();
				res += f.query(b.size()) - f.query(lb);
			}
		}
	}

	return res;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 180 ms 293796 KB Output is correct
2 Correct 167 ms 293848 KB Output is correct
3 Correct 166 ms 293880 KB Output is correct
4 Correct 167 ms 293864 KB Output is correct
5 Correct 165 ms 293816 KB Output is correct
6 Correct 167 ms 293936 KB Output is correct
7 Correct 167 ms 293880 KB Output is correct
8 Correct 167 ms 293880 KB Output is correct
9 Correct 183 ms 293880 KB Output is correct
10 Correct 168 ms 293880 KB Output is correct
11 Correct 168 ms 293880 KB Output is correct
12 Correct 167 ms 293880 KB Output is correct
13 Correct 165 ms 293880 KB Output is correct
14 Correct 167 ms 293848 KB Output is correct
15 Correct 167 ms 294136 KB Output is correct
16 Correct 167 ms 293880 KB Output is correct
17 Correct 169 ms 294008 KB Output is correct
18 Correct 172 ms 293880 KB Output is correct
19 Correct 166 ms 293884 KB Output is correct
20 Correct 181 ms 294008 KB Output is correct
21 Correct 175 ms 293844 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 180 ms 293796 KB Output is correct
2 Correct 167 ms 293848 KB Output is correct
3 Correct 166 ms 293880 KB Output is correct
4 Correct 167 ms 293864 KB Output is correct
5 Correct 165 ms 293816 KB Output is correct
6 Correct 167 ms 293936 KB Output is correct
7 Correct 167 ms 293880 KB Output is correct
8 Correct 167 ms 293880 KB Output is correct
9 Correct 183 ms 293880 KB Output is correct
10 Correct 168 ms 293880 KB Output is correct
11 Correct 168 ms 293880 KB Output is correct
12 Correct 167 ms 293880 KB Output is correct
13 Correct 165 ms 293880 KB Output is correct
14 Correct 167 ms 293848 KB Output is correct
15 Correct 167 ms 294136 KB Output is correct
16 Correct 167 ms 293880 KB Output is correct
17 Correct 169 ms 294056 KB Output is correct
18 Correct 185 ms 294184 KB Output is correct
19 Correct 169 ms 294056 KB Output is correct
20 Correct 169 ms 294136 KB Output is correct
21 Correct 186 ms 294440 KB Output is correct
22 Correct 173 ms 294344 KB Output is correct
23 Correct 170 ms 294340 KB Output is correct
24 Correct 170 ms 294136 KB Output is correct
25 Correct 169 ms 294008 KB Output is correct
26 Correct 172 ms 293880 KB Output is correct
27 Correct 166 ms 293884 KB Output is correct
28 Correct 181 ms 294008 KB Output is correct
29 Correct 175 ms 293844 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 180 ms 293796 KB Output is correct
2 Correct 167 ms 293848 KB Output is correct
3 Correct 166 ms 293880 KB Output is correct
4 Correct 167 ms 293864 KB Output is correct
5 Correct 165 ms 293816 KB Output is correct
6 Correct 167 ms 293936 KB Output is correct
7 Correct 167 ms 293880 KB Output is correct
8 Correct 167 ms 293880 KB Output is correct
9 Correct 183 ms 293880 KB Output is correct
10 Correct 168 ms 293880 KB Output is correct
11 Correct 168 ms 293880 KB Output is correct
12 Correct 167 ms 293880 KB Output is correct
13 Correct 165 ms 293880 KB Output is correct
14 Correct 167 ms 293848 KB Output is correct
15 Correct 167 ms 294136 KB Output is correct
16 Correct 167 ms 293880 KB Output is correct
17 Correct 169 ms 294056 KB Output is correct
18 Correct 185 ms 294184 KB Output is correct
19 Correct 169 ms 294056 KB Output is correct
20 Correct 169 ms 294136 KB Output is correct
21 Correct 186 ms 294440 KB Output is correct
22 Correct 173 ms 294344 KB Output is correct
23 Correct 170 ms 294340 KB Output is correct
24 Correct 170 ms 294136 KB Output is correct
25 Correct 198 ms 295168 KB Output is correct
26 Correct 206 ms 295076 KB Output is correct
27 Correct 182 ms 295196 KB Output is correct
28 Correct 185 ms 295460 KB Output is correct
29 Correct 200 ms 296732 KB Output is correct
30 Correct 201 ms 296808 KB Output is correct
31 Correct 199 ms 296412 KB Output is correct
32 Correct 198 ms 296436 KB Output is correct
33 Correct 169 ms 294008 KB Output is correct
34 Correct 172 ms 293880 KB Output is correct
35 Correct 166 ms 293884 KB Output is correct
36 Correct 181 ms 294008 KB Output is correct
37 Correct 175 ms 293844 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 180 ms 293796 KB Output is correct
2 Correct 167 ms 293848 KB Output is correct
3 Correct 166 ms 293880 KB Output is correct
4 Correct 167 ms 293864 KB Output is correct
5 Correct 165 ms 293816 KB Output is correct
6 Correct 167 ms 293936 KB Output is correct
7 Correct 167 ms 293880 KB Output is correct
8 Correct 167 ms 293880 KB Output is correct
9 Correct 183 ms 293880 KB Output is correct
10 Correct 168 ms 293880 KB Output is correct
11 Correct 168 ms 293880 KB Output is correct
12 Correct 167 ms 293880 KB Output is correct
13 Correct 165 ms 293880 KB Output is correct
14 Correct 167 ms 293848 KB Output is correct
15 Correct 167 ms 294136 KB Output is correct
16 Correct 167 ms 293880 KB Output is correct
17 Correct 169 ms 294056 KB Output is correct
18 Correct 185 ms 294184 KB Output is correct
19 Correct 169 ms 294056 KB Output is correct
20 Correct 169 ms 294136 KB Output is correct
21 Correct 186 ms 294440 KB Output is correct
22 Correct 173 ms 294344 KB Output is correct
23 Correct 170 ms 294340 KB Output is correct
24 Correct 170 ms 294136 KB Output is correct
25 Correct 198 ms 295168 KB Output is correct
26 Correct 206 ms 295076 KB Output is correct
27 Correct 182 ms 295196 KB Output is correct
28 Correct 185 ms 295460 KB Output is correct
29 Correct 200 ms 296732 KB Output is correct
30 Correct 201 ms 296808 KB Output is correct
31 Correct 199 ms 296412 KB Output is correct
32 Correct 198 ms 296436 KB Output is correct
33 Correct 316 ms 315548 KB Output is correct
34 Correct 346 ms 316680 KB Output is correct
35 Correct 347 ms 316672 KB Output is correct
36 Correct 339 ms 321724 KB Output is correct
37 Correct 408 ms 306348 KB Output is correct
38 Correct 391 ms 306540 KB Output is correct
39 Correct 424 ms 306396 KB Output is correct
40 Correct 379 ms 305556 KB Output is correct
41 Correct 367 ms 309748 KB Output is correct
42 Correct 384 ms 314116 KB Output is correct
43 Correct 671 ms 326928 KB Output is correct
44 Correct 624 ms 327032 KB Output is correct
45 Correct 383 ms 310820 KB Output is correct
46 Correct 426 ms 310764 KB Output is correct
47 Correct 573 ms 323020 KB Output is correct
48 Correct 586 ms 323228 KB Output is correct
49 Correct 169 ms 294008 KB Output is correct
50 Correct 172 ms 293880 KB Output is correct
51 Correct 166 ms 293884 KB Output is correct
52 Correct 181 ms 294008 KB Output is correct
53 Correct 175 ms 293844 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 194 ms 294464 KB Output is correct
2 Correct 201 ms 294392 KB Output is correct
3 Correct 197 ms 293880 KB Output is correct
4 Correct 201 ms 294012 KB Output is correct
5 Correct 169 ms 294292 KB Output is correct
6 Correct 169 ms 294400 KB Output is correct
7 Correct 190 ms 294552 KB Output is correct
8 Correct 178 ms 294276 KB Output is correct
9 Correct 174 ms 294284 KB Output is correct
10 Correct 169 ms 294052 KB Output is correct
11 Correct 180 ms 294312 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 167 ms 293880 KB Output is correct
2 Correct 1189 ms 382924 KB Output is correct
3 Correct 2555 ms 485644 KB Output is correct
4 Correct 2521 ms 486536 KB Output is correct
5 Correct 2500 ms 486452 KB Output is correct
6 Correct 411 ms 319096 KB Output is correct
7 Correct 668 ms 340984 KB Output is correct
8 Correct 720 ms 343932 KB Output is correct
9 Correct 169 ms 294008 KB Output is correct
10 Correct 172 ms 293880 KB Output is correct
11 Correct 166 ms 293884 KB Output is correct
12 Correct 181 ms 294008 KB Output is correct
13 Correct 175 ms 293844 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 180 ms 293796 KB Output is correct
2 Correct 167 ms 293848 KB Output is correct
3 Correct 166 ms 293880 KB Output is correct
4 Correct 167 ms 293864 KB Output is correct
5 Correct 165 ms 293816 KB Output is correct
6 Correct 167 ms 293936 KB Output is correct
7 Correct 167 ms 293880 KB Output is correct
8 Correct 167 ms 293880 KB Output is correct
9 Correct 183 ms 293880 KB Output is correct
10 Correct 168 ms 293880 KB Output is correct
11 Correct 168 ms 293880 KB Output is correct
12 Correct 167 ms 293880 KB Output is correct
13 Correct 165 ms 293880 KB Output is correct
14 Correct 167 ms 293848 KB Output is correct
15 Correct 167 ms 294136 KB Output is correct
16 Correct 167 ms 293880 KB Output is correct
17 Correct 169 ms 294056 KB Output is correct
18 Correct 185 ms 294184 KB Output is correct
19 Correct 169 ms 294056 KB Output is correct
20 Correct 169 ms 294136 KB Output is correct
21 Correct 186 ms 294440 KB Output is correct
22 Correct 173 ms 294344 KB Output is correct
23 Correct 170 ms 294340 KB Output is correct
24 Correct 170 ms 294136 KB Output is correct
25 Correct 198 ms 295168 KB Output is correct
26 Correct 206 ms 295076 KB Output is correct
27 Correct 182 ms 295196 KB Output is correct
28 Correct 185 ms 295460 KB Output is correct
29 Correct 200 ms 296732 KB Output is correct
30 Correct 201 ms 296808 KB Output is correct
31 Correct 199 ms 296412 KB Output is correct
32 Correct 198 ms 296436 KB Output is correct
33 Correct 316 ms 315548 KB Output is correct
34 Correct 346 ms 316680 KB Output is correct
35 Correct 347 ms 316672 KB Output is correct
36 Correct 339 ms 321724 KB Output is correct
37 Correct 408 ms 306348 KB Output is correct
38 Correct 391 ms 306540 KB Output is correct
39 Correct 424 ms 306396 KB Output is correct
40 Correct 379 ms 305556 KB Output is correct
41 Correct 367 ms 309748 KB Output is correct
42 Correct 384 ms 314116 KB Output is correct
43 Correct 671 ms 326928 KB Output is correct
44 Correct 624 ms 327032 KB Output is correct
45 Correct 383 ms 310820 KB Output is correct
46 Correct 426 ms 310764 KB Output is correct
47 Correct 573 ms 323020 KB Output is correct
48 Correct 586 ms 323228 KB Output is correct
49 Correct 194 ms 294464 KB Output is correct
50 Correct 201 ms 294392 KB Output is correct
51 Correct 197 ms 293880 KB Output is correct
52 Correct 201 ms 294012 KB Output is correct
53 Correct 169 ms 294292 KB Output is correct
54 Correct 169 ms 294400 KB Output is correct
55 Correct 190 ms 294552 KB Output is correct
56 Correct 178 ms 294276 KB Output is correct
57 Correct 174 ms 294284 KB Output is correct
58 Correct 169 ms 294052 KB Output is correct
59 Correct 180 ms 294312 KB Output is correct
60 Correct 167 ms 293880 KB Output is correct
61 Correct 1189 ms 382924 KB Output is correct
62 Correct 2555 ms 485644 KB Output is correct
63 Correct 2521 ms 486536 KB Output is correct
64 Correct 2500 ms 486452 KB Output is correct
65 Correct 411 ms 319096 KB Output is correct
66 Correct 668 ms 340984 KB Output is correct
67 Correct 720 ms 343932 KB Output is correct
68 Correct 2498 ms 476912 KB Output is correct
69 Correct 2509 ms 557196 KB Output is correct
70 Correct 3128 ms 556952 KB Output is correct
71 Correct 3016 ms 637356 KB Output is correct
72 Correct 3569 ms 418576 KB Output is correct
73 Correct 4452 ms 540588 KB Output is correct
74 Correct 4452 ms 555680 KB Output is correct
75 Execution timed out 5086 ms 704704 KB Time limit exceeded
76 Halted 0 ms 0 KB -