Submission #65790446

---

Source Code Expand

// 綺麗に解きなおし

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

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

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

  int n, m;
  cin >> n >> m;

  // サイズが不揃いな二次元配列を受け取る
  vector<vector<int>> a(m);
  for (int i=0; i<m; i++) {
    int num;
    cin >> num;
    a.at(i).resize(num);
    for (int j=0; j<num; j++) {
      cin >> a.at(i).at(j);
    }
  }

  // Bの方は、逆写像で受け取る
  // すなわち、各食材が何日目に食べられるようになるかを入れる
  // 食材0はデータが来ないが、あって害になるわけでもないので適当に-1で埋めておく
  // また、食材nを受け取れるよう、sizeはn+1としておく。
  vector<int> b_inverse(n+1,-1);
  for (int i=0; i<n; i++) {
    int tmp;
    cin >> tmp;
    b_inverse.at(tmp) = i;
  }

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

  // 一旦「その日に食べられるようになる料理数」を格納
  // 最後に累積和を取って「その日【までに】食べられるようになる料理数」にして出力する
  vector<int> result(n,0);

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

  // 各料理について、何日目に食べられるようになるかを計算し、resultに反映
  for (const vector<int>& dish : a) {
    int day = -1;
    for (int ingredient : dish) {
      day = max(day,b_inverse.at(ingredient));
    }
    result.at(day)++;
  }

  // 予定通り累積和を取る
  for (int i=0; i<n-1; i++) {
    result.at(i+1) += result.at(i);
  }

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

  for (size_t i=0; i<result.size(); i++) {
    cout << result.at(i) << endl;
  }

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

  return 0;

}

Submission Info

Judge Result