oj.uz

Submission #724447

# Submission time Handle Problem Language Result Execution time Memory
724447 2023-04-15T09:37:30 Z rt144 Bridges (APIO19_bridges) C++17
100 / 100
 1763 ms 11244 KB 
#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e4+5, maxm = 1e5+5;
int n, m, q, ans[maxm];
struct edge { int u, v, w; } es[maxm];
struct upd { int isu, t, x, y; };
struct DSU {
	int p[maxn], sz[maxn];
	stack<int> h;
	DSU() { iota(p, p+n+1, 0); fill(sz, sz+n+1, 1); }
	int get(int x) { while(p[x]!=x) x=p[x]; return x; }
	bool merge(int x, int y, bool rec) {
		int a = get(x), b = get(y);
		if (a==b) return false;
		if (sz[b]>sz[a]) swap(a, b);
		sz[a] += sz[b]; p[b] = a;
		if (rec) h.push(b);
		return true;
	}
	void rollback(int t) {
		while (h.size()>t) {
			int b = h.top(); h.pop();
			sz[p[b]] -= sz[b];
			p[b] = b;
		}
	}
};

int32_t main() {
	cin.tie(nullptr)->sync_with_stdio(false);
	cin >> n >> m; const int B = 1000;
	for (int i=1;i<=m;i++) cin >> es[i].u >> es[i].v >> es[i].w;
	int q; cin >> q; vector<upd> qs(q);
	for (int i=0;i<q;i++) cin >> qs[i].isu >> qs[i].x >> qs[i].y, qs[i].t=i;
	for (int b=0;b<q;b+=B) {
		int bsz = min(q-b, B);
		bitset<maxm> isch = 0ull;
		vector<upd> ch, qu;
		for (int i=b;i<b+bsz;i++) {
			if (qs[i].isu==1) isch[qs[i].x]=1, ch.push_back(qs[i]);
			else qu.push_back(qs[i]);
		}
		vector<edge> un; for (int i=1;i<=m;i++) if (!isch[i]) un.push_back(es[i]);
		sort(begin(un), end(un), [](edge a, edge b) { return a.w<b.w; });
		sort(begin(qu), end(qu), [](upd a, upd b) { return a.y>b.y; });
		vector<vector<int>> to_merge(bsz);
		for (int i=b;i<b+bsz;i++) {
			if (qs[i].isu==1) es[qs[i].x].w = qs[i].y;
			else for (auto [_,ct,bi,_1]:ch) if (es[bi].w>=qs[i].y) to_merge[i-b].push_back(bi);
		}
		DSU dsu;
		for (auto [_, t, s, cw] : qu) {
			while (!un.empty() && un.back().w>=cw) {
				auto [u, v, _] = un.back();
				dsu.merge(u, v, false); un.pop_back();
			}
			int pt = dsu.h.size();
			for (int j : to_merge[t-b])
				dsu.merge(es[j].u, es[j].v, true);
			ans[t] = dsu.sz[dsu.get(s)];
			dsu.rollback(pt);
		}
	}
	for (int i=0;i<q;i++) if (ans[i]) cout << ans[i] << '\n';
}

Compilation message

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

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 724 KB Output is correct
2 Correct 1 ms 724 KB Output is correct
3 Correct 34 ms 1688 KB Output is correct
4 Correct 5 ms 1108 KB Output is correct
5 Correct 34 ms 1920 KB Output is correct
6 Correct 25 ms 2132 KB Output is correct
7 Correct 29 ms 2224 KB Output is correct
8 Correct 35 ms 1756 KB Output is correct
9 Correct 36 ms 2940 KB Output is correct
10 Correct 36 ms 1764 KB Output is correct
11 Correct 36 ms 1776 KB Output is correct
12 Correct 43 ms 2060 KB Output is correct
13 Correct 44 ms 1936 KB Output is correct
14 Correct 38 ms 1816 KB Output is correct
15 Correct 43 ms 1852 KB Output is correct
16 Correct 34 ms 2388 KB Output is correct
17 Correct 35 ms 2168 KB Output is correct

Subtask #2
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 1115 ms 5400 KB Output is correct
2 Correct 1126 ms 8328 KB Output is correct
3 Correct 1127 ms 8408 KB Output is correct
4 Correct 1179 ms 8132 KB Output is correct
5 Correct 1134 ms 8452 KB Output is correct
6 Correct 1693 ms 8348 KB Output is correct
7 Correct 1734 ms 8640 KB Output is correct
8 Correct 1736 ms 8552 KB Output is correct
9 Correct 31 ms 4172 KB Output is correct
10 Correct 998 ms 7372 KB Output is correct
11 Correct 917 ms 7084 KB Output is correct
12 Correct 907 ms 7340 KB Output is correct
13 Correct 943 ms 7528 KB Output is correct

Subtask #3
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 911 ms 4588 KB Output is correct
2 Correct 728 ms 6048 KB Output is correct
3 Correct 1088 ms 7620 KB Output is correct
4 Correct 896 ms 6900 KB Output is correct
5 Correct 31 ms 4292 KB Output is correct
6 Correct 1028 ms 7196 KB Output is correct
7 Correct 821 ms 6748 KB Output is correct
8 Correct 707 ms 6984 KB Output is correct
9 Correct 568 ms 6584 KB Output is correct
10 Correct 515 ms 6808 KB Output is correct
11 Correct 593 ms 6248 KB Output is correct
12 Correct 520 ms 6448 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1125 ms 6632 KB Output is correct
2 Correct 35 ms 2892 KB Output is correct
3 Correct 127 ms 4836 KB Output is correct
4 Correct 73 ms 4836 KB Output is correct
5 Correct 644 ms 6648 KB Output is correct
6 Correct 1133 ms 6656 KB Output is correct
7 Correct 653 ms 6692 KB Output is correct
8 Correct 542 ms 4660 KB Output is correct
9 Correct 551 ms 4692 KB Output is correct
10 Correct 545 ms 4712 KB Output is correct
11 Correct 905 ms 6144 KB Output is correct
12 Correct 856 ms 6056 KB Output is correct
13 Correct 888 ms 6220 KB Output is correct
14 Correct 607 ms 6696 KB Output is correct
15 Correct 641 ms 6628 KB Output is correct
16 Correct 1147 ms 6640 KB Output is correct
17 Correct 1175 ms 6924 KB Output is correct
18 Correct 1136 ms 6824 KB Output is correct
19 Correct 1152 ms 6832 KB Output is correct
20 Correct 988 ms 6588 KB Output is correct
21 Correct 986 ms 6556 KB Output is correct
22 Correct 1098 ms 6592 KB Output is correct
23 Correct 669 ms 6220 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1115 ms 5400 KB Output is correct
2 Correct 1126 ms 8328 KB Output is correct
3 Correct 1127 ms 8408 KB Output is correct
4 Correct 1179 ms 8132 KB Output is correct
5 Correct 1134 ms 8452 KB Output is correct
6 Correct 1693 ms 8348 KB Output is correct
7 Correct 1734 ms 8640 KB Output is correct
8 Correct 1736 ms 8552 KB Output is correct
9 Correct 31 ms 4172 KB Output is correct
10 Correct 998 ms 7372 KB Output is correct
11 Correct 917 ms 7084 KB Output is correct
12 Correct 907 ms 7340 KB Output is correct
13 Correct 943 ms 7528 KB Output is correct
14 Correct 911 ms 4588 KB Output is correct
15 Correct 728 ms 6048 KB Output is correct
16 Correct 1088 ms 7620 KB Output is correct
17 Correct 896 ms 6900 KB Output is correct
18 Correct 31 ms 4292 KB Output is correct
19 Correct 1028 ms 7196 KB Output is correct
20 Correct 821 ms 6748 KB Output is correct
21 Correct 707 ms 6984 KB Output is correct
22 Correct 568 ms 6584 KB Output is correct
23 Correct 515 ms 6808 KB Output is correct
24 Correct 593 ms 6248 KB Output is correct
25 Correct 520 ms 6448 KB Output is correct
26 Correct 1125 ms 8320 KB Output is correct
27 Correct 1412 ms 8720 KB Output is correct
28 Correct 1162 ms 8148 KB Output is correct
29 Correct 800 ms 7704 KB Output is correct
30 Correct 1340 ms 8360 KB Output is correct
31 Correct 1367 ms 8588 KB Output is correct
32 Correct 1370 ms 8448 KB Output is correct
33 Correct 1166 ms 8172 KB Output is correct
34 Correct 1183 ms 8084 KB Output is correct
35 Correct 1148 ms 8432 KB Output is correct
36 Correct 858 ms 7736 KB Output is correct
37 Correct 835 ms 7624 KB Output is correct
38 Correct 840 ms 7768 KB Output is correct
39 Correct 665 ms 7596 KB Output is correct
40 Correct 664 ms 7516 KB Output is correct
41 Correct 661 ms 7692 KB Output is correct
42 Correct 654 ms 7480 KB Output is correct
43 Correct 645 ms 7512 KB Output is correct
44 Correct 646 ms 7676 KB Output is correct

Subtask #6
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 724 KB Output is correct
2 Correct 1 ms 724 KB Output is correct
3 Correct 34 ms 1688 KB Output is correct
4 Correct 5 ms 1108 KB Output is correct
5 Correct 34 ms 1920 KB Output is correct
6 Correct 25 ms 2132 KB Output is correct
7 Correct 29 ms 2224 KB Output is correct
8 Correct 35 ms 1756 KB Output is correct
9 Correct 36 ms 2940 KB Output is correct
10 Correct 36 ms 1764 KB Output is correct
11 Correct 36 ms 1776 KB Output is correct
12 Correct 43 ms 2060 KB Output is correct
13 Correct 44 ms 1936 KB Output is correct
14 Correct 38 ms 1816 KB Output is correct
15 Correct 43 ms 1852 KB Output is correct
16 Correct 34 ms 2388 KB Output is correct
17 Correct 35 ms 2168 KB Output is correct
18 Correct 1115 ms 5400 KB Output is correct
19 Correct 1126 ms 8328 KB Output is correct
20 Correct 1127 ms 8408 KB Output is correct
21 Correct 1179 ms 8132 KB Output is correct
22 Correct 1134 ms 8452 KB Output is correct
23 Correct 1693 ms 8348 KB Output is correct
24 Correct 1734 ms 8640 KB Output is correct
25 Correct 1736 ms 8552 KB Output is correct
26 Correct 31 ms 4172 KB Output is correct
27 Correct 998 ms 7372 KB Output is correct
28 Correct 917 ms 7084 KB Output is correct
29 Correct 907 ms 7340 KB Output is correct
30 Correct 943 ms 7528 KB Output is correct
31 Correct 911 ms 4588 KB Output is correct
32 Correct 728 ms 6048 KB Output is correct
33 Correct 1088 ms 7620 KB Output is correct
34 Correct 896 ms 6900 KB Output is correct
35 Correct 31 ms 4292 KB Output is correct
36 Correct 1028 ms 7196 KB Output is correct
37 Correct 821 ms 6748 KB Output is correct
38 Correct 707 ms 6984 KB Output is correct
39 Correct 568 ms 6584 KB Output is correct
40 Correct 515 ms 6808 KB Output is correct
41 Correct 593 ms 6248 KB Output is correct
42 Correct 520 ms 6448 KB Output is correct
43 Correct 1125 ms 6632 KB Output is correct
44 Correct 35 ms 2892 KB Output is correct
45 Correct 127 ms 4836 KB Output is correct
46 Correct 73 ms 4836 KB Output is correct
47 Correct 644 ms 6648 KB Output is correct
48 Correct 1133 ms 6656 KB Output is correct
49 Correct 653 ms 6692 KB Output is correct
50 Correct 542 ms 4660 KB Output is correct
51 Correct 551 ms 4692 KB Output is correct
52 Correct 545 ms 4712 KB Output is correct
53 Correct 905 ms 6144 KB Output is correct
54 Correct 856 ms 6056 KB Output is correct
55 Correct 888 ms 6220 KB Output is correct
56 Correct 607 ms 6696 KB Output is correct
57 Correct 641 ms 6628 KB Output is correct
58 Correct 1147 ms 6640 KB Output is correct
59 Correct 1175 ms 6924 KB Output is correct
60 Correct 1136 ms 6824 KB Output is correct
61 Correct 1152 ms 6832 KB Output is correct
62 Correct 988 ms 6588 KB Output is correct
63 Correct 986 ms 6556 KB Output is correct
64 Correct 1098 ms 6592 KB Output is correct
65 Correct 669 ms 6220 KB Output is correct
66 Correct 1125 ms 8320 KB Output is correct
67 Correct 1412 ms 8720 KB Output is correct
68 Correct 1162 ms 8148 KB Output is correct
69 Correct 800 ms 7704 KB Output is correct
70 Correct 1340 ms 8360 KB Output is correct
71 Correct 1367 ms 8588 KB Output is correct
72 Correct 1370 ms 8448 KB Output is correct
73 Correct 1166 ms 8172 KB Output is correct
74 Correct 1183 ms 8084 KB Output is correct
75 Correct 1148 ms 8432 KB Output is correct
76 Correct 858 ms 7736 KB Output is correct
77 Correct 835 ms 7624 KB Output is correct
78 Correct 840 ms 7768 KB Output is correct
79 Correct 665 ms 7596 KB Output is correct
80 Correct 664 ms 7516 KB Output is correct
81 Correct 661 ms 7692 KB Output is correct
82 Correct 654 ms 7480 KB Output is correct
83 Correct 645 ms 7512 KB Output is correct
84 Correct 646 ms 7676 KB Output is correct
85 Correct 1536 ms 11240 KB Output is correct
86 Correct 156 ms 7268 KB Output is correct
87 Correct 104 ms 7188 KB Output is correct
88 Correct 930 ms 10184 KB Output is correct
89 Correct 1612 ms 11168 KB Output is correct
90 Correct 951 ms 10660 KB Output is correct
91 Correct 1216 ms 8156 KB Output is correct
92 Correct 1191 ms 8268 KB Output is correct
93 Correct 1718 ms 8156 KB Output is correct
94 Correct 1354 ms 10028 KB Output is correct
95 Correct 1362 ms 10072 KB Output is correct
96 Correct 1542 ms 9444 KB Output is correct
97 Correct 766 ms 9296 KB Output is correct
98 Correct 746 ms 8976 KB Output is correct
99 Correct 1530 ms 11244 KB Output is correct
100 Correct 1593 ms 11004 KB Output is correct
101 Correct 1604 ms 11044 KB Output is correct
102 Correct 1590 ms 11140 KB Output is correct
103 Correct 1763 ms 9816 KB Output is correct
104 Correct 1706 ms 9792 KB Output is correct
105 Correct 1316 ms 9632 KB Output is correct
106 Correct 1039 ms 9164 KB Output is correct
107 Correct 1074 ms 10140 KB Output is correct
108 Correct 1570 ms 10200 KB Output is correct
109 Correct 1138 ms 8280 KB Output is correct