結果

問題 No.1370 置換門松列
ユーザー SSRSSSRS
提出日時 2021-01-29 22:39:41
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 180 ms / 2,000 ms
コード長 1,256 bytes
コンパイル時間 1,835 ms
コンパイル使用メモリ 175,688 KB
実行使用メモリ 8,576 KB
最終ジャッジ日時 2024-09-14 19:22:09
合計ジャッジ時間 3,884 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int main(){
  int N, M;
  cin >> N >> M;
  vector<int> a(N);
  for (int i = 0; i < N; i++){
    cin >> a[i];
    a[i]--;
  }
  bool ok = true;
  for (int i = 0; i < N - 2; i++){
    if (a[i] == a[i + 2]){
      ok = false;
    }
  }
  if (!ok){
    cout << "No" << endl;
  } else {
    vector<vector<int>> E(M);
    for (int i = 0; i < N - 1; i++){
      if (i % 2 == 0){
        E[a[i]].push_back(a[i + 1]);
      } else {
        E[a[i + 1]].push_back(a[i]);
      }
    }
    vector<int> d(M, 0);
    for (int i = 0; i < M; i++){
      for (int j : E[i]){
        d[j]++;
      }
    }
    queue<int> Q;
    for (int i = 0; i < M; i++){
      if (d[i] == 0){
        Q.push(i);
      }
    }
    int cnt = 0;
    vector<int> ans(M, -1);
    while (!Q.empty()){
      int v = Q.front();
      Q.pop();
      cnt++;
      ans[v] = cnt;
      for (int w : E[v]){
        d[w]--;
        if (d[w] == 0){
          Q.push(w);
        }
      }
    }
    for (int i = 0; i < M; i++){
      if (ans[i] == -1){
        ok = false;
      }
    }
    if (!ok){
      cout << "No" << endl;
    } else {
      cout << "Yes" << endl;
      for (int i = 0; i < M; i++){
        cout << ans[i] << endl;
      }
    }
  }
}
0