結果

問題 No.1477 Lamps on Graph
ユーザー monnu
提出日時 2021-04-16 20:41:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 268 ms / 2,000 ms
コード長 1,018 bytes
コンパイル時間 1,747 ms
コンパイル使用メモリ 177,876 KB
実行使用メモリ 11,412 KB
最終ジャッジ日時 2024-07-02 23:10:26
合計ジャッジ時間 7,879 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
//#include <atcoder/all>
//using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MAX 200003
#define MOD 998244353
#define INF 1000000000

int main(){
  int N,M;
  cin>>N>>M;
  vector<int> A(N);
  vector<pair<int,int>> order(N);
  for(int i=0;i<N;i++){
    cin>>A[i];
    order[i].first=A[i];
    order[i].second=i;
  }
  sort(order.begin(),order.end());
  Graph G(N);
  for(int i=0;i<M;i++){
    int u,v;
    cin>>u>>v;
    u--;v--;
    G[u].push_back(v);
    G[v].push_back(u);
  }
  int K;
  cin>>K;
  vector<int> light(N,0);
  for(int i=0;i<K;i++){
    int B;
    cin>>B;
    B--;
    light[B]=1;
  }

  vector<int> ans;
  for(int i=0;i<N;i++){
    int v=order[i].second;
    if(light[v]==1){
      ans.push_back(v);
      light[v]=0;
      for(auto nv:G[v]){
        if(A[nv]>A[v]){
          light[nv]=1-light[nv];
        }
      }
    }
  }

  cout<<ans.size()<<endl;
  for(int i=0;i<ans.size();i++){
    cout<<ans[i]+1<<endl;
  }
}
0