oj.uz

Submission #1084721

# Submission time Handle Problem Language Result Execution time Memory
1084721 2024-09-06T18:28:06 Z avighna Firefighting (NOI20_firefighting) C++17
100 / 100
 215 ms 51652 KB 
#include <bits/stdc++.h>

typedef long long ll;

int main() {
  std::ios_base::sync_with_stdio(false);
  std::cin.tie(nullptr);

  ll n, k;
  std::cin >> n >> k;
  std::vector<std::vector<std::pair<ll, ll>>> adj(n + 1);
  for (ll i = 0, u, v, w; i < n - 1; ++i) {
    std::cin >> u >> v >> w;
    adj[u].push_back({v, w});
    adj[v].push_back({u, w});
  }

  std::vector<ll> ans;
  std::function<std::pair<bool, ll>(ll, ll, ll)> dfs;
  dfs = [&](ll x, ll par, ll parent_edge) {
    ll furthest_unprotected = 0, station_reachability = -1e15;
    for (auto &[to, w] : adj[x]) {
      if (to == par) {
        continue;
      }
      auto res = dfs(to, x, w);
      if (res.first) {
        station_reachability = std::max(station_reachability, res.second);
      } else {
        furthest_unprotected = std::max(furthest_unprotected, res.second);
      }
    }
    std::pair<bool, ll> ret;
    if (station_reachability >= furthest_unprotected) {
      ret = {true, station_reachability - parent_edge};
    } else {
      ret = {false, furthest_unprotected + parent_edge};
    }
    if (!ret.first and ret.second > k) {
      ans.push_back(x);
      ret = {true, k - parent_edge};
    }
    if (ret.first and ret.second < 0) {
      ret = {false, 0};
    }
    return ret;
  };
  dfs(1, 0, 1e15);

  std::cout << ans.size() << '\n';
  for (auto &i : ans) {
    std::cout << i << ' ';
  }
  std::cout << '\n';
}

Subtask #1
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 173 ms 34640 KB Output is correct
2 Correct 215 ms 34752 KB Output is correct
3 Correct 48 ms 13108 KB Output is correct
4 Correct 162 ms 33724 KB Output is correct
5 Correct 0 ms 344 KB Output is correct
6 Correct 0 ms 344 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 344 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 0 ms 448 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 0 ms 344 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 1 ms 344 KB Output is correct
14 Correct 0 ms 348 KB Output is correct
15 Correct 0 ms 348 KB Output is correct

Subtask #3
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 452 KB Output is correct
11 Correct 0 ms 348 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 1 ms 604 KB Output is correct
15 Correct 0 ms 348 KB Output is correct
16 Correct 0 ms 348 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 344 KB Output is correct
23 Correct 1 ms 348 KB Output is correct
24 Correct 0 ms 348 KB Output is correct
25 Correct 0 ms 348 KB Output is correct
26 Correct 0 ms 348 KB Output is correct

Subtask #4
19.0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 161 ms 34764 KB Output is correct
2 Correct 68 ms 17360 KB Output is correct
3 Correct 64 ms 17876 KB Output is correct
4 Correct 63 ms 15508 KB Output is correct
5 Correct 0 ms 344 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 118 ms 30944 KB Output is correct
8 Correct 128 ms 30864 KB Output is correct
9 Correct 126 ms 31164 KB Output is correct
10 Correct 148 ms 31160 KB Output is correct
11 Correct 179 ms 34848 KB Output is correct
12 Correct 85 ms 21452 KB Output is correct
13 Correct 49 ms 13388 KB Output is correct
14 Correct 94 ms 20164 KB Output is correct
15 Correct 98 ms 24264 KB Output is correct
16 Correct 103 ms 25880 KB Output is correct
17 Correct 87 ms 21880 KB Output is correct
18 Correct 100 ms 21448 KB Output is correct

Subtask #5
26.0 / 26.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 604 KB Output is correct
2 Correct 1 ms 604 KB Output is correct
3 Correct 1 ms 420 KB Output is correct
4 Correct 1 ms 604 KB Output is correct
5 Correct 1 ms 860 KB Output is correct
6 Correct 1 ms 860 KB Output is correct
7 Correct 1 ms 856 KB Output is correct
8 Correct 1 ms 860 KB Output is correct
9 Correct 1 ms 724 KB Output is correct
10 Correct 2 ms 860 KB Output is correct
11 Correct 2 ms 856 KB Output is correct
12 Correct 1 ms 600 KB Output is correct
13 Correct 2 ms 860 KB Output is correct
14 Correct 1 ms 860 KB Output is correct
15 Correct 1 ms 604 KB Output is correct
16 Correct 1 ms 580 KB Output is correct
17 Correct 1 ms 600 KB Output is correct
18 Correct 2 ms 604 KB Output is correct
19 Correct 1 ms 604 KB Output is correct
20 Correct 1 ms 468 KB Output is correct
21 Correct 1 ms 604 KB Output is correct

Subtask #6
38.0 / 38.0

# Verdict Execution time Memory Grader output
1 Correct 147 ms 29268 KB Output is correct
2 Correct 119 ms 27728 KB Output is correct
3 Correct 131 ms 29944 KB Output is correct
4 Correct 59 ms 14168 KB Output is correct
5 Correct 205 ms 45248 KB Output is correct
6 Correct 179 ms 43160 KB Output is correct
7 Correct 138 ms 49096 KB Output is correct
8 Correct 143 ms 47672 KB Output is correct
9 Correct 165 ms 40652 KB Output is correct
10 Correct 146 ms 39528 KB Output is correct
11 Correct 156 ms 51652 KB Output is correct
12 Correct 93 ms 20308 KB Output is correct
13 Correct 156 ms 42256 KB Output is correct
14 Correct 180 ms 37828 KB Output is correct
15 Correct 127 ms 30292 KB Output is correct
16 Correct 119 ms 28500 KB Output is correct
17 Correct 145 ms 28244 KB Output is correct
18 Correct 122 ms 30028 KB Output is correct
19 Correct 79 ms 21064 KB Output is correct
20 Correct 51 ms 12232 KB Output is correct
21 Correct 139 ms 30288 KB Output is correct