Submission #69714484

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;

/////////////////// メイン ///////////////////

int main () {
  
  //////////////////// 入力 ////////////////////

  int n;
  cin >> n;

  vector<int> a(n);
  for (int i=0; i<n; i++) {
    cin >> a.at(i);
  }

  //////////////// 出力変数定義 ////////////////

  bool result = true;

  // vectorの方はaをそのまま使う

  //////////////////// 処理 ////////////////////

  // -1以外に何が何個あるかカウントしつつ、-1の場所を記録する
  vector<int> count(n+1,0);
  vector<int> blank;
  for (int i=0; i<n; i++) {
    if (a.at(i)==-1) blank.emplace_back(i); 
    else {
      count.at(a.at(i))++;
    }
  }

  // 1つも登場してない数をカウント
  vector<int> absence;
  for (int i=1; i<=n; i++) {
    if (count.at(i)==0) absence.emplace_back(i);
  }

  // -1のところに、未使用の数を順番に入れる（未使用の数の個数は空白の個数以上にあるはず）
  // 入れたら、countを更新
  for (size_t i=0; i<blank.size(); i++) {
    a.at(blank.at(i)) = absence.at(i);
    count.at(absence.at(i))++;
  }

  // カウントが0じゃないところがあったら、falseにする
  for (int i=1; i<=n; i++) {
    if (count.at(i)!=1) result = false;
  }

  //////////////////// 出力 ////////////////////

  if (result) {
    cout << "Yes" << endl;
    for (size_t i=0; i<a.size(); i++) {
      cout << a.at(i);
      if (i!=a.size()-1) {
        cout << " ";
      }
    }
    cout << endl;
  } else {
    cout << "No" << endl;
  }

  //////////////////// 終了 ////////////////////

  return 0;

}

Submission Info

Judge Result