166632025-05-07 19:44:21sztomiFertőzési sorozat (50 pont)cpp17Wrong answer 17/5012ms512 KiB
#include <bits/stdc++.h>

using namespace std;

int n;
vector<int> felosztas; // bad practice
vector<vector<int>> graf;
vector<int> volt;
vector<int> sor;
vector<int> kell;
int akt_volt = 1;

bool jo_felosztas(){
    int elozo = felosztas[sor[0]];
    int akt_max = felosztas[sor[0]];
    bool kezdo = true;
    for(int i = 0; i < sor.size(); i++){
        if(felosztas[sor[i]] < akt_max || felosztas[sor[i]]-akt_max > 1){
            return false;
        }
        // ugy valt, hogy meg nem vegzett minden elotte levovel
        if(felosztas[sor[i]]-akt_max == 1){
            kezdo = false;
            elozo = akt_max;
        }
        if((kell[elozo] && !kezdo)){
            return false;
        }


        akt_max = max(felosztas[sor[i]], akt_max);
        kell[akt_max]--;
    }
    return true;
}

void bfs(int kezd){
    kell.assign(n, 0);
    queue<int> q;
    q.push(kezd);
    volt[kezd] = akt_volt;
    felosztas[kezd] = 0;
    kell[0] = 1;
    int akt;
    while(!q.empty()){
        akt = q.front();
        //cout << "akt: " <<akt << " lepes: " << lepes << "\n";
        q.pop();

        for(auto x : graf[akt]){
            if(volt[x] == akt_volt){
                continue;
            }
            felosztas[x] = felosztas[akt] + 1;
            kell[felosztas[x]]++;
            volt[x] = akt_volt;
            q.push(x);
        }
    }

    int sz = 0;
    for(int i = 0; i < n; i++){
        if(kell[i] == 0) break;
        sz += kell[i];
        kell[i] = sz;
    }


    akt_volt++;
}


int main()
{
    int m, k;
    cin >> n >> m >> k;
    sor.resize(k);
    for(auto &x : sor){
        cin >> x;
        x--;
    }

    graf.assign(n, vector<int>());

    volt.assign(n, 0);
    int a, b;
    for(int i = 0; i < m; i++){
        cin >> a >> b;
        a--;
        b--;
        graf[a].push_back(b);
        graf[b].push_back(a);
    }


    vector<int> jok;
    felosztas.assign(n, -1);
    for(int i = 0; i < n; i++){
        bfs(i);
        /*
        for(int j = 0; j < n; j++){
            cout << felosztas[j] << " ";
        }
        cout << "\n";
        */
        if(jo_felosztas()){
            //cout << "jo felosztas\n";
            jok.push_back(i+1);
        }
    }

    cout << jok.size() << "\n";
    for(auto x : jok){
        cout << x << " ";
    }
    cout << "\n";

}
SubtaskSumTestVerdictTimeMemory
base17/50
1Wrong answer0/01ms316 KiB
2Wrong answer0/01ms316 KiB
3Wrong answer0/04ms316 KiB
4Wrong answer0/21ms316 KiB
5Wrong answer0/21ms316 KiB
6Wrong answer0/23ms316 KiB
7Wrong answer0/23ms316 KiB
8Wrong answer0/24ms508 KiB
9Wrong answer0/24ms316 KiB
10Wrong answer0/212ms440 KiB
11Accepted1/11ms316 KiB
12Accepted2/24ms508 KiB
13Accepted2/24ms432 KiB
14Accepted2/24ms424 KiB
15Accepted2/24ms424 KiB
16Accepted2/24ms316 KiB
17Wrong answer0/24ms320 KiB
18Wrong answer0/14ms388 KiB
19Wrong answer0/14ms316 KiB
20Accepted1/14ms508 KiB
21Wrong answer0/112ms316 KiB
22Accepted1/112ms316 KiB
23Wrong answer0/19ms440 KiB
24Wrong answer0/18ms440 KiB
25Accepted1/18ms444 KiB
26Wrong answer0/19ms436 KiB
27Wrong answer0/112ms440 KiB
28Wrong answer0/19ms440 KiB
29Accepted1/110ms316 KiB
30Wrong answer0/18ms512 KiB
31Wrong answer0/18ms316 KiB
32Wrong answer0/110ms500 KiB
33Wrong answer0/112ms316 KiB
34Wrong answer0/112ms508 KiB
35Wrong answer0/112ms316 KiB
36Wrong answer0/112ms316 KiB
37Wrong answer0/112ms316 KiB
38Accepted1/112ms440 KiB
39Accepted1/110ms316 KiB
40Wrong answer0/112ms436 KiB