oj.uz

답안 #792867

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
792867 2023-07-25T10:09:50 Z phoenix0423 다리 (APIO19_bridges) C++17
100 / 100
 1862 ms 156288 KB 
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define fastio ios::sync_with_stdio(false), cin.tie(0)
#pragma GCC optimize("Ofast")
#define pb push_back
const int maxn = 1e5 + 5;
ll u[maxn], v[maxn], w[maxn];
ll t[maxn], x[maxn], y[maxn], ch[maxn]; // [bridge, new weight] or [start, car weight]
ll ans[maxn];
vector<int> to_join[maxn];
int n, m, q;
const int b = 1000;

struct dsu_save{
	int u, rku, v, rkv;
	dsu_save(){}
	dsu_save(int _u, int _rku, int _v, int _rkv) : u(_u), rku(_rku), v(_v), rkv(_rkv){}
};
struct DSU{
	vector<int> par, siz;
	int n;
	stack<int> stk;
	DSU(){}
	DSU(int _n) : n(_n){}
	void init(){
		siz = vector<int>(n, 1);
		par.resize(n);
		for(int i = 0; i < n; i++) par[i] = i;
	}
	int root(int x){
		return x == par[x] ? x : root(par[x]);
	}
	void unite(int x, int y){
		x = root(x);
		y = root(y);
		if(x == y) return;
		if(siz[x] > siz[y]) swap(x, y);
		stk.push(x);
		par[x] = y;
		siz[y] += siz[x];
	}
	void roll_back(int sz){
		while(stk.size() > sz){
			auto u = stk.top(); stk.pop();
			siz[par[u]] -= siz[u];
			par[u] = u;
		}
	}
};
int main(void){

	fastio;
	cin>>n>>m;
	DSU dsu(n + 1);
	for(int i = 1; i <= m; i++){
		cin >> u[i] >> v[i] >> w[i];
	}
	cin>>q;
	for(int i = 1; i <= q; i++){
		cin >> t[i] >> x[i] >> y[i];
	}
	for(int l = 1; l <= q; l += b){
		int r = min(q + 1, l + b);
		fill(ch, ch + maxn, false);
		dsu.init();
		vector<int> ask, upd, unchanged;

		for(int i = l; i < r; i++){
			if(t[i] == 1){
				ch[x[i]] = true;
				upd.pb(i);
			}
			else ask.pb(i);
		}
		for(int i = 1; i <= m; i++){
			if(!ch[i]) unchanged.pb(i);
		}
		for(int i = l; i < r; i++){
			if(t[i] == 1) w[x[i]] = y[i];
			else{
				to_join[i].clear();
				for(auto j : upd){
					if(w[x[j]] >= y[i]) to_join[i].pb(x[j]);
				}
			}
		}
		sort(ask.begin(), ask.end(), [](int a, int b){ return y[a] > y[b];});
		sort(unchanged.begin(), unchanged.end(), [](int a, int b){ return w[a] > w[b];});
		// processing queries
		int unptr = 0;
		for(auto i : ask){
			while(unptr < unchanged.size() && w[unchanged[unptr]] >= y[i]){ // no need to roll back
				dsu.unite(u[unchanged[unptr]], v[unchanged[unptr]]);
				unptr++;
			}
			int prev_sz = dsu.stk.size();
			for(auto j : to_join[i]){
				dsu.unite(u[j], v[j]);
			}
			ans[i] = dsu.siz[dsu.root(x[i])];
			dsu.roll_back(prev_sz);
		}
	}
	for(int i = 1; i <= q; i++) if(t[i] == 2) cout << ans[i] << "\n";
}

Compilation message

bridges.cpp: In member function 'void DSU::roll_back(int)':
bridges.cpp:44:20: warning: comparison of integer expressions of different signedness: 'std::stack<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   44 |   while(stk.size() > sz){
      |         ~~~~~~~~~~~^~~~
bridges.cpp: In function 'int main()':
bridges.cpp:93:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   93 |    while(unptr < unchanged.size() && w[unchanged[unptr]] >= y[i]){ // no need to roll back
      |          ~~~~~~^~~~~~~~~~~~~~~~~~

Subtask #1
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 3412 KB Output is correct
2 Correct 2 ms 3452 KB Output is correct
3 Correct 31 ms 10572 KB Output is correct
4 Correct 5 ms 3848 KB Output is correct
5 Correct 32 ms 11288 KB Output is correct
6 Correct 24 ms 11568 KB Output is correct
7 Correct 26 ms 11468 KB Output is correct
8 Correct 31 ms 9900 KB Output is correct
9 Correct 33 ms 15780 KB Output is correct
10 Correct 38 ms 10744 KB Output is correct
11 Correct 32 ms 10496 KB Output is correct
12 Correct 37 ms 13096 KB Output is correct
13 Correct 35 ms 11212 KB Output is correct
14 Correct 35 ms 10420 KB Output is correct
15 Correct 39 ms 11220 KB Output is correct
16 Correct 27 ms 14024 KB Output is correct
17 Correct 30 ms 13152 KB Output is correct

Subtask #2
16.0 / 16.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1051 ms 95956 KB Output is correct
2 Correct 1060 ms 98268 KB Output is correct
3 Correct 1046 ms 98644 KB Output is correct
4 Correct 1020 ms 98200 KB Output is correct
5 Correct 1022 ms 98848 KB Output is correct
6 Correct 1200 ms 156288 KB Output is correct
7 Correct 1195 ms 152768 KB Output is correct
8 Correct 1215 ms 152624 KB Output is correct
9 Correct 32 ms 8108 KB Output is correct
10 Correct 715 ms 123052 KB Output is correct
11 Correct 637 ms 122172 KB Output is correct
12 Correct 947 ms 78200 KB Output is correct
13 Correct 915 ms 71356 KB Output is correct

Subtask #3
17.0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 781 ms 88920 KB Output is correct
2 Correct 467 ms 104804 KB Output is correct
3 Correct 903 ms 118524 KB Output is correct
4 Correct 746 ms 90304 KB Output is correct
5 Correct 33 ms 8060 KB Output is correct
6 Correct 821 ms 109960 KB Output is correct
7 Correct 738 ms 82272 KB Output is correct
8 Correct 649 ms 66308 KB Output is correct
9 Correct 571 ms 58408 KB Output is correct
10 Correct 541 ms 46824 KB Output is correct
11 Correct 609 ms 55540 KB Output is correct
12 Correct 551 ms 45212 KB Output is correct

Subtask #4
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1542 ms 31388 KB Output is correct
2 Correct 30 ms 8140 KB Output is correct
3 Correct 179 ms 8852 KB Output is correct
4 Correct 79 ms 8984 KB Output is correct
5 Correct 807 ms 32956 KB Output is correct
6 Correct 1491 ms 34440 KB Output is correct
7 Correct 767 ms 32952 KB Output is correct
8 Correct 703 ms 32024 KB Output is correct
9 Correct 675 ms 32028 KB Output is correct
10 Correct 681 ms 31888 KB Output is correct
11 Correct 1113 ms 33408 KB Output is correct
12 Correct 1094 ms 33544 KB Output is correct
13 Correct 1165 ms 33464 KB Output is correct
14 Correct 706 ms 33044 KB Output is correct
15 Correct 767 ms 33060 KB Output is correct
16 Correct 1527 ms 34740 KB Output is correct
17 Correct 1541 ms 34636 KB Output is correct
18 Correct 1510 ms 34856 KB Output is correct
19 Correct 1484 ms 34800 KB Output is correct
20 Correct 1278 ms 33792 KB Output is correct
21 Correct 1262 ms 33848 KB Output is correct
22 Correct 1463 ms 34064 KB Output is correct
23 Correct 859 ms 30492 KB Output is correct

Subtask #5
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1051 ms 95956 KB Output is correct
2 Correct 1060 ms 98268 KB Output is correct
3 Correct 1046 ms 98644 KB Output is correct
4 Correct 1020 ms 98200 KB Output is correct
5 Correct 1022 ms 98848 KB Output is correct
6 Correct 1200 ms 156288 KB Output is correct
7 Correct 1195 ms 152768 KB Output is correct
8 Correct 1215 ms 152624 KB Output is correct
9 Correct 32 ms 8108 KB Output is correct
10 Correct 715 ms 123052 KB Output is correct
11 Correct 637 ms 122172 KB Output is correct
12 Correct 947 ms 78200 KB Output is correct
13 Correct 915 ms 71356 KB Output is correct
14 Correct 781 ms 88920 KB Output is correct
15 Correct 467 ms 104804 KB Output is correct
16 Correct 903 ms 118524 KB Output is correct
17 Correct 746 ms 90304 KB Output is correct
18 Correct 33 ms 8060 KB Output is correct
19 Correct 821 ms 109960 KB Output is correct
20 Correct 738 ms 82272 KB Output is correct
21 Correct 649 ms 66308 KB Output is correct
22 Correct 571 ms 58408 KB Output is correct
23 Correct 541 ms 46824 KB Output is correct
24 Correct 609 ms 55540 KB Output is correct
25 Correct 551 ms 45212 KB Output is correct
26 Correct 1043 ms 98528 KB Output is correct
27 Correct 1081 ms 127584 KB Output is correct
28 Correct 1010 ms 96692 KB Output is correct
29 Correct 832 ms 53688 KB Output is correct
30 Correct 1160 ms 114140 KB Output is correct
31 Correct 1119 ms 116516 KB Output is correct
32 Correct 1098 ms 116924 KB Output is correct
33 Correct 1050 ms 97020 KB Output is correct
34 Correct 1047 ms 97280 KB Output is correct
35 Correct 1050 ms 97528 KB Output is correct
36 Correct 859 ms 60604 KB Output is correct
37 Correct 880 ms 59376 KB Output is correct
38 Correct 916 ms 57836 KB Output is correct
39 Correct 808 ms 44060 KB Output is correct
40 Correct 764 ms 43180 KB Output is correct
41 Correct 804 ms 42720 KB Output is correct
42 Correct 768 ms 41480 KB Output is correct
43 Correct 734 ms 40864 KB Output is correct
44 Correct 720 ms 40272 KB Output is correct

Subtask #6
27.0 / 27.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 3412 KB Output is correct
2 Correct 2 ms 3452 KB Output is correct
3 Correct 31 ms 10572 KB Output is correct
4 Correct 5 ms 3848 KB Output is correct
5 Correct 32 ms 11288 KB Output is correct
6 Correct 24 ms 11568 KB Output is correct
7 Correct 26 ms 11468 KB Output is correct
8 Correct 31 ms 9900 KB Output is correct
9 Correct 33 ms 15780 KB Output is correct
10 Correct 38 ms 10744 KB Output is correct
11 Correct 32 ms 10496 KB Output is correct
12 Correct 37 ms 13096 KB Output is correct
13 Correct 35 ms 11212 KB Output is correct
14 Correct 35 ms 10420 KB Output is correct
15 Correct 39 ms 11220 KB Output is correct
16 Correct 27 ms 14024 KB Output is correct
17 Correct 30 ms 13152 KB Output is correct
18 Correct 1051 ms 95956 KB Output is correct
19 Correct 1060 ms 98268 KB Output is correct
20 Correct 1046 ms 98644 KB Output is correct
21 Correct 1020 ms 98200 KB Output is correct
22 Correct 1022 ms 98848 KB Output is correct
23 Correct 1200 ms 156288 KB Output is correct
24 Correct 1195 ms 152768 KB Output is correct
25 Correct 1215 ms 152624 KB Output is correct
26 Correct 32 ms 8108 KB Output is correct
27 Correct 715 ms 123052 KB Output is correct
28 Correct 637 ms 122172 KB Output is correct
29 Correct 947 ms 78200 KB Output is correct
30 Correct 915 ms 71356 KB Output is correct
31 Correct 781 ms 88920 KB Output is correct
32 Correct 467 ms 104804 KB Output is correct
33 Correct 903 ms 118524 KB Output is correct
34 Correct 746 ms 90304 KB Output is correct
35 Correct 33 ms 8060 KB Output is correct
36 Correct 821 ms 109960 KB Output is correct
37 Correct 738 ms 82272 KB Output is correct
38 Correct 649 ms 66308 KB Output is correct
39 Correct 571 ms 58408 KB Output is correct
40 Correct 541 ms 46824 KB Output is correct
41 Correct 609 ms 55540 KB Output is correct
42 Correct 551 ms 45212 KB Output is correct
43 Correct 1542 ms 31388 KB Output is correct
44 Correct 30 ms 8140 KB Output is correct
45 Correct 179 ms 8852 KB Output is correct
46 Correct 79 ms 8984 KB Output is correct
47 Correct 807 ms 32956 KB Output is correct
48 Correct 1491 ms 34440 KB Output is correct
49 Correct 767 ms 32952 KB Output is correct
50 Correct 703 ms 32024 KB Output is correct
51 Correct 675 ms 32028 KB Output is correct
52 Correct 681 ms 31888 KB Output is correct
53 Correct 1113 ms 33408 KB Output is correct
54 Correct 1094 ms 33544 KB Output is correct
55 Correct 1165 ms 33464 KB Output is correct
56 Correct 706 ms 33044 KB Output is correct
57 Correct 767 ms 33060 KB Output is correct
58 Correct 1527 ms 34740 KB Output is correct
59 Correct 1541 ms 34636 KB Output is correct
60 Correct 1510 ms 34856 KB Output is correct
61 Correct 1484 ms 34800 KB Output is correct
62 Correct 1278 ms 33792 KB Output is correct
63 Correct 1262 ms 33848 KB Output is correct
64 Correct 1463 ms 34064 KB Output is correct
65 Correct 859 ms 30492 KB Output is correct
66 Correct 1043 ms 98528 KB Output is correct
67 Correct 1081 ms 127584 KB Output is correct
68 Correct 1010 ms 96692 KB Output is correct
69 Correct 832 ms 53688 KB Output is correct
70 Correct 1160 ms 114140 KB Output is correct
71 Correct 1119 ms 116516 KB Output is correct
72 Correct 1098 ms 116924 KB Output is correct
73 Correct 1050 ms 97020 KB Output is correct
74 Correct 1047 ms 97280 KB Output is correct
75 Correct 1050 ms 97528 KB Output is correct
76 Correct 859 ms 60604 KB Output is correct
77 Correct 880 ms 59376 KB Output is correct
78 Correct 916 ms 57836 KB Output is correct
79 Correct 808 ms 44060 KB Output is correct
80 Correct 764 ms 43180 KB Output is correct
81 Correct 804 ms 42720 KB Output is correct
82 Correct 768 ms 41480 KB Output is correct
83 Correct 734 ms 40864 KB Output is correct
84 Correct 720 ms 40272 KB Output is correct
85 Correct 1798 ms 100920 KB Output is correct
86 Correct 180 ms 15032 KB Output is correct
87 Correct 111 ms 19560 KB Output is correct
88 Correct 1050 ms 112908 KB Output is correct
89 Correct 1788 ms 99608 KB Output is correct
90 Correct 1030 ms 121496 KB Output is correct
91 Correct 1095 ms 98420 KB Output is correct
92 Correct 1063 ms 98252 KB Output is correct
93 Correct 1238 ms 136040 KB Output is correct
94 Correct 1442 ms 99908 KB Output is correct
95 Correct 1437 ms 99872 KB Output is correct
96 Correct 1541 ms 140620 KB Output is correct
97 Correct 881 ms 65660 KB Output is correct
98 Correct 958 ms 62604 KB Output is correct
99 Correct 1828 ms 102112 KB Output is correct
100 Correct 1821 ms 100944 KB Output is correct
101 Correct 1862 ms 101180 KB Output is correct
102 Correct 1800 ms 101168 KB Output is correct
103 Correct 1659 ms 146084 KB Output is correct
104 Correct 1666 ms 145732 KB Output is correct
105 Correct 1398 ms 73424 KB Output is correct
106 Correct 1161 ms 53416 KB Output is correct
107 Correct 1387 ms 80036 KB Output is correct
108 Correct 1734 ms 135860 KB Output is correct
109 Correct 1136 ms 137652 KB Output is correct