結果

問題 No.3396 ChRisTmas memory
コンテスト
ユーザー KumaTachiRen
提出日時 2025-11-12 21:07:12
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,011 ms / 4,000 ms
コード長 1,336 bytes
コンパイル時間 765 ms
コンパイル使用メモリ 90,832 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-12-02 23:31:50
合計ジャッジ時間 12,229 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <numeric>
using namespace std;
using ll = long long;

pair<ll, ll> inv_mod(ll x, ll m) {
  ll a = m, b = x % m;
  ll u = 0, v = 1;
  while (b > 0) {
    ll q = a / b;
    swap(a -= q * b, b);
    swap(u -= q * v, v);
  }
  if (u < 0) u += m / a;
  return {a, u};
}

int main() {
  int q;
  cin >> q;
  vector<int> M, Y;
  while (q--) {
    int t;
    cin >> t;
    if (t == 1) {
      int m, r;
      cin >> m >> r;
      if (!Y.empty() && Y.back() < 0)
        M.push_back(m), Y.push_back(-1);
      else {
        ll p = 1, y = r;
        for (int i = 0; i < (int)Y.size(); i++) {
          y -= Y[i] * p % m;
          if (y < 0) y += m;
          p = p * M[i] % m;
        }
        auto [g, x] = inv_mod(p, m);
        if (y % g != 0) {
          M.push_back(1), Y.push_back(-1);
        } else {
          m /= g;
          y = m == 1 ? 0 : (y / g) * x % m;
          M.push_back(m), Y.push_back(y);
        }
      }
    } else if (t == 2) {
      int k;
      cin >> k;
      while (k--) M.pop_back(), Y.pop_back();
    } else {
      int m;
      cin >> m;
      if (!Y.empty() && Y.back() < 0)
        cout << -1 << "\n";
      else {
        ll x = 0;
        for (int i = (int)Y.size() - 1; i >= 0; i--) x = (x * M[i] + Y[i]) % m;
        cout << x << "\n";
      }
    }
  }
}
0