8114 2024. 01. 12 13:50:59 adam Tom és Jerry 1 (80) cpp17 Futási hiba 0/80 108ms 19388 KiB
#include <bits/stdc++.h>

using namespace std;
vector<vector<int>> labyrinth_jerry;
vector<vector<int>> labyrinth_tom;

vector<pair<int, int>> bfs_jerry (int from, int to, vector<vector<int>> &labyrinth) {
    queue<vector<int>> q; // first: node, second: depth;
    vector<bool> been_here(labyrinth.size(), false);
    vector<int> path;
    vector<pair<int, int>> depths;
    q.push({from, 0, -1});
    while(!q.empty()) {
        if (q.front()[0] == to) {
            depths[q.front()[0]] = pair(q.front()[0], q.front()[1]);
            return depths;
        }
        for (int i = 0; i < labyrinth[q.front()[0]].size(); i++) {
            if(been_here[labyrinth[q.front()[0]][i]]) continue;
            q.push({labyrinth[q.front()[0]][i], q.front()[1]+1, q.front()[0]});
            been_here[labyrinth[q.front()[0]][i]] = true;

        }
        if (path.empty() || q.front()[2] == path[path.size() -1]) {
            depths.push_back(pair(q.front()[0], q.front()[1]));
            path.push_back(q.front()[0]);
        };
        q.pop();
    }
    return depths;
}
map<int, int> bfs_tom (int from, vector<vector<int>> &labyrinth) {
    queue<vector<int>> q; // first: node, second: depth;
    vector<bool> been_here(labyrinth.size(), false);
    vector<int> path;
    map<int, int> depths;
    q.push({from, 0, -1});
    while(!q.empty()) {
        for (int i = 0; i < labyrinth[q.front()[0]].size(); i++) {
            if(been_here[labyrinth[q.front()[0]][i]]) continue;
            q.push({labyrinth[q.front()[0]][i], q.front()[1]+1, q.front()[0]});
            been_here[labyrinth[q.front()[0]][i]] = true;

        }
        if (path.empty() || q.front()[2] == path[path.size() -1]) {
            depths[q.front()[0]] = q.front()[1];
            path.push_back(q.front()[0]);
        };
        q.pop();
    }
    return depths;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int point_count, road_count, tom_pos, tries_count, hole_pos;
    cin >> point_count >> road_count >> tom_pos >> tries_count >> hole_pos;
    labyrinth_jerry.assign(road_count, vector(0, 0));
    labyrinth_tom.assign(road_count, vector(0, 0));
    for (int i = 0; i < road_count; i++) {
        int road_from, road_to, width;
        cin >> road_from >> road_to >> width;

        if (width == 2) {
            labyrinth_jerry[road_from].push_back(road_to);
            labyrinth_jerry[road_to].push_back(road_from);
            labyrinth_tom[road_from].push_back(road_to);
            labyrinth_tom[road_to].push_back(road_from);
        } else {
            labyrinth_jerry[road_from].push_back(road_to);
            labyrinth_jerry[road_to].push_back(road_from);
        }
    }



    vector<int> tries(tries_count, 0);
    for (int i = 0; i < tries_count; i++) {
        cin >> tries[i];
    }
    map<int, int> tom_path = bfs_tom(tom_pos, labyrinth_tom);

    for (int i = 0; i < tries.size(); i++) {
        vector<pair<int, int>> jerry_depths = bfs_jerry(tries[i], hole_pos, labyrinth_jerry);
        string log;
        for (pair<int, int> jerr : jerry_depths) {
            if (tom_path.count(jerr.first) != 0 && tom_path[jerr.first] <= jerr.second) {
                log = "NEM";
                continue;
            }
        }
        if(log.length() == 0) log = "IGEN";

        cout << log << endl;

    }
    return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 0/80
1 Elfogadva 0/0 3ms 1888 KiB
2 Futási hiba 0/0 4ms 2616 KiB
3 Futási hiba 0/4 4ms 2780 KiB
4 Hibás válasz 0/4 3ms 2384 KiB
5 Hibás válasz 0/4 3ms 2392 KiB
6 Futási hiba 0/4 3ms 2492 KiB
7 Hibás válasz 0/4 3ms 2696 KiB
8 Hibás válasz 0/4 4ms 3084 KiB
9 Futási hiba 0/4 4ms 3208 KiB
10 Futási hiba 0/4 4ms 3404 KiB
11 Hibás válasz 0/4 10ms 4732 KiB
12 Hibás válasz 0/4 14ms 4964 KiB
13 Futási hiba 0/4 16ms 6896 KiB
14 Hibás válasz 0/4 54ms 10508 KiB
15 Futási hiba 0/4 35ms 12708 KiB
16 Futási hiba 0/4 41ms 14516 KiB
17 Futási hiba 0/4 108ms 19388 KiB
18 Hibás válasz 0/4 50ms 13296 KiB
19 Futási hiba 0/4 46ms 14020 KiB
20 Futási hiba 0/4 46ms 14000 KiB
21 Futási hiba 0/4 37ms 12036 KiB
22 Futási hiba 0/4 92ms 18972 KiB