# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
123602 | johutha | 악어의 지하 도시 (IOI11_crocodile) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "crocodile.h"
#include <vector>
#define int int64_t
using namespace std;
vector<int> vals;
vector<vector<pair<int,int>>> adjlist;
int dfs(int curr, int prev)
{
if (vals[curr] != -1) return vals[curr];
vals[curr] = 1e14;
int fr = 1e14;
int sec = 1e14;
for (auto p : adjlist[curr])
{
int next = p.first;
if (next == prev) continue;
int val = dfs(next, curr) + p.second;
if (val <= fr)
{
sec = fr;
fr = val;
}
else if (val <= sec)
{
sec = val;
}
}
return vals[curr] = sec;
}
int travel_plan(int n, int m, int R[][2], int L[], int k, int P[])
{
vals.resize(n, -1);
for (int i = 0; i < k; i++)
{
vals[P[i]] = 0;
}
adjlist.resize(n);
for (int i = 0; i < m; i++)
{
adjlist[R[i][0]].push_back({R[i][1], L[i]});
adjlist[R[i][1]].push_back({R[i][0], L[i]});
}
return dfs(0, -1);
}