Submission #870052

# Submission time Handle Problem Language Result Execution time Memory
870052 2023-11-06T19:17:21 Z ThommyDB Closing Time (IOI23_closing) C++17
Compilation error
0 ms 0 KB
#include<bits/stdc++.h>

using namespace std;

vector<vector<pair<int, int>>> adj;
vector<bool> visited;
vector<int> distX, distY;

void dfs(int curr, long long d, vector<long long> &dist){
    visited[curr] = true;
    dist[curr] =d;
    for(pair<int, int> u : adj[curr]){
        if(!visited[u.first]){
            dfs(u.first, d+u.second, dist);
        }
    }
}

int solve(int N, long long K, int mx){
    int ans = 0;
    priority_queue<pair<int, int>> q;

    for(int i = 0; i < N; i++){
        q.push({min(distX[i], distY[i]), i});
    }
    vector<int> cnt(N, 0);
    while(!q.empty()){
        while(!q.empty() && (K < q.top().first || cnt[q.top().second] > mx-1)) q.pop();
        if(q.empty()) break;

        int t = q.top().first;
        int x = q.top().second;
        q.pop();
        ans++;
        cnt[x]++;
        K-=t;
        if(cnt[x]==1 && mx == 2){
            q.push({abs(distX[x] - distY[x]), x});
        }
    }
    if(K<0) return 0;
    else return ans;
}

int max_score(int N, int X, int Y, long long K, vector<int> U, vector<int> V, vector<int> W){
    adj.assign(N, vector<pair<int, int>>(0));
    for(int i = 0; i < N-1; i++){
        adj[U[i]].push_back({V[i], W[i]});
        adj[V[i]].push_back({U[i], W[i]});
    }
    distX.resize(N, 0);
    distY.resize(N, 0);
    visited.assign(N, false);
    dfs(0, 0, distX);
    visited.assign(N, false);
    dfs(0, 0, distY);
    return max(solve(N, K, 1), solve(N, K, 2));
}

Compilation message

closing.cpp: In function 'int max_score(int, int, int, long long int, std::vector<int>, std::vector<int>, std::vector<int>)':
closing.cpp:54:15: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
   54 |     dfs(0, 0, distX);
      |               ^~~~~
closing.cpp:9:52: note: in passing argument 3 of 'void dfs(int, long long int, std::vector<long long int>&)'
    9 | void dfs(int curr, long long d, vector<long long> &dist){
      |                                 ~~~~~~~~~~~~~~~~~~~^~~~
closing.cpp:56:15: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
   56 |     dfs(0, 0, distY);
      |               ^~~~~
closing.cpp:9:52: note: in passing argument 3 of 'void dfs(int, long long int, std::vector<long long int>&)'
    9 | void dfs(int curr, long long d, vector<long long> &dist){
      |                                 ~~~~~~~~~~~~~~~~~~~^~~~