190812025-11-21 18:57:42szjTom és Jerry 1 (80)cpp17Accepted 80/80209ms6832 KiB
#include <bits/stdc++.h>
using namespace std;
struct el
{
    int csucs, szell;
}ur;

int main()
{
    //ifstream cin("f.txt");
    int n, m, tom, prob, firs, i, a, b, c, j, eg, ma=1000001;
    cin >> n >> m >> tom >> prob >> eg;
    tom--;
    eg--;
    vector<vector<el>> vec(n);
    int tav1[n];
    fill(tav1, tav1+n, ma);
    for(i=0; i<m; i++)
    {
        cin >> a >> b >> c;
        a--;b--;
        ur.csucs=a;
        ur.szell = c;
        vec[b].push_back(ur);
        ur.csucs=b;
        vec[a].push_back(ur);

    }
    queue<int> q;
    q.push(tom);
    tav1[tom]=0;
    while(!q.empty())
    {
        a = q.front();
        q.pop();
        for(i=0; i<vec[a].size(); i++)
        {

            if(vec[a][i].szell==2 && tav1[vec[a][i].csucs]==ma)
            {
                tav1[vec[a][i].csucs]=tav1[a]+1;
                q.push(vec[a][i].csucs);
            }
        }
    }
    priority_queue<pair<int, int>>que;
    que.push({tav1[eg], eg});
    int szok[n];
    fill(szok, szok+n, 0);
    szok[eg] = tav1[eg];
    while(!que.empty())
    {
        a=que.top().second;
        b = que.top().first;
        que.pop();
        if(b>0)
        {
            szok[a]=1;
            for(auto x:vec[a])if(!szok[x.csucs])que.push({min(b-1, tav1[x.csucs]), x.csucs});
        }
    }
    for(i=0; i<prob; i++)
    {
        cin >> a;
        a--;
        if(szok[a])cout << "IGEN\n";
        else cout << "NEM\n";
    }
    return 0;
}
SubtaskSumTestVerdictTimeMemory
base80/80
1Accepted0/01ms316 KiB
2Accepted0/03ms316 KiB
3Accepted4/41ms512 KiB
4Accepted4/41ms316 KiB
5Accepted4/41ms316 KiB
6Accepted4/41ms316 KiB
7Accepted4/41ms316 KiB
8Accepted4/42ms316 KiB
9Accepted4/44ms316 KiB
10Accepted4/43ms316 KiB
11Accepted4/417ms1004 KiB
12Accepted4/412ms1076 KiB
13Accepted4/430ms1620 KiB
14Accepted4/470ms2740 KiB
15Accepted4/4209ms6832 KiB
16Accepted4/467ms3932 KiB
17Accepted4/4103ms4488 KiB
18Accepted4/4100ms4272 KiB
19Accepted4/487ms3632 KiB
20Accepted4/481ms3636 KiB
21Accepted4/475ms3236 KiB
22Accepted4/4179ms6336 KiB