5734 2023. 09. 10 17:40:42 kukkerman A lehető legkevesebb metróval utazás (40 pont) cpp17 Elfogadva 40/40 361ms 24112 KiB
#include <iostream>
#include <vector>
#include <unordered_set>
#include <deque>
#include <algorithm>

void beolvas(std::istream &in, size_t &ind, size_t &erk, std::vector<std::vector<size_t>> &vonalak) {
    size_t n, m;
    in >> n >> m >> ind >> erk;

    vonalak.resize(n);
    for (auto i = 0u; i < n; i++) {
        size_t a;
        in >> a;

        vonalak[i].resize(a);
        for (auto j = 0u; j < a; j++) {
            in >> vonalak[i][j];
        }
    }
}

bool van_kozos_allomas(const std::vector<size_t> &vonal1, const std::vector<size_t> &vonal2) {
    const auto h1 = vonal1.size();
    const auto h2 = vonal2.size();

    auto i = 0u;
    auto j = 0u;
    while (i < h1 && j < h2 && vonal1[i] != vonal2[j]) {
        if (vonal1[i] < vonal2[j]) {
            i++;
        } else {
            j++;
        }
    }

    return i < h1 && j < h2;
}

void feldolgoz(size_t ind, size_t erk, std::vector<std::vector<size_t>> &vonalak) {
    const auto n = vonalak.size();

    std::vector<bool> erk_vonal(n, false);
    std::deque<std::pair<size_t, size_t>> sor;

    for (auto v = 0u; v < n; v++) {
        for (auto a : vonalak[v]) {
            if (a == ind) {
                sor.emplace_back(v, v);

            } else if (a == erk) {
                erk_vonal[v] = true;
            }
        }

        std::sort(vonalak[v].begin(), vonalak[v].end());
    }

    std::vector<std::unordered_set<size_t>> sz(n);
    for (auto v1 = 0u; v1 < n - 1; v1++) {
        for (auto v2 = v1 + 1; v2 < n; v2++) {
            if (van_kozos_allomas(vonalak[v1], vonalak[v2])) {
                sz[v1].insert(v2);
                sz[v2].insert(v1);
            }
        }
    }

    std::vector<size_t> elozo_vonal(n, n);
    std::vector<size_t> utvonal;
    while (!sor.empty() && utvonal.empty()) {
        const auto [v, elozo] = sor.front();
        sor.pop_front();

        if (elozo_vonal[v] == n) {
            elozo_vonal[v] = elozo;

            if (erk_vonal[v]) {
                size_t i = v;
                while (elozo_vonal[i] != i) {
                    utvonal.push_back(i);
                    i = elozo_vonal[i];
                }
                utvonal.push_back(i);

            } else {
                for (auto csatl : sz[v]) {
                    sor.emplace_back(csatl, v);
                }
            }
        }
    }

    if (!utvonal.empty()) {
        const auto hossz = utvonal.size();
        std::cout << hossz << std::endl;
        for (auto i = 1u; i <= hossz; i++) {
            std::cout << utvonal[hossz - i] + 1 << ' ';
        }

    } else {
        std::cout << -1;
    }
    std::cout << std::endl;
}

int main() {
    std::vector<std::vector<size_t>> vonalak;
    size_t ind, erk;

    beolvas(std::cin, ind, erk, vonalak);
    feldolgoz(ind, erk, vonalak);

    return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 40/40
1 Elfogadva 0/0 3ms 1808 KiB
2 Elfogadva 0/0 8ms 2552 KiB
3 Elfogadva 2/2 2ms 2416 KiB
4 Elfogadva 2/2 2ms 2396 KiB
5 Elfogadva 2/2 3ms 2612 KiB
6 Elfogadva 2/2 2ms 2772 KiB
7 Elfogadva 2/2 4ms 2916 KiB
8 Elfogadva 2/2 4ms 2944 KiB
9 Elfogadva 2/2 4ms 3072 KiB
10 Elfogadva 2/2 4ms 3284 KiB
11 Elfogadva 2/2 3ms 3576 KiB
12 Elfogadva 2/2 9ms 4076 KiB
13 Elfogadva 2/2 9ms 4240 KiB
14 Elfogadva 2/2 8ms 4508 KiB
15 Elfogadva 2/2 356ms 23792 KiB
16 Elfogadva 2/2 358ms 23696 KiB
17 Elfogadva 2/2 356ms 23984 KiB
18 Elfogadva 2/2 361ms 24112 KiB
19 Elfogadva 2/2 7ms 4612 KiB
20 Elfogadva 2/2 8ms 4636 KiB
21 Elfogadva 2/2 4ms 4584 KiB
22 Elfogadva 2/2 8ms 4816 KiB