Submission #980400

#	Time	Username	Problem	Language	Result	Execution time	Memory
980400		vjudge1	Road Closures (APIO21_roads)	C++17	0 / 100	769 ms	1048576 KiB

roads

#include "roads.h"
#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define all(a) a.begin(), a.end()

vector<vector<vector<ll>>> dp;
vector<vector<array<ll, 2>>> graph;
vector<bool> used;

void dfs(int n, int p, int dist, int k){
  vector<array<ll, 2>> d;
  for(auto x : graph[n]){
    if(x[0] != p){
      dfs(x[0], n, x[1], k);
      d.push_back({dp[x[0]][k][1] - dp[x[0]][k][0], x[0]});
    }
  }
  sort(all(d));
  ll s1 = 0;
  if((int)graph[n].size() - k > d.size())
    dp[n][k][0] = 0;
  else{
    for (int i = 0; i < (int)graph[n].size() - k; i++)
      s1 += dp[d[i][1]][k][1], used[d[i][1]] = 1;
    for (auto x : graph[n])
      if (x[0] != p and !used[x[0]])
        s1 += dp[x[0]][k][0];
    dp[n][k][0] = s1;
    for (int i = 0; i < (int)graph[n].size() - k; i++)
      used[d[i][1]] = 0;
  }
  if(n != 0){
    s1 = 0;
    for(int i = 0; i < (int)graph[n].size() - k - 1; i ++)
      s1 += dp[d[i][1]][k][1], used[d[i][1]] = 1;
    s1 += dist;
    for(auto x : graph[n])
      if(x[0] != p and !used[x[0]])
        s1 += dp[x[0]][k][0];
    dp[n][k][1] = s1;
    for(int i = 0; i < (int)graph[n].size() - k - 1; i ++)
      used[d[i][1]] = 0;
  }
}

vector<ll> minimum_closure_costs(int n, vector<int> u, vector<int> v, vector<int> w) {
  graph.resize(n);
  for(int i = 0; i < n - 1; i ++)
    graph[u[i]].push_back({v[i], w[i]}), graph[v[i]].push_back({u[i], w[i]});
  used.resize(n);
  dp.resize(n, vector<vector<ll>>(n, vector<ll>(2, 0)));
  for(int i = 0; i < n; i ++)
    dfs(0, -1, 0, i);
  vector<ll> ans(n);
  for(int i = 0; i < n; i ++)
    ans[i] = dp[0][i][0];
  return ans;
}

Compilation message (stderr)

roads.cpp: In function 'void dfs(int, int, int, int)':
roads.cpp:22:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::array<long long int, 2> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   22 |   if((int)graph[n].size() - k > d.size())
      |      ~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 7
• Subtask 3: 0 / 14
• Subtask 4: 0 / 10
• Subtask 5: 0 / 17
• Subtask 6: 0 / 25
• Subtask 7: 0 / 22