Submission #1150020

#TimeUsernameProblemLanguageResultExecution timeMemory
1150020knhatdevRace (IOI11_race)C++20
43 / 100
3097 ms49104 KiB
#include<bits/stdc++.h>
#define ii pair<int,int>
#define fi first
#define se second
// #define int long long
#define task ""
using namespace std;

const int maxn = 1e6 + 10, mod = 1e9 + 7;
const int oo = 1e9;
int n, k, ans, mi[maxn], ma_h;
int sz[maxn], h[maxn], par[maxn], root;
bool is_centroid[maxn];
vector<ii> g[maxn];

void dfs(int u, int par){
    sz[u] = 1;
    for(ii k : g[u]){
        int v = k.fi, c = k.se;
        if(v == par || is_centroid[v]) continue;
        dfs(v, u);
        sz[u] += sz[v];
    }
}
int find_centroid(int u, int sz_tree, int par){
    for(ii k : g[u]){
        int v = k.fi, c = k.se;
        if(v != par && !is_centroid[v] && sz[v] > sz_tree/2)
            return find_centroid(v, sz_tree, u);
    }
    return u;
}

void get(int u, int par, int h, int w, int t){
    if(w > k) return;
    if(t == 0)
        ans = min(ans, mi[k - w] + h);
    else
        mi[w] = min(mi[w], h);
    ma_h = max(ma_h, w);
    for(ii k : g[u]){
        int v = k.fi, c = k.se;
        if(v == par || is_centroid[v]) continue;
        get(v, u, h + 1, w + c, t);
    }
}

void build_centroid(int u, int par){
    dfs(u, -1);
    int centroid = find_centroid(u, sz[u], -1);
    // cout << u << ' ' << sz[u] << '\n';
    // cout << u << ' ' << centroid << '\n';
    if(root == 0) root = centroid;
    is_centroid[centroid] = 1;
    for(ii k : g[centroid]){
        int v = k.fi, c = k.se;
        if(!is_centroid[v]){
            get(v, centroid, 1, c, 0);
            get(v, centroid, 1, c, 1);
        }
    }
    for(int i = 1; i <= ma_h; i++) mi[i] = oo;
    for(ii k : g[centroid]){
        int v = k.fi;
        if(!is_centroid[v]) build_centroid(v, centroid);
    }
}
int best_path(int N, int K, int H[][2], int L[]){
    n = N; k = K;
    for(int i = 0; i < n - 1; i++){
        H[i][0]++; H[i][1]++;
        g[H[i][0]].push_back(ii(H[i][1], L[i]));
        g[H[i][1]].push_back(ii(H[i][0], L[i]));
    }
    // memset(is_centroid, 0, sizeof(is_centroid));
    for(int i = 1; i <= k; i++)
        mi[i] = oo;
    ans = oo;
    build_centroid(1, -1);
    return (ans == oo ? -1 : ans);
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...