24602023-01-13 10:49:12TuruTamasTom és Jerry 3cpp11Hibás válasz 11/5086ms9316 KiB
#include <bits/stdc++.h>
using namespace std;

int T, N, Tom, Jerry, K;

#define vvig vector<vector<int>> &graph
#define vi vector<int>

inline int bfs(vector<int> &d, vvig, int kezd) {
    queue<int> q;
    q.push(kezd);
    d[kezd] = 0;
    int k;
    while (!q.empty())
    {
        k = q.front();
        q.pop();
        for (int x : graph[k]) {
            if (d[x] != -1) continue;
            d[x] = d[k] + 1;
            q.push(x);
        }
    }

    return k;
}

void dfs_visit(vector<bool> &visited, vvig, int a, vi &d) {
    visited[a] = true;
    for (int x: graph[a]) {
        if (visited[x] || d[x] <= K) continue;
        dfs_visit(visited, graph, x, d);
    }
}

void dfs_Jerry(vvig, vi &d_tom, int loc, vector<bool> &visited, int dist, bool &success, bool &end) {
    visited[loc] = true;
    for (int x : graph[loc]) {
        if (end) return;
        if (visited[x] || d_tom[x] <= dist + 2) continue;
        if (d_tom[x] > K) {
            success = true;
            end = true;
            return;
        }
        dfs_Jerry(graph, d_tom, x, visited, dist + 1, success, end);
    }
}



int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin >> T;
    for (size_t q = 0; q < T; q++)
    {
        cin >> N >> Tom >> Jerry >> K;
        Tom--; Jerry--;
        int a, b;
        if (K > N) {
            cout << "IGEN\n";
            for (size_t i = 0; i < N - 1; i++)
            {
                cin >> a >> b;
            }
            continue;
        }

        vector<vi> graph(N);

        bool f = false;
        for (size_t i = 0; i < N - 1; i++)
        {
            cin >> a >> b;
            a--; b--;
            if ((a == Tom && b == Jerry) || (b == Tom && a == Jerry)) {
                f = true;
            }
            if (!f){
                graph[a].push_back(b);
                graph[b].push_back(a);
            }
        }
        if (f) {
            cout << "IGEN\n";
            continue;
        }
        
        vi d_tom(N, -1);
        int max_d_tom = bfs(d_tom, graph, Tom);

        if(d_tom[Jerry] == 1)
        {
            cout << "IGEN\n";
            continue;
        }

        bool success = false;
        vector<bool> visited(N, false);
        bool end = false;
        dfs_Jerry(graph, d_tom, Jerry, visited, 0, success, end);
        if (!success) {
            cout << "IGEN\n";
            continue;
        }
        
        vi d2(N, -1);
        int nagy_tav = bfs(d2, graph, max_d_tom)/2;
        // if (K == 1 && nagy_tav >= 4) {
        //     cout << "NEM\n";
        //     continue;
        // }
        cout << (nagy_tav <= K ? "IGEN\n" : "NEM\n");
    } // q (question)
    cout << '\n';
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base11/50
1Hibás válasz0/03ms1824 KiB
2Hibás válasz0/02ms2300 KiB
3Hibás válasz0/52ms2296 KiB
4Hibás válasz0/13ms2392 KiB
5Hibás válasz0/13ms2512 KiB
6Hibás válasz0/13ms2720 KiB
7Hibás válasz0/13ms2940 KiB
8Hibás válasz0/13ms2940 KiB
9Hibás válasz0/13ms3144 KiB
10Hibás válasz0/13ms3216 KiB
11Hibás válasz0/23ms3348 KiB
12Hibás válasz0/23ms3432 KiB
13Hibás válasz0/13ms3420 KiB
14Elfogadva2/283ms6284 KiB
15Hibás válasz0/279ms5168 KiB
16Hibás válasz0/286ms6476 KiB
17Hibás válasz0/282ms7912 KiB
18Elfogadva2/285ms7020 KiB
19Hibás válasz0/286ms6908 KiB
20Hibás válasz0/257ms9316 KiB
21Hibás válasz0/278ms5492 KiB
22Hibás válasz0/278ms5796 KiB
23Hibás válasz0/382ms8356 KiB
24Elfogadva2/283ms7184 KiB
25Elfogadva3/382ms7208 KiB
26Hibás válasz0/282ms8776 KiB
27Elfogadva2/276ms5804 KiB
28Hibás válasz0/376ms5816 KiB