Submission #1280027

#TimeUsernameProblemLanguageResultExecution timeMemory
1280027dex111222333444555경주 (Race) (IOI11_race)C++17
Compilation error
0 ms0 KiB
#include <bits/stdc++.h>
#include "race.h"
#define ii pair<int, int>
#define fi first
#define se second
#define sz(v) ((int)(v).size())
#define all(v) begin((v)), end(v)
#define TIME (1.0 * clock() / CLOCKS_PER_SEC)
#define dbg(x) "[" #x " = " << x << "]"
using namespace std;
const int MAXN = 2e5 + 5, infINT = 1e9 + 23737, MAXV = 1e6 + 5;
const int dx[4] = {-1, 0, 0, 1};
const int dy[4] = {0, -1, 1, 0};
template<class X, class Y> bool minimize(X &x, const Y &y){return x > y ? x = y, 1: 0;}
template<class X, class Y> bool maximize(X &x, const Y &y){return x < y ? x = y, 1: 0;}
int numNode, numReq, edge[MAXN][2], len[MAXN], best[MAXV], sz[MAXN], res = MAXN + 1;
bool del[MAXN];
vector<ii> adj[MAXN];
vector<int> can;

void updateAns(int u, bool type, int sumDist, int step = 1, int par = 0){
    if (sumDist > numReq) return;
    can.push_back(sumDist);
    if (type){
        minimize(res, best[numReq - sumDist] + step);
    }else{
        minimize(best[sumDist], step);
    }
    for(ii x: adj[u]) if (x.fi != par && !del[x.fi]){
        int v = x.fi, w = x.se;
        updateAns(v, type, sumDist + w, step + 1, u);
    }
}

int getSize(int u, int par = 0){
    sz[u] = 1;
    for(ii x: adj[u]) if (x.fi != par && !del[x.fi]){
        int v = x.fi, w = x.se;
        sz[u] += getSize(v, u);
    }
    return sz[u];
}

int getCentroid(int sizeU, int u, int par = 0){
    for(ii x: adj[u]) if (x.fi != par && !del[x.fi]){
        int v = x.fi, w = x.se;
        if (sz[v] > (sizeU >> 1)) return getCentroid(sizeU, v, u);
    }
    return u;
}

void decomp(int u = 1){
    int centroid = getCentroid(getSize(u), u);
    del[centroid] = 1;

    can.clear();

    for(ii x: adj[centroid]) if (!del[x.fi]) {
        int v = x.fi, w = x.se;
        updateAns(v, true, w);
        updateAns(v, false, w);
    }

    for(int sumDist: can) best[sumDist] = infINT;

    for(ii x: adj[centroid]) if (!del[x.fi]){
        int v = x.fi, w = x.se;
        decomp(v);
    }
}

int best_path(int numNode, int numReq, int edge[][2], int len[])
{
    for(int i = 0; i < numNode - 1; i++){
        int u = edge[i][0] + 1, v = edge[i][1] + 1;
        adj[u].push_back({v, len[i]});
        adj[v].push_back({u, len[i]});
    }
    memset(best, 0x3f, sizeof best);
    best[0] = 0;
    decomp();
    cerr << (res <= MAXN ? res: -1) << '\n';
    return (res <= MAXN ? res: -1);
}

void solve(){
    int numNode = 11, numReq = 12;
    int edge[10][2] = {{0,1},{0,2},{2,3},{3,4},{4,5},{0,6},{6,7},{6,8},{8,9},{8,10}};
    int len[10] = {3,4,5,4,6,3,2,5,6,7};
    cout << best_path(N, K, H, L);
    return 0;
    cout << best_path(numNode, numReq, edge, len) << '\n';
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int t = 1;
    // cin >> t;
    while(t--){
        input();
        solve();
    }
    cerr << TIME << ".s\n";
}

Compilation message (stderr)

race.cpp: In function 'void solve()':
race.cpp:90:23: error: 'N' was not declared in this scope
   90 |     cout << best_path(N, K, H, L);
      |                       ^
race.cpp:90:26: error: 'K' was not declared in this scope
   90 |     cout << best_path(N, K, H, L);
      |                          ^
race.cpp:90:29: error: 'H' was not declared in this scope
   90 |     cout << best_path(N, K, H, L);
      |                             ^
race.cpp:90:32: error: 'L' was not declared in this scope
   90 |     cout << best_path(N, K, H, L);
      |                                ^
race.cpp:91:12: error: return-statement with a value, in function returning 'void' [-fpermissive]
   91 |     return 0;
      |            ^
race.cpp: In function 'int main()':
race.cpp:100:9: error: 'input' was not declared in this scope; did you mean 'int'?
  100 |         input();
      |         ^~~~~
      |         int