| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1000820 | Tsog | 경주 (Race) (IOI11_race) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define pb push_back
#define mp make_pair
using namespace std;
const int INF = INT_MAX;
void dfs(int v, int parent, vector<vector<int>>& dp, const vector<vector<pair<int,int>>>& adj) {
for (int i = 0; i < adj[v].size(); ++i) {
int u = adj[v][i].first;
int cos = adj[v][i].second;
if (u == parent) {
continue;
}
for (int k = cos; k <= 100; ++k) {
if (dp[v][k - cos] != INF) {
dp[u][k] = min(dp[u][k], dp[v][k - cos] + 1);
}
}
dfs(u, v, dp, adj);
}
}
int best_path(int N, int K, vector<vector<int>>& H, vector<int>& L) {
vector<vector<pair<int,int>>> adj(N);
for (int i = 0; i < N - 1; ++i) {
int u = H[i][0], v = H[i][1];
adj[u].push_back(mp(v,L[i]));
adj[v].push_back(mp(u,L[i]));
}
vector<vector<int>> dp(N, vector<int>(200001, INF));
for(int i = 0 ; i < N ; i ++){
dp[i][0] = 0;
}
dfs(0, -1, dp, adj);
int result = INF;
for (int v = 0; v < N; ++v) {
result = min(result, dp[v][K]);
}
return (result == INF ? -1 : result);
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int N;
int K;
int a , b , l , i;
vector<vector<int>> H;
vector<int> L;
cin >> N >> K;
for(i = 1 ; i < N ; i++){
cin >> a >> b;
H.pb({a , b});
}
for(i = 1 ; i < N ; i++){
cin >> l;
L.pb(l);
}
int min_highways = best_path(N, K, H, L);
cout << min_highways;
}