제출 #961085

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
961085		okkoo	경주 (Race) (IOI11_race)	C++17	0 / 100	7 ms	13400 KiB

race

#include <iostream>
#include <vector>
#include <queue>
#include <string.h>
using namespace std;
#define fastIO ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);

const int mxn = 2e5;
vector<vector<pair<int, int> > > adj(mxn+1, vector<pair<int, int> >());
int K;
int ans = 1e9;
bool processed[mxn+1];
int sz[mxn+1];
int lengths[mxn*5+1];

int getSize(int node, int p){
    sz[node] = 1;
    for(pair<int, int> tmp: adj[node]){
        int to = tmp.first;
        if(!processed[to] && to!=p) sz[node] += getSize(to, node);
    }
    return sz[node];
}

int getCentroid(int node, int p, int mxSize){
    for(pair<int, int> tmp: adj[node]){
        int to = tmp.first;
        if(!processed[to] && to!=p && sz[to] > mxSize) return getCentroid(to, node, mxSize);
    }
    return node;
}

queue<pair<int, int> > q;
queue<int> l;

void dfs(int node, int p, int edgeLen, int edgeCnt){
    if(K < edgeLen) return;
    ans = min(ans, lengths[edgeLen-K] + edgeCnt);
    q.push({edgeLen, edgeCnt});
    l.push(edgeLen);
    for(pair<int, int> tmp: adj[node]){
        int to = tmp.first;
        if(!processed[to] && to!=p) dfs(to, node, edgeLen + tmp.second, edgeCnt+1);
    }
    if(!p){
        while(!q.empty()){
            lengths[q.front().first] = min(lengths[q.front().first], q.front().second);
            q.pop();
        }
    }
}

void centroidDecomposition(int node){
    int centroid = getCentroid(node, 0, getSize(node, 0)/2);
    processed[centroid] = 1;
    lengths[0] = 0;
    for(pair<int, int> tmp: adj[node]){
        int to = tmp.first;
        if(!processed[to]) dfs(node, 0, tmp.second, 1);
    }
    while(!l.empty()){
        lengths[l.front()] = 1e9;
        l.pop();
    }
    for(pair<int, int> tmp: adj[node]){
        int to = tmp.first;
        if(!processed[to]) centroidDecomposition(to);
    }
}

int best_path(int n, int k, int h[][2], int l[]){
    fastIO;
    K = k;
    memset(lengths, 0x3f, sizeof(lengths));
    for(int i=0; i<n-1; i++){
        adj[h[i][0]].push_back({h[i][1], l[i]});
        adj[h[i][1]].push_back({h[i][0], l[i]});
    }

    centroidDecomposition(1);

    return (ans == 1e9 ? -1 : ans);
}

• Subtask 1: 0 / 9
• Subtask 2: 0 / 12
• Subtask 3: 0 / 22
• Subtask 4: 0 / 57