oj.uz

Submission #363969

# Submission time Handle Problem Language Result Execution time Memory
363969 2021-02-07T17:35:55 Z penguinhacker Bridges (APIO19_bridges) C++14
100 / 100
 2934 ms 11052 KB 
#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define ar array

const int mxN = 100000;
const int B = 750;
int n, m, q, ans[mxN];
int par[mxN], sz[mxN];

ar<int, 3> e[mxN];
ar<int, 3> qry[mxN];

vector<int> rb;

void reset() {
	rb.clear();
	fill(sz, sz + n, 1);
	iota(par, par + n, 0);
}

int find(int i) {
	while(i != par[i]) i = par[i];
	return i;
}

void merge(int a, int b) {
	a = find(a), b = find(b);
	if (a == b) return;
	if (sz[a] < sz[b]) swap(a, b);
	rb.push_back(b);
	sz[a] += sz[b];
	par[b] = a;
}

void rollback(int k) {
	while(rb.size() > k) {
		int b = rb.back();
		rb.pop_back();
		sz[par[b]] -= sz[b];
		par[b] = b;
	}
}

vector<int> to_join[B];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin >> n >> m;
	for (int i = 0; i < m; ++i) cin >> e[i][0] >> e[i][1] >> e[i][2], --e[i][0], --e[i][1];
	cin >> q;
	for (int i = 0; i < q; ++i) cin >> qry[i][0] >> qry[i][1] >> qry[i][2], --qry[i][1];

	for (int l = 0; l < q; l += B) {
		reset();
		int r = min(q, l + B);
		vector<bool> changed(m);
		for (int i = l; i < r; ++i) if (qry[i][0] == 1) changed[qry[i][1]] = 1;

		vector<int> upd, unchanged, ask;
		for (int i = 0; i < m; ++i) {
			if (changed[i]) upd.push_back(i);
			if (!changed[i]) unchanged.push_back(i);
		}
		for (int i = l; i < r; ++i) {
			if (qry[i][0] == 1) e[qry[i][1]][2] = qry[i][2];
			else {
				to_join[i - l].clear();
				for (int ind : upd) if (e[ind][2] >= qry[i][2]) to_join[i - l].push_back(ind);
				ask.push_back(i);
			}
		}
		sort(unchanged.begin(), unchanged.end(), [&](int a, int b) {return e[a][2] < e[b][2];});
		sort(ask.begin(), ask.end(), [&](int a, int b) {return qry[a][2] > qry[b][2];});
		for (int i : ask) {
			while(!unchanged.empty()) {
				int x = unchanged.back();
				if (e[x][2] < qry[i][2]) break;
				merge(e[x][0], e[x][1]);
				unchanged.pop_back();
			}
			int prv_sz = rb.size();
			for (int x : to_join[i - l]) merge(e[x][0], e[x][1]);
			ans[i] = sz[find(qry[i][1])];
			rollback(prv_sz);
		}
	}
	for (int i = 0; i < q; ++i) if (qry[i][0] == 2) cout << ans[i] << "\n";
	return 0;
}

Compilation message

bridges.cpp: In function 'void rollback(int)':
bridges.cpp:38:18: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   38 |  while(rb.size() > k) {
      |        ~~~~~~~~~~^~~

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 30 ms 2028 KB Output is correct
4 Correct 4 ms 620 KB Output is correct
5 Correct 13 ms 1004 KB Output is correct
6 Correct 10 ms 876 KB Output is correct
7 Correct 13 ms 1004 KB Output is correct
8 Correct 14 ms 1004 KB Output is correct
9 Correct 17 ms 1004 KB Output is correct
10 Correct 14 ms 1004 KB Output is correct
11 Correct 14 ms 1004 KB Output is correct
12 Correct 17 ms 1004 KB Output is correct
13 Correct 20 ms 1260 KB Output is correct
14 Correct 18 ms 1132 KB Output is correct
15 Correct 16 ms 1004 KB Output is correct
16 Correct 15 ms 1004 KB Output is correct
17 Correct 15 ms 1004 KB Output is correct

Subtask #2
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 1703 ms 5448 KB Output is correct
2 Correct 1661 ms 5496 KB Output is correct
3 Correct 1683 ms 5476 KB Output is correct
4 Correct 1665 ms 5776 KB Output is correct
5 Correct 1697 ms 5464 KB Output is correct
6 Correct 2259 ms 5496 KB Output is correct
7 Correct 2250 ms 5200 KB Output is correct
8 Correct 2291 ms 5284 KB Output is correct
9 Correct 40 ms 2156 KB Output is correct
10 Correct 1235 ms 5452 KB Output is correct
11 Correct 1183 ms 5280 KB Output is correct
12 Correct 1577 ms 4348 KB Output is correct
13 Correct 1501 ms 6640 KB Output is correct

Subtask #3
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1266 ms 4720 KB Output is correct
2 Correct 749 ms 4076 KB Output is correct
3 Correct 1401 ms 4852 KB Output is correct
4 Correct 1231 ms 4628 KB Output is correct
5 Correct 39 ms 2156 KB Output is correct
6 Correct 1350 ms 4732 KB Output is correct
7 Correct 1179 ms 4804 KB Output is correct
8 Correct 1012 ms 4688 KB Output is correct
9 Correct 915 ms 3596 KB Output is correct
10 Correct 872 ms 3952 KB Output is correct
11 Correct 920 ms 6560 KB Output is correct
12 Correct 827 ms 6224 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 2380 ms 5420 KB Output is correct
2 Correct 41 ms 3564 KB Output is correct
3 Correct 247 ms 4552 KB Output is correct
4 Correct 108 ms 4668 KB Output is correct
5 Correct 1202 ms 7616 KB Output is correct
6 Correct 2323 ms 8932 KB Output is correct
7 Correct 1237 ms 7388 KB Output is correct
8 Correct 1197 ms 6540 KB Output is correct
9 Correct 1190 ms 6632 KB Output is correct
10 Correct 1196 ms 6372 KB Output is correct
11 Correct 1784 ms 7876 KB Output is correct
12 Correct 1763 ms 8260 KB Output is correct
13 Correct 1806 ms 7916 KB Output is correct
14 Correct 1050 ms 7652 KB Output is correct
15 Correct 1116 ms 7524 KB Output is correct
16 Correct 2361 ms 8788 KB Output is correct
17 Correct 2400 ms 8860 KB Output is correct
18 Correct 2447 ms 8916 KB Output is correct
19 Correct 2415 ms 9132 KB Output is correct
20 Correct 2036 ms 8536 KB Output is correct
21 Correct 2007 ms 8096 KB Output is correct
22 Correct 2307 ms 8804 KB Output is correct
23 Correct 1352 ms 6968 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1703 ms 5448 KB Output is correct
2 Correct 1661 ms 5496 KB Output is correct
3 Correct 1683 ms 5476 KB Output is correct
4 Correct 1665 ms 5776 KB Output is correct
5 Correct 1697 ms 5464 KB Output is correct
6 Correct 2259 ms 5496 KB Output is correct
7 Correct 2250 ms 5200 KB Output is correct
8 Correct 2291 ms 5284 KB Output is correct
9 Correct 40 ms 2156 KB Output is correct
10 Correct 1235 ms 5452 KB Output is correct
11 Correct 1183 ms 5280 KB Output is correct
12 Correct 1577 ms 4348 KB Output is correct
13 Correct 1501 ms 6640 KB Output is correct
14 Correct 1266 ms 4720 KB Output is correct
15 Correct 749 ms 4076 KB Output is correct
16 Correct 1401 ms 4852 KB Output is correct
17 Correct 1231 ms 4628 KB Output is correct
18 Correct 39 ms 2156 KB Output is correct
19 Correct 1350 ms 4732 KB Output is correct
20 Correct 1179 ms 4804 KB Output is correct
21 Correct 1012 ms 4688 KB Output is correct
22 Correct 915 ms 3596 KB Output is correct
23 Correct 872 ms 3952 KB Output is correct
24 Correct 920 ms 6560 KB Output is correct
25 Correct 827 ms 6224 KB Output is correct
26 Correct 1698 ms 5280 KB Output is correct
27 Correct 2025 ms 5448 KB Output is correct
28 Correct 1800 ms 5460 KB Output is correct
29 Correct 1478 ms 5504 KB Output is correct
30 Correct 2020 ms 5476 KB Output is correct
31 Correct 2031 ms 5336 KB Output is correct
32 Correct 2059 ms 5248 KB Output is correct
33 Correct 1837 ms 5332 KB Output is correct
34 Correct 1804 ms 5460 KB Output is correct
35 Correct 1785 ms 5524 KB Output is correct
36 Correct 1506 ms 5432 KB Output is correct
37 Correct 1529 ms 5384 KB Output is correct
38 Correct 1478 ms 5440 KB Output is correct
39 Correct 1329 ms 4196 KB Output is correct
40 Correct 1309 ms 4380 KB Output is correct
41 Correct 1335 ms 4148 KB Output is correct
42 Correct 1251 ms 6440 KB Output is correct
43 Correct 1292 ms 6612 KB Output is correct
44 Correct 1234 ms 6276 KB Output is correct

Subtask #6
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 30 ms 2028 KB Output is correct
4 Correct 4 ms 620 KB Output is correct
5 Correct 13 ms 1004 KB Output is correct
6 Correct 10 ms 876 KB Output is correct
7 Correct 13 ms 1004 KB Output is correct
8 Correct 14 ms 1004 KB Output is correct
9 Correct 17 ms 1004 KB Output is correct
10 Correct 14 ms 1004 KB Output is correct
11 Correct 14 ms 1004 KB Output is correct
12 Correct 17 ms 1004 KB Output is correct
13 Correct 20 ms 1260 KB Output is correct
14 Correct 18 ms 1132 KB Output is correct
15 Correct 16 ms 1004 KB Output is correct
16 Correct 15 ms 1004 KB Output is correct
17 Correct 15 ms 1004 KB Output is correct
18 Correct 1703 ms 5448 KB Output is correct
19 Correct 1661 ms 5496 KB Output is correct
20 Correct 1683 ms 5476 KB Output is correct
21 Correct 1665 ms 5776 KB Output is correct
22 Correct 1697 ms 5464 KB Output is correct
23 Correct 2259 ms 5496 KB Output is correct
24 Correct 2250 ms 5200 KB Output is correct
25 Correct 2291 ms 5284 KB Output is correct
26 Correct 40 ms 2156 KB Output is correct
27 Correct 1235 ms 5452 KB Output is correct
28 Correct 1183 ms 5280 KB Output is correct
29 Correct 1577 ms 4348 KB Output is correct
30 Correct 1501 ms 6640 KB Output is correct
31 Correct 1266 ms 4720 KB Output is correct
32 Correct 749 ms 4076 KB Output is correct
33 Correct 1401 ms 4852 KB Output is correct
34 Correct 1231 ms 4628 KB Output is correct
35 Correct 39 ms 2156 KB Output is correct
36 Correct 1350 ms 4732 KB Output is correct
37 Correct 1179 ms 4804 KB Output is correct
38 Correct 1012 ms 4688 KB Output is correct
39 Correct 915 ms 3596 KB Output is correct
40 Correct 872 ms 3952 KB Output is correct
41 Correct 920 ms 6560 KB Output is correct
42 Correct 827 ms 6224 KB Output is correct
43 Correct 2380 ms 5420 KB Output is correct
44 Correct 41 ms 3564 KB Output is correct
45 Correct 247 ms 4552 KB Output is correct
46 Correct 108 ms 4668 KB Output is correct
47 Correct 1202 ms 7616 KB Output is correct
48 Correct 2323 ms 8932 KB Output is correct
49 Correct 1237 ms 7388 KB Output is correct
50 Correct 1197 ms 6540 KB Output is correct
51 Correct 1190 ms 6632 KB Output is correct
52 Correct 1196 ms 6372 KB Output is correct
53 Correct 1784 ms 7876 KB Output is correct
54 Correct 1763 ms 8260 KB Output is correct
55 Correct 1806 ms 7916 KB Output is correct
56 Correct 1050 ms 7652 KB Output is correct
57 Correct 1116 ms 7524 KB Output is correct
58 Correct 2361 ms 8788 KB Output is correct
59 Correct 2400 ms 8860 KB Output is correct
60 Correct 2447 ms 8916 KB Output is correct
61 Correct 2415 ms 9132 KB Output is correct
62 Correct 2036 ms 8536 KB Output is correct
63 Correct 2007 ms 8096 KB Output is correct
64 Correct 2307 ms 8804 KB Output is correct
65 Correct 1352 ms 6968 KB Output is correct
66 Correct 1698 ms 5280 KB Output is correct
67 Correct 2025 ms 5448 KB Output is correct
68 Correct 1800 ms 5460 KB Output is correct
69 Correct 1478 ms 5504 KB Output is correct
70 Correct 2020 ms 5476 KB Output is correct
71 Correct 2031 ms 5336 KB Output is correct
72 Correct 2059 ms 5248 KB Output is correct
73 Correct 1837 ms 5332 KB Output is correct
74 Correct 1804 ms 5460 KB Output is correct
75 Correct 1785 ms 5524 KB Output is correct
76 Correct 1506 ms 5432 KB Output is correct
77 Correct 1529 ms 5384 KB Output is correct
78 Correct 1478 ms 5440 KB Output is correct
79 Correct 1329 ms 4196 KB Output is correct
80 Correct 1309 ms 4380 KB Output is correct
81 Correct 1335 ms 4148 KB Output is correct
82 Correct 1251 ms 6440 KB Output is correct
83 Correct 1292 ms 6612 KB Output is correct
84 Correct 1234 ms 6276 KB Output is correct
85 Correct 2857 ms 10396 KB Output is correct
86 Correct 279 ms 6088 KB Output is correct
87 Correct 155 ms 6676 KB Output is correct
88 Correct 1635 ms 8812 KB Output is correct
89 Correct 2813 ms 10660 KB Output is correct
90 Correct 1674 ms 9048 KB Output is correct
91 Correct 1819 ms 8464 KB Output is correct
92 Correct 1814 ms 8544 KB Output is correct
93 Correct 2361 ms 8216 KB Output is correct
94 Correct 2309 ms 9416 KB Output is correct
95 Correct 2310 ms 9300 KB Output is correct
96 Correct 2602 ms 8940 KB Output is correct
97 Correct 1426 ms 7608 KB Output is correct
98 Correct 1446 ms 10368 KB Output is correct
99 Correct 2837 ms 10148 KB Output is correct
100 Correct 2848 ms 10200 KB Output is correct
101 Correct 2934 ms 10484 KB Output is correct
102 Correct 2887 ms 10936 KB Output is correct
103 Correct 2892 ms 9568 KB Output is correct
104 Correct 2854 ms 9552 KB Output is correct
105 Correct 2322 ms 11052 KB Output is correct
106 Correct 1940 ms 10648 KB Output is correct
107 Correct 2268 ms 8212 KB Output is correct
108 Correct 2841 ms 9880 KB Output is correct
109 Correct 1968 ms 8196 KB Output is correct