Submission #843877

# Submission time Handle Problem Language Result Execution time Memory
843877 2023-09-04T16:22:50 Z nonono Closing Time (IOI23_closing) C++17
Compilation error
0 ms 0 KB
#include "closing.h"
#include <bits/stdc++.h>
using namespace std;

using Graph = vector<vector<pair<int, long long>>> ;

void dfs(int u, int par, const Graph& g, vector<long long>& dist) {
    for(auto [v, w] : g[u]) {
        if(v == par) continue ;
        
        dist[v] = dist[u] + w;
        dfs(v, u, d);
    }
}

int sub1(int N, long long K) {
    vector<long long> a;
    
    for(int i = 0; i < N; i ++) {
        a.push_back(d[0][i]);
        a.push_back(d[1][i]);
    }
    
    sort(a.begin(), a.end());
    
    long long sum = 0;
    int ans = 0;
    
    for(long long x : a) {
        if(sum + x > K) break ;
        sum += x;
        ans += 1;
    }
    
    return ans;
}

int max_score(int N, int X, int Y, long long K, vector<int> U, vector<int> V, vector<int> W) {
    
    Graph g(N);
    vector<int> dX(N, 0), dY(N, 0);
    
    for(int i = 0; i < N - 1; i ++) {
        g[U[i]].push_back({V[i], W[i]});
        g[V[i]].push_back({U[i], W[i]});
    }
    
    dfs(X, X, g, dX);
    dfs(Y, Y, g, dY);
    
    if(dX[Y] > 2 * K) return sub1(N, K);
}

// int main() {
//     int N, X, Y;
//     long long K;
//     cin >> N >> X >> Y >> K;
    
//     vector<int> U(N), V(N), W(N);
    
//     for(int i = 0; i < N - 1; i ++) {
//         cin >> U[i] >> V[i] >> W[i];
//     }
    
//     cout << max_score(N, X, Y, K, U, V, W) << "\n";
//     return 0;
// }

Compilation message

closing.cpp: In function 'void dfs(int, int, const Graph&, std::vector<long long int>&)':
closing.cpp:12:19: error: 'd' was not declared in this scope
   12 |         dfs(v, u, d);
      |                   ^
closing.cpp: In function 'int sub1(int, long long int)':
closing.cpp:20:21: error: 'd' was not declared in this scope
   20 |         a.push_back(d[0][i]);
      |                     ^
closing.cpp: In function 'int max_score(int, int, int, long long int, std::vector<int>, std::vector<int>, std::vector<int>)':
closing.cpp:48:18: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
   48 |     dfs(X, X, g, dX);
      |                  ^~
closing.cpp:7:61: note: in passing argument 4 of 'void dfs(int, int, const Graph&, std::vector<long long int>&)'
    7 | void dfs(int u, int par, const Graph& g, vector<long long>& dist) {
      |                                          ~~~~~~~~~~~~~~~~~~~^~~~
closing.cpp:49:18: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
   49 |     dfs(Y, Y, g, dY);
      |                  ^~
closing.cpp:7:61: note: in passing argument 4 of 'void dfs(int, int, const Graph&, std::vector<long long int>&)'
    7 | void dfs(int u, int par, const Graph& g, vector<long long>& dist) {
      |                                          ~~~~~~~~~~~~~~~~~~~^~~~
closing.cpp:40:14: warning: control reaches end of non-void function [-Wreturn-type]
   40 |     Graph g(N);
      |              ^