結果

問題 No.1477 Lamps on Graph
コンテスト
ユーザー se1ka2
提出日時 2021-04-16 20:16:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,428 bytes
コンパイル時間 814 ms
コンパイル使用メモリ 80,836 KB
実行使用メモリ 13,136 KB
最終ジャッジ日時 2024-07-02 22:14:53
合計ジャッジ時間 8,981 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 12 WA * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <queue>
using namespace std;

struct Graph
{
    int n;
    std::vector<std::vector<int>> g;
    
    Graph(){}
    
    Graph(int n) : n(n){
        g.resize(n);
    }
    
    void add_edge(int from, int to){
        g[from].push_back(to);
    }
};

int main()
{
    int n, m;
    cin >> n >> m;
    int a[100005];
    for(int i = 0; i < n; i++) cin >> a[i];
    Graph g(n), r(n);
    for(int i = 0; i < m; i++){
        int u, v;
        cin >> u >> v;
        u--; v--;
        if(a[u] < a[v]){
            g.add_edge(u, v);
            r.add_edge(v, u);
        }
        if(a[u] > a[v]){
            g.add_edge(v, u);
            r.add_edge(u, v);
        }
    }
    int k;
    cin >> k;
    bool b[100005]{0};
    for(int i = 0; i < k; i++){
        int u;
        cin >> u;
        u--;
        b[u] = true;
    }
    bool c[100005]{0};
    queue<int> que;
    for(int u = 0; u < n; u++){
        if(r.g[u].size() == 0){
            c[u] = true;
            que.push(u);
        }
    }
    vector<int> ans;
    while(que.size()){
        int u = que.front();
        que.pop();
        if(b[u]) ans.push_back(u);
        for(int v : g.g[u]){
            if(b[u]) b[v] = !b[v];
            if(!c[v]){
                c[v] = true;
                que.push(v);
            }
        }
    }
    cout << (int)ans.size() << endl;
    for(int u : ans) cout << u + 1 << endl;
    cout << endl;
}
0