提出 #66785833

---

ソースコード 拡げる

// 綺麗に解きなおし

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

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

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

  int n, q;
  cin >> n >> q;

  // vectorを受け取るが、0始まりに直しておく
  vector<int> b(q);
  for (int i=0; i<q; i++) {
    cin >> b.at(i);
    b.at(i)--;
  }

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

  // 結果格納用vector
  vector<int> result(q,0);

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

  // 今それぞれの箱に何個ボールが入っているか管理するvector
  vector<int> balls(n,0);

  // クエリを1つずつ処理
  for (int i=0; i<q; i++) {

    // どの箱に入れるか
    int box = -1;

    // 0以外の場合（0始まりにした処理の都合で-1になっている）
    if (b.at(i)!=-1) {

      // その番号の箱に入れる
      box = b.at(i);

    // 0の場合（0始まりにした処理の都合で-1になっている）
    } else {

      // ボール数の最小値
      // 絶対にこれより大きくならないという値で初期化しておく
      int min_ball = q+1;

      // 最小のものを探す
      for (int j=0; j<n; j++) {
        if (balls.at(j)<min_ball) {
          min_ball = balls.at(j);
          box = j;
        }
      }
      
    }

    // 指定の箱にボールを入れる
    // 結果の方には1始まりに戻して入れる
    balls.at(box)++;
    result.at(i) = box+1;

  }

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

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

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

  return 0;

}

提出情報

ジャッジ結果