Submission #1360253

#TimeUsernameProblemLanguageResultExecution timeMemory
1360253altern23Designated Cities (JOI19_designated_cities)C++20
0 / 100
145 ms40172 KiB
#include <bits/stdc++.h>
using namespace std;

#define ll long long

const int MAXN = 2e5;

vector <pair<ll, ll>> adj[MAXN+5];
ll M;

ll dist[MAXN+5], ans[MAXN+5], tmp[MAXN+5], P[MAXN+5];

vector <pair<ll, ll>> dp[MAXN+5];

void dfs(ll idx, ll par) {
      dp[idx].push_back({dist[idx], idx});
      for (auto [i, j] : adj[idx]) {
            if (i == par) M -= j;
            else {
                  dist[i] = dist[idx]+j;
                  P[i] = idx;
                  dfs(i, idx);
                  dp[idx].push_back({dp[i].back().first, dp[i].back().second});
            }
      }
      sort(dp[idx].begin(), dp[idx].end());
}

void init(ll idx, ll par) {
      for (auto [i, j] : adj[idx]) {
            if (i != par) {
                  init(i, idx);
                  dist[i] = dist[idx]+j;
            }
      }
}

int main() {
      ios_base::sync_with_stdio(0); cin.tie(0);
      int tc = 1;
      // cin >> tc;
      while (tc--) {
            ll N; cin >> N;

            for (int i = 0; i <= N; i++) {
                  ans[i] = 1e18;
                  tmp[i] = 1e18;
            }
            
            for (int i = 2; i <= N; i++) {
                  ll u, v, a, b; cin >> u >> v >> a >> b;
                  M += a+b;
                  adj[u].push_back({v, a});
                  adj[v].push_back({u, b});

            }

            ll root = 0;
            init(1, -1);
            
            for (int i = 1; i <= N; i++) {
                  if (dist[root] < dist[i]) root = i;
            }

            for (int i = 1; i <= N; i++) {
                  dist[i] = 0;
            }

            dfs(root, -1);

            priority_queue<pair<ll, ll>> pq;

            vector <ll> vis(N+5);

            pq.push(dp[root].back());
            ll x = 1;
            tmp[x] = M;
            ans[x] = min(ans[x], tmp[x]);

            while (!pq.empty()) {
                  auto [val, idx] = pq.top(); pq.pop();
                  tmp[x+1] = tmp[x]-val;
                  ans[x+1] = min(ans[x+1], tmp[x+1]);
                  x++;
                  while (idx) {
                        dp[idx].pop_back();
                        if (!dp[idx].empty()) {
                              pq.push({dp[idx].back().first-dist[idx], dp[idx].back().second});
                        }
                        if (vis[idx]) break;
                        vis[idx] = 1;
                        idx = P[idx];
                  }
            }

            for (int i = 1; i <= N; i++) ans[i] = min(ans[i-1], ans[i]);

            ll Q; cin >> Q;
            for (int i = 1; i <= Q; i++) {
                  ll z; cin >> z;
                  cout << ans[z] << "\n";
            }
      }
}

/*

*/
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...