제출 #984906

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
984906		54skyxenon	Crocodile's Underground City (IOI11_crocodile)	C++17	46 / 100	3 ms	4956 KiB

crocodile

// https://oj.uz/problem/view/IOI11_crocodile

/** Needed for linking!!! */
#include "crocodile.h"

#include <bits/stdc++.h>
using namespace std;

#define INF 2000000000

vector<map<int, int>> graph;
vector<bool> is_exit;
vector<bool> visited;

int dfs(int curr) {
    if (is_exit[curr]) {
        return 0;
    }

    vector<int> distances;
    for (auto [nei, weight] : graph[curr]) {
        if (!visited[nei]) {
            visited[nei] = true;
            distances.push_back(min(INF, weight + dfs(nei)));
        }
        else {
            distances.push_back(INF);
        }
    }

    sort(distances.begin(), distances.end());
    assert(distances.size() > 1);
    return distances[1];
}

int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]) {
    graph.resize(N);
    is_exit.resize(N);
    visited.resize(N);

    for (int i = 0; i < M; i++) {
        graph[R[i][0]][R[i][1]] = graph[R[i][1]][R[i][0]] = L[i];
    }

    for (int i = 0; i < K; i++) {
        is_exit[P[i]] = true;
    }

    visited[0] = true;
    return dfs(0); 
}

• Subtask 1: 46 / 46
• Subtask 2: 0 / 43
• Subtask 3: 0 / 11