126142024-12-26 21:16:15BucsMateKerékpártúra (50 pont)cpp17Runtime error 24/50130ms3640 KiB
#include <iostream>
#include <vector>
#include <queue>

using namespace std;

void BFS_rev(bool canReturn[], vector<vector<int>> &adj_rev, int start)
{
    queue<int> q;
    q.push(start);
    canReturn[start] = true;

    while(!q.empty()){
        int curr = q.front();
        q.pop();
        for(int i = 0; i < adj_rev[curr].size(); i++){
            int newNode = adj_rev[curr][i];
            if(!canReturn[newNode]){
                q.push(newNode);
                canReturn[newNode] = true;
            }
        }
    }
}

void BFS_main(bool canReturn[], bool valid[], vector<vector<int>> &adj, int start)
{
    queue<int> q;
    q.push(start);
    valid[start] = true;

    while(!q.empty()){
        int curr = q.front();
        q.pop();
        if(!canReturn[curr]){
            continue;
        }
        for(int i = 0; i < adj[curr].size(); i++){
            int newNode = adj[curr][i];
            if(!valid[newNode]){
                valid[newNode] = true;
                q.push(newNode);
            }
        }
    }
}

int main()
{
    int N, M, K;
    cin >> N >> M >> K;
    vector<vector<int>> adj(N+1, vector<int>()), adj_rev(N+1, vector<int>());
    for(int i = 0; i < M; i++){
        int a, b;
        cin >> a >> b;
        adj[a].push_back(b);
        adj_rev[b].push_back(a);
    }
    bool canReturn[1001] = {};
    BFS_rev(canReturn, adj_rev, K);

    bool valid[1001] = {};
    BFS_main(canReturn, valid, adj, K);

    vector<int> sol;

    for(int i = 1; i <= N; i++){
        if(valid[i] && i != K){
            sol.push_back(i);
        }
    }

    cout << sol.size() << endl;
    for(int i = 0; i < sol.size(); i++){
        cout << sol[i] << " ";
    }
    cout << endl;
    return 0;
}
SubtaskSumTestVerdictTimeMemory
base24/50
1Accepted0/01ms320 KiB
2Runtime error0/019ms1560 KiB
3Accepted2/21ms320 KiB
4Accepted2/21ms508 KiB
5Accepted2/21ms320 KiB
6Accepted2/21ms320 KiB
7Accepted2/21ms320 KiB
8Accepted2/22ms320 KiB
9Accepted2/23ms320 KiB
10Accepted2/24ms456 KiB
11Accepted2/24ms320 KiB
12Accepted2/212ms584 KiB
13Accepted2/210ms712 KiB
14Accepted2/220ms948 KiB
15Runtime error0/332ms1592 KiB
16Runtime error0/437ms1772 KiB
17Runtime error0/450ms2088 KiB
18Runtime error0/345ms1848 KiB
19Runtime error0/339ms1844 KiB
20Runtime error0/3115ms3128 KiB
21Runtime error0/3130ms3384 KiB
22Runtime error0/3130ms3640 KiB