Submission #1279399

#	Time	Username	Problem	Language	Result	Execution time	Memory
1279399		dhuyyyy	Robots (APIO13_robots)	C++20	0 / 100	1 ms	332 KiB

robots

#include <bits/stdc++.h>
//#include "race.h"
#define fi first
#define se second
using namespace std;

using ii = pair<int,int>;

const int maxn = 2e5+5;
const int maxk = 1e6+5;


int res = 1e9 + 9;

int pos[maxk], sub[maxn];

bool num[maxn];

vector <ii> adj[maxn];

void dfs(int u,int p){
    sub[u] = 1;
    for (ii it : adj[u]){
        if (it.fi == p || num[it.fi]) continue;
        dfs(it.fi,u);
        sub[u] += sub[it.fi];
    }
}
int centroid(int u,int p,int n){
    for (ii it : adj[u]){
        if (it.fi == p || num[it.fi]) continue;
        if (sub[it.fi] * 2 > sub[u]) return centroid(it.fi,u,n);
    }
    return u;
}
void getans(int u,int p,int cur,int km,int K){
    if (km > K) return;
    res = min(res,cur + pos[K - km]);
    for (ii it : adj[u]){
        if (it.fi == p || num[it.fi]) continue;
        getans(it.fi,u,cur + 1,km + it.se,K);
    }
    pos[km] = min(pos[km],cur);
}
void solve(int u,int K){
    dfs(u,0);
    int m = sub[u];
    int root = centroid(u,0,m);
    num[root] = 1;
    for (ii it : adj[root]){
        if (num[it.fi]) continue;
        getans(it.fi,root,1,it.se,K);
    }
    for (ii it : adj[u]){
        if (num[it.fi]) continue;
        solve(it.fi,K);
    }
}
int best_path(int N, int K, int H[][2], int L[]){
    for (int i = 1; i <= K; i++) pos[i] = 1e9 + 9;
    for (int i = 0; i < N - 1; i++){
        H[i][0]++; H[i][1]++;
        adj[H[i][0]].push_back({H[i][1],L[i]});
        adj[H[i][1]].push_back({H[i][0],L[i]});
    }
    solve(1,K);
    return (res <= 0 || res > N ? -1 : res);
}

signed main(){
    ios_base::sync_with_stdio(0); cin.tie(0);

    cout << 5;

    return 0;
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 20
• Subtask 3: 0 / 30
• Subtask 4: 0 / 40