oj.uz

답안 #977201

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
977201 2024-05-07T13:44:49 Z saayan007 경주 (Race) (IOI11_race) C++17
0 / 100
 2 ms 11612 KB 
#include "race.h"
#include "bits/stdc++.h"
using namespace std;
#warning notusing64bitintegers

#define rep(i, a, b) for(int i = (a); i <= (b); ++i)
#define repd(i, a, b) for(int i = (a); i >= (b); --i)

#define em emplace
#define eb emplace_back

const int mxN = 2e5L + 10;
const int mxK = 1e6L + 10;
int n, k;
vector<pair<int, int>> adj[mxN];
int ans = mxN;
int sz[mxN];
bool proc[mxN] = {};
int dp[mxK];

int get_sz(int x, int p) {
    sz[x] = 1;
    for(auto [y, w] : adj[x]) {
        if(proc[y] || y == p) continue;
        sz[x] += get_sz(y, x);
    }
    return sz[x];
}

int get_cen(int x, int p, int tot) {
    for(auto [y, w] : adj[x]) {
        if(proc[y] || y == p || sz[y] * 2 <= tot) continue;
        return get_cen(y, x, tot);
    }
    return x;
}

void dfs(int x, int p, int dep, int dis, bool flag) {
    if(dis > k) return;
    if(flag) {
        dp[dis] = mxN;
    }
    else {
        if(dp[k - dis] != mxN) ans = min(ans, dep + dp[k - dis]);
        dp[dis] = min(dp[dis], dep);
    }

    for(auto [y, w] : adj[x]) {
        if(proc[y] || y == p) continue;
        dfs(y, x, dep + 1, dis + w, flag);
    }
}

void solve(int x = 1, int p = -1) {
    int c = get_cen(x, p, get_sz(x, p));
    dfs(c, -1, 0, 0, 0);
    dfs(c, -1, 0, 0, 1);
    proc[c] = 1;
    for(auto [d, w] : adj[c]) {
        if(proc[d]) continue;
        solve(d, c);
    }
}

int best_path(int N, int K, int H[][2], int L[])
{
    rep(i, 0, K) dp[i] = mxN;
    n = N; k = K;
    rep(i, 0, N - 2) {
        adj[H[i][0] + 1].eb(H[i][1] + 1, L[i]);
        adj[H[i][1] + 1].eb(H[i][0] + 1, L[i]);
    }

    solve();
    return (ans > n - 1 ? -1 : ans);
}

Compilation message

race.cpp:4:2: warning: #warning notusing64bitintegers [-Wcpp]
    4 | #warning notusing64bitintegers
      |  ^~~~~~~

Subtask #1
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 11612 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 11612 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 11612 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 57.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 11612 KB Output isn't correct
2 Halted 0 ms 0 KB -