24252023-01-13 09:19:24rennTom és Jerry 3cpp11Wrong answer 3/50167ms7952 KiB
#include    <bits/stdc++.h>
using namespace std;

typedef vector<vector<int>> gr;
typedef pair<int, int> pii;

inline int bfs(gr &graf, int kezdo, vector<int> &tavok)
{
    queue<int> next;
    next.push(kezdo);

    int i;
    while(!next.empty())
    {
        i = next.front();
        next.pop();

        for(auto &x : graf[i])
        {
            if(tavok[x] > -1) continue;

            next.push(x);
            tavok[x] = tavok[i]+1;
        }
    }

    return i;
}

inline int bfsm(gr &graf, int kezdo)
{
    queue<int> next;
    next.push(kezdo);
    vector<int> tavok(graf.size(), -1);
    vector<int> prev(graf.size(), -1);

    int i, j;
    while(!next.empty())
    {
        i = next.front();
        next.pop();

        for(auto &x : graf[i])
        {
            if(tavok[x] > -1) continue;

            prev[x] = i;
            next.push(x);
            tavok[x] = tavok[i]+1;
        }
    }

    j = tavok[i]/2;
 
    //j = ((tavok[j] & 1)^1) ? tavok[j]+1 : tavok[j];

    return j+1;
}

int jerry(gr &graf, int j, int &t, vector<int> &tomtavok)
{
    vector<int> jertavok(graf.size(), -1);
    jertavok[j] = 0;

    queue<int> next;
    next.push(j);

    int i;
    while(!next.empty())
    {
        i = next.front();
        next.pop();

        for(auto &x : graf[i])
        {
            if(jertavok[x] > -1 || tomtavok[x] <= jertavok[i]+2) continue;

            next.push(x);
            jertavok[x] = jertavok[i]+1;
        }
    }

    return tomtavok[i];
}

int main()
{
    int T;
    cin >> T;

    int n, t, j, k, nn, a, b;
    int kozep;

    while(T--)
    {
        cin >> n >> t >> j >> k;
        j--;
        t--;
        nn = n-1;

        gr graf(n);
        
        vector<int> tomtavok(n, -1);
        tomtavok[t] = 0;

        while(nn--)
        {
            cin >> a >> b;

            a--; b--;
            graf[a].push_back(b);
            graf[b].push_back(a);

        }

        nn = t;
        
        nn = bfs(graf, nn, tomtavok);

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

        if(jerry(graf, j, t, tomtavok) < k-1)
        {
            cout << "IGEN\n";
            continue;
        }

        kozep = bfsm(graf, nn);

        cout << (kozep < k ? "IGEN" : "NEM") << "\n";
    }

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base3/50
1Accepted0/03ms1680 KiB
2Accepted0/03ms1852 KiB
3Wrong answer0/52ms2052 KiB
4Wrong answer0/13ms2304 KiB
5Wrong answer0/13ms2492 KiB
6Wrong answer0/13ms2692 KiB
7Wrong answer0/13ms2764 KiB
8Wrong answer0/13ms2732 KiB
9Wrong answer0/13ms2728 KiB
10Accepted1/13ms2856 KiB
11Wrong answer0/23ms3060 KiB
12Wrong answer0/23ms3268 KiB
13Wrong answer0/13ms3340 KiB
14Wrong answer0/2162ms6416 KiB
15Wrong answer0/2156ms4948 KiB
16Wrong answer0/2166ms6308 KiB
17Wrong answer0/2153ms7708 KiB
18Wrong answer0/2158ms6176 KiB
19Wrong answer0/2167ms6452 KiB
20Wrong answer0/2104ms7700 KiB
21Wrong answer0/2153ms5052 KiB
22Accepted2/2164ms5072 KiB
23Wrong answer0/3155ms7952 KiB
24Wrong answer0/2160ms6376 KiB
25Wrong answer0/3165ms6648 KiB
26Wrong answer0/2152ms7836 KiB
27Wrong answer0/2151ms5504 KiB
28Wrong answer0/3152ms5388 KiB