結果

問題 No.2994 べき内積
コンテスト
ユーザー 2qbingxuan
提出日時 2024-12-22 04:14:42
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,810 bytes
コンパイル時間 3,034 ms
コンパイル使用メモリ 248,356 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-22 04:14:48
合計ジャッジ時間 5,133 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define all(x) begin(x), end(x)
#ifdef CKISEKI
#define safe cerr << __PRETTY_FUNCTION__ << " line " << __LINE__ << " safe\n"
#define debug(a...) debug_(#a, a)
#define orange(a...) orange_(#a, a)
#include <experimental/iterator>
void debug_(auto s, auto ...a) {
  cerr << "\e[1;32m(" << s << ") = (";
  int f = 0;
  (..., (cerr << (f++ ? ", " : "") << a));
  cerr << ")\e[0m\n";
}
void orange_(auto s, auto L, auto R) {
  cerr << "\e[1;33m[ " << s << " ] = [ ";
  using namespace experimental;
  copy(L, R, make_ostream_joiner(cerr, ", "));
  cerr << " ]\e[0m\n";
}
#else
#define safe ((void)0)
#define debug(...) safe
#define orange(...) safe
#endif

using lld = int64_t;

int main() {
  constexpr int p = 1009;
  constexpr int p2 = p * p;
  constexpr int mod = p * (p - 1);

  cin.tie(nullptr)->sync_with_stdio(false);
  int M, N;
  cin >> M >> N;

  const lld inf = 1e15;
  lld K_capped = 0, K_mod = 0;

  {
    vector<int> k(M + 1);
    for (int i = 0; i <= M; i++) {
      cin >> k[i];
    }
    for (int i = M; i >= 0; i--) {
      K_capped = min(inf, K_capped * p + k[i]);
      K_mod = (K_mod * p + k[i]) % mod;
    }
  }


  vector<int> e(N + 1), r(N + 1);
  r[0] = 1;

  for (int i = 0; i <= N; i++)
    cin >> e[i];

  auto mul = [&](const auto &a, const auto &b) {
    vector<int> c(N + 1);
    for (int i = 0; i <= N; i++) {
      for (int j = 0; i + j <= N; j++) {
        c[i + j] += a[i] * b[j];
        if (c[i + j] >= p2) c[i + j] -= p2;
      }
    }
    for (int i = 0; i <= N; i++)
      c[i] %= p;
    return c;
  };

  // lld K = K_mod + (K_capped - K_mod) / mod * mod;
  lld K = K_mod;
  while (K) {
    if (K & 1) r = mul(r, e);
    e = mul(e, e);
    K >>= 1;
  }
  for (int i = 0; i <= N; i++)
    cout << r[i] << (i==N ? '\n' : ' ');
}
0