Submission #639030

#	Time	Username	Problem	Language	Result	Execution time	Memory
639030		thienbao1602	Race (IOI11_race)	C++17	Compilation error	0 ms	0 KiB

race

#include    <bits/stdc++.h>
#define ll long long
#define pi pair<ll, ll>
#define fi first
#define se second
#define sz(x) (int)(x.size())
using namespace std;

const int N = 2e5 + 55;
const ll inf = 1e17;

int n, dist[N];
ll sum[N], K, bestWay = inf;
map<ll, ll> info[N];
vector<pi> edges[N];

void dfs(int u, int par, ll curSum, int deep)
{
    dist[u] = deep;
    sum[u] = curSum;
    info[u][curSum] = deep;

    for(auto [r, v] : edges[u])
    {
        if (v != par)
        {
            dfs(v, u, curSum + r, deep + 1);
        }
    }
}

void merge_small_to_large(int u, int par)
{
    ll target = K + 2 * sum[u];
    for(auto [r, v] : edges[u])
    {
        if (v != par)
        {
            merge_small_to_large(v, u);
            if (sz(info[u]) < sz(info[v])) swap(info[u], info[v]);
            for(auto x : info[v])
            {
                if (info[u].find(target - x.fi) != info[u].end())
                {
                    bestWay = min(bestWay, info[u][target - x.fi] + x.se  - 2 * dist[u]);
                }
            }

            for(auto x : info[v])
            {
                if (info[u].find(x.fi) == info[u].end())
                {
                    info[u][x.fi] = x.se;
                } else
                {
                    info[u][x.fi] = min(info[u][x.fi], x.se);
                }
            }
        }
    }
}

ll best_path(int n, ll K, vector<vector<ll>> H, vector<ll> L)
{
    for(int i=0; i<n-1; i++)
    {
        edges[H[i][0]].push_back({L[i], H[i][1]});
        edges[H[i][1]].push_back({L[i], H[i][0]});
    }

    dfs(0, n + 1, 0, 0);
    merge_small_to_large(0, n + 1);
    return (bestWay == inf ? -1 : bestWay);
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin >> n >> K;
    vector<ll> L(n);
    vector<vector<ll>> H(n, vector<ll>(2));
    for(int i=0; i<n-1; i++)
    {
        cin >> H[i][0] >> H[i][1] >> L[i];
    }
    cout << best_path(n, K, H, L);
    return 0;
}

Compilation message (stderr)

/usr/bin/ld: /tmp/ccZqUoAa.o: in function `main':
race.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/cc6Xd9rb.o:grader.cpp:(.text.startup+0x0): first defined here
/usr/bin/ld: /tmp/cc6Xd9rb.o: in function `main':
grader.cpp:(.text.startup+0x28): undefined reference to `best_path(int, int, int (*) [2], int*)'
collect2: error: ld returned 1 exit status