oj.uz

Submission #748586

# Submission time Handle Problem Language Result Execution time Memory
748586 2023-05-26T14:05:26 Z NeroZein Bridges (APIO19_bridges) C++17
100 / 100
 1883 ms 12184 KB 
#include "bits/stdc++.h"
using namespace std;

#ifdef Nero
#include "Deb.h"
#else
#define deb(...)
#endif

const int B = 1003;
const int N = 50004;
const int Q = 100005;

stack<int> stk; 
int p[N], sz[N]; 

inline int get (int v) {
  while (p[v] != v) v = p[v];
  return v;
}

inline void unite (int u, int v) {
  u = get(u); v = get(v);
  if (u == v) return;
  if (sz[u] > sz[v]) swap(u, v); 
  sz[v] += sz[u];
  p[u] = v; 
  stk.push(u); 
}

inline void rollback (int time) {
  while (time < (int) stk.size()) {
    int u = stk.top();
    stk.pop(); 
    sz[p[u]] -= sz[u];
    p[u] = u; 
  }
}

int n, m;
int ans[Q]; 
bitset<Q> changed; 
int u[Q], v[Q], d[Q]; 
int t[Q], x[Q], y[Q]; 
vector<int> to_join[B];

void reset () {
  for (int i = 1; i <= n; ++i) {
    p[i] = i; sz[i] = 1; 
  }
  changed.reset(); 
  while (stk.size()) stk.pop(); 
}

signed main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  cin >> n >> m;
  for (int i = 1; i <= m; ++i) {
    cin >> u[i] >> v[i] >> d[i]; 
  }
  int q;
  cin >> q; 
  for (int i = 1; i <= q; ++i) {
    cin >> t[i] >> x[i] >> y[i]; 
  }
  for (int l = 1; l <= q; l += B) {
    reset(); 
    vector<int> ask, upd;
    for (int i = l; i < min(q + 1, l + B); ++i) {
      if (t[i] == 1) {
        changed[x[i]] = 1; 
        upd.push_back(i); 
      } else {
        ask.push_back(i);
      }
    }
    for (int i = l; i < min(q + 1, l + B); ++i) {
      if (t[i] == 1) {
        d[x[i]] = y[i]; 
      } else {
        to_join[i - l].clear(); 
        for (int j : upd) {
          if (d[x[j]] >= y[i]) {
            to_join[i - l].push_back(x[j]); 
          }
        }
      }
    }
    vector<int> unchanged; 
    for (int i = 1; i <= m; ++i) {
      if (!changed[i]) {
        unchanged.push_back(i); 
      }
    }
    sort(ask.begin(), ask.end(), [&](int i, int j) {
      return y[i] > y[j]; 
    }); 
    sort(unchanged.begin(), unchanged.end(), [&](int i, int j) {
      return d[i] > d[j]; 
    }); 
    int ptr = 0; 
    for (int i : ask) {
      while (ptr < (int) unchanged.size() && d[unchanged[ptr]] >= y[i]) {
        unite(u[unchanged[ptr]], v[unchanged[ptr]]); 
        ptr++; 
      }
      int time = (int) stk.size();
      for (int j : to_join[i - l]) {
        unite(u[j], v[j]); 
      }
      ans[i] = sz[get(x[i])]; 
      rollback(time); 
    }
  }
  for (int i = 1; i <= q; ++i) {
    if (t[i] == 2) {
      cout << ans[i] << ' '; 
    }
  }
  return 0;
}

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 352 KB Output is correct
3 Correct 29 ms 2968 KB Output is correct
4 Correct 4 ms 640 KB Output is correct
5 Correct 30 ms 3292 KB Output is correct
6 Correct 21 ms 2788 KB Output is correct
7 Correct 29 ms 3792 KB Output is correct
8 Correct 33 ms 2724 KB Output is correct
9 Correct 31 ms 4344 KB Output is correct
10 Correct 33 ms 3024 KB Output is correct
11 Correct 33 ms 2768 KB Output is correct
12 Correct 40 ms 2896 KB Output is correct
13 Correct 38 ms 2900 KB Output is correct
14 Correct 36 ms 2796 KB Output is correct
15 Correct 40 ms 2784 KB Output is correct
16 Correct 32 ms 3928 KB Output is correct
17 Correct 33 ms 3532 KB Output is correct

Subtask #2
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 1173 ms 9116 KB Output is correct
2 Correct 1196 ms 9180 KB Output is correct
3 Correct 1189 ms 9036 KB Output is correct
4 Correct 1239 ms 8920 KB Output is correct
5 Correct 1215 ms 9232 KB Output is correct
6 Correct 1724 ms 10464 KB Output is correct
7 Correct 1734 ms 10316 KB Output is correct
8 Correct 1738 ms 10440 KB Output is correct
9 Correct 33 ms 3432 KB Output is correct
10 Correct 955 ms 9396 KB Output is correct
11 Correct 879 ms 9208 KB Output is correct
12 Correct 998 ms 6852 KB Output is correct
13 Correct 1046 ms 10268 KB Output is correct

Subtask #3
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 990 ms 8332 KB Output is correct
2 Correct 678 ms 8408 KB Output is correct
3 Correct 1112 ms 9408 KB Output is correct
4 Correct 919 ms 8044 KB Output is correct
5 Correct 32 ms 3452 KB Output is correct
6 Correct 1065 ms 7828 KB Output is correct
7 Correct 874 ms 7812 KB Output is correct
8 Correct 770 ms 7816 KB Output is correct
9 Correct 632 ms 6048 KB Output is correct
10 Correct 569 ms 6092 KB Output is correct
11 Correct 719 ms 9368 KB Output is correct
12 Correct 622 ms 9388 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1487 ms 8732 KB Output is correct
2 Correct 41 ms 3404 KB Output is correct
3 Correct 155 ms 4460 KB Output is correct
4 Correct 69 ms 4572 KB Output is correct
5 Correct 676 ms 7332 KB Output is correct
6 Correct 1434 ms 8716 KB Output is correct
7 Correct 694 ms 7228 KB Output is correct
8 Correct 743 ms 6224 KB Output is correct
9 Correct 710 ms 6384 KB Output is correct
10 Correct 749 ms 6252 KB Output is correct
11 Correct 1117 ms 7708 KB Output is correct
12 Correct 1061 ms 7832 KB Output is correct
13 Correct 1069 ms 7616 KB Output is correct
14 Correct 668 ms 7384 KB Output is correct
15 Correct 636 ms 7420 KB Output is correct
16 Correct 1338 ms 8784 KB Output is correct
17 Correct 1341 ms 8908 KB Output is correct
18 Correct 1323 ms 8776 KB Output is correct
19 Correct 1362 ms 8776 KB Output is correct
20 Correct 1142 ms 8072 KB Output is correct
21 Correct 1133 ms 8016 KB Output is correct
22 Correct 1297 ms 8644 KB Output is correct
23 Correct 766 ms 6948 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1173 ms 9116 KB Output is correct
2 Correct 1196 ms 9180 KB Output is correct
3 Correct 1189 ms 9036 KB Output is correct
4 Correct 1239 ms 8920 KB Output is correct
5 Correct 1215 ms 9232 KB Output is correct
6 Correct 1724 ms 10464 KB Output is correct
7 Correct 1734 ms 10316 KB Output is correct
8 Correct 1738 ms 10440 KB Output is correct
9 Correct 33 ms 3432 KB Output is correct
10 Correct 955 ms 9396 KB Output is correct
11 Correct 879 ms 9208 KB Output is correct
12 Correct 998 ms 6852 KB Output is correct
13 Correct 1046 ms 10268 KB Output is correct
14 Correct 990 ms 8332 KB Output is correct
15 Correct 678 ms 8408 KB Output is correct
16 Correct 1112 ms 9408 KB Output is correct
17 Correct 919 ms 8044 KB Output is correct
18 Correct 32 ms 3452 KB Output is correct
19 Correct 1065 ms 7828 KB Output is correct
20 Correct 874 ms 7812 KB Output is correct
21 Correct 770 ms 7816 KB Output is correct
22 Correct 632 ms 6048 KB Output is correct
23 Correct 569 ms 6092 KB Output is correct
24 Correct 719 ms 9368 KB Output is correct
25 Correct 622 ms 9388 KB Output is correct
26 Correct 1274 ms 9120 KB Output is correct
27 Correct 1540 ms 10492 KB Output is correct
28 Correct 1272 ms 9276 KB Output is correct
29 Correct 891 ms 8920 KB Output is correct
30 Correct 1437 ms 10320 KB Output is correct
31 Correct 1464 ms 10324 KB Output is correct
32 Correct 1494 ms 10260 KB Output is correct
33 Correct 1254 ms 9104 KB Output is correct
34 Correct 1257 ms 9160 KB Output is correct
35 Correct 1255 ms 9004 KB Output is correct
36 Correct 961 ms 8704 KB Output is correct
37 Correct 942 ms 8832 KB Output is correct
38 Correct 935 ms 8552 KB Output is correct
39 Correct 771 ms 7248 KB Output is correct
40 Correct 775 ms 7076 KB Output is correct
41 Correct 768 ms 7228 KB Output is correct
42 Correct 778 ms 9920 KB Output is correct
43 Correct 770 ms 10036 KB Output is correct
44 Correct 760 ms 9696 KB Output is correct

Subtask #6
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 352 KB Output is correct
3 Correct 29 ms 2968 KB Output is correct
4 Correct 4 ms 640 KB Output is correct
5 Correct 30 ms 3292 KB Output is correct
6 Correct 21 ms 2788 KB Output is correct
7 Correct 29 ms 3792 KB Output is correct
8 Correct 33 ms 2724 KB Output is correct
9 Correct 31 ms 4344 KB Output is correct
10 Correct 33 ms 3024 KB Output is correct
11 Correct 33 ms 2768 KB Output is correct
12 Correct 40 ms 2896 KB Output is correct
13 Correct 38 ms 2900 KB Output is correct
14 Correct 36 ms 2796 KB Output is correct
15 Correct 40 ms 2784 KB Output is correct
16 Correct 32 ms 3928 KB Output is correct
17 Correct 33 ms 3532 KB Output is correct
18 Correct 1173 ms 9116 KB Output is correct
19 Correct 1196 ms 9180 KB Output is correct
20 Correct 1189 ms 9036 KB Output is correct
21 Correct 1239 ms 8920 KB Output is correct
22 Correct 1215 ms 9232 KB Output is correct
23 Correct 1724 ms 10464 KB Output is correct
24 Correct 1734 ms 10316 KB Output is correct
25 Correct 1738 ms 10440 KB Output is correct
26 Correct 33 ms 3432 KB Output is correct
27 Correct 955 ms 9396 KB Output is correct
28 Correct 879 ms 9208 KB Output is correct
29 Correct 998 ms 6852 KB Output is correct
30 Correct 1046 ms 10268 KB Output is correct
31 Correct 990 ms 8332 KB Output is correct
32 Correct 678 ms 8408 KB Output is correct
33 Correct 1112 ms 9408 KB Output is correct
34 Correct 919 ms 8044 KB Output is correct
35 Correct 32 ms 3452 KB Output is correct
36 Correct 1065 ms 7828 KB Output is correct
37 Correct 874 ms 7812 KB Output is correct
38 Correct 770 ms 7816 KB Output is correct
39 Correct 632 ms 6048 KB Output is correct
40 Correct 569 ms 6092 KB Output is correct
41 Correct 719 ms 9368 KB Output is correct
42 Correct 622 ms 9388 KB Output is correct
43 Correct 1487 ms 8732 KB Output is correct
44 Correct 41 ms 3404 KB Output is correct
45 Correct 155 ms 4460 KB Output is correct
46 Correct 69 ms 4572 KB Output is correct
47 Correct 676 ms 7332 KB Output is correct
48 Correct 1434 ms 8716 KB Output is correct
49 Correct 694 ms 7228 KB Output is correct
50 Correct 743 ms 6224 KB Output is correct
51 Correct 710 ms 6384 KB Output is correct
52 Correct 749 ms 6252 KB Output is correct
53 Correct 1117 ms 7708 KB Output is correct
54 Correct 1061 ms 7832 KB Output is correct
55 Correct 1069 ms 7616 KB Output is correct
56 Correct 668 ms 7384 KB Output is correct
57 Correct 636 ms 7420 KB Output is correct
58 Correct 1338 ms 8784 KB Output is correct
59 Correct 1341 ms 8908 KB Output is correct
60 Correct 1323 ms 8776 KB Output is correct
61 Correct 1362 ms 8776 KB Output is correct
62 Correct 1142 ms 8072 KB Output is correct
63 Correct 1133 ms 8016 KB Output is correct
64 Correct 1297 ms 8644 KB Output is correct
65 Correct 766 ms 6948 KB Output is correct
66 Correct 1274 ms 9120 KB Output is correct
67 Correct 1540 ms 10492 KB Output is correct
68 Correct 1272 ms 9276 KB Output is correct
69 Correct 891 ms 8920 KB Output is correct
70 Correct 1437 ms 10320 KB Output is correct
71 Correct 1464 ms 10324 KB Output is correct
72 Correct 1494 ms 10260 KB Output is correct
73 Correct 1254 ms 9104 KB Output is correct
74 Correct 1257 ms 9160 KB Output is correct
75 Correct 1255 ms 9004 KB Output is correct
76 Correct 961 ms 8704 KB Output is correct
77 Correct 942 ms 8832 KB Output is correct
78 Correct 935 ms 8552 KB Output is correct
79 Correct 771 ms 7248 KB Output is correct
80 Correct 775 ms 7076 KB Output is correct
81 Correct 768 ms 7228 KB Output is correct
82 Correct 778 ms 9920 KB Output is correct
83 Correct 770 ms 10036 KB Output is correct
84 Correct 760 ms 9696 KB Output is correct
85 Correct 1748 ms 10960 KB Output is correct
86 Correct 164 ms 6572 KB Output is correct
87 Correct 96 ms 6624 KB Output is correct
88 Correct 1044 ms 10876 KB Output is correct
89 Correct 1719 ms 11488 KB Output is correct
90 Correct 1031 ms 11068 KB Output is correct
91 Correct 1303 ms 9152 KB Output is correct
92 Correct 1308 ms 9052 KB Output is correct
93 Correct 1833 ms 10440 KB Output is correct
94 Correct 1482 ms 10120 KB Output is correct
95 Correct 1469 ms 10464 KB Output is correct
96 Correct 1645 ms 12056 KB Output is correct
97 Correct 760 ms 8264 KB Output is correct
98 Correct 827 ms 11216 KB Output is correct
99 Correct 1744 ms 11316 KB Output is correct
100 Correct 1730 ms 10960 KB Output is correct
101 Correct 1794 ms 11264 KB Output is correct
102 Correct 1795 ms 11372 KB Output is correct
103 Correct 1789 ms 12040 KB Output is correct
104 Correct 1883 ms 12184 KB Output is correct
105 Correct 1398 ms 12104 KB Output is correct
106 Correct 1209 ms 11484 KB Output is correct
107 Correct 1297 ms 8468 KB Output is correct
108 Correct 1765 ms 11224 KB Output is correct
109 Correct 1198 ms 10484 KB Output is correct