#364608 (C++17) No.856 増える演算

提出ソース

結果

問題	No.856 増える演算
ユーザー	0w1
提出日時	2019-08-01 17:46:37
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	TLE
実行時間	-
コード長	2,544 bytes
コンパイル時間	2,287 ms
コンパイル使用メモリ	210,908 KB
実行使用メモリ	17,024 KB
最終ジャッジ日時	2024-07-05 07:43:59
合計ジャッジ時間	10,128 ms
ジャッジサーバーID （参考情報）	judge4 / judge5

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 10,624 KB
testcase_01	AC	1 ms 5,248 KB
testcase_02	AC	1 ms 5,376 KB
testcase_03	AC	1 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	1 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	1 ms 5,376 KB
testcase_13	AC	2 ms 5,376 KB
testcase_14	AC	2 ms 5,376 KB
testcase_15	AC	2 ms 5,376 KB
testcase_16	AC	2 ms 5,376 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	2 ms 5,376 KB
testcase_19	AC	2 ms 5,376 KB
testcase_20	AC	1 ms 5,376 KB
testcase_21	AC	2 ms 5,376 KB
testcase_22	AC	2 ms 5,376 KB
testcase_23	AC	3 ms 5,376 KB
testcase_24	AC	3 ms 5,376 KB
testcase_25	AC	2 ms 5,376 KB
testcase_26	AC	2 ms 5,376 KB
testcase_27	AC	1 ms 5,376 KB
testcase_28	AC	3 ms 5,376 KB
testcase_29	AC	3 ms 5,376 KB
testcase_30	AC	2 ms 5,376 KB
testcase_31	AC	2 ms 5,376 KB
testcase_32	AC	3 ms 5,376 KB
testcase_33	AC	5 ms 5,376 KB
testcase_34	AC	10 ms 5,376 KB
testcase_35	AC	6 ms 5,376 KB
testcase_36	AC	10 ms 5,376 KB
testcase_37	AC	9 ms 6,940 KB
testcase_38	AC	2 ms 6,944 KB
testcase_39	AC	3 ms 6,944 KB
testcase_40	AC	4 ms 6,940 KB
testcase_41	AC	4 ms 6,940 KB
testcase_42	AC	9 ms 6,940 KB
testcase_43	AC	4 ms 6,940 KB
testcase_44	AC	2 ms 6,944 KB
testcase_45	AC	3 ms 6,940 KB
testcase_46	AC	5 ms 6,944 KB
testcase_47	AC	5 ms 6,944 KB
testcase_48	AC	7 ms 6,944 KB
testcase_49	AC	5 ms 6,940 KB
testcase_50	AC	6 ms 6,940 KB
testcase_51	AC	8 ms 6,944 KB
testcase_52	AC	9 ms 6,940 KB
testcase_53	AC	61 ms 16,460 KB
testcase_54	AC	48 ms 15,864 KB
testcase_55	AC	58 ms 16,260 KB
testcase_56	AC	47 ms 16,000 KB
testcase_57	AC	62 ms 16,256 KB
testcase_58	AC	55 ms 16,128 KB
testcase_59	AC	69 ms 16,552 KB
testcase_60	AC	51 ms 16,040 KB
testcase_61	AC	70 ms 16,576 KB
testcase_62	AC	67 ms 16,640 KB
testcase_63	AC	35 ms 15,744 KB
testcase_64	AC	66 ms 16,384 KB
testcase_65	AC	44 ms 16,000 KB
testcase_66	AC	51 ms 16,068 KB
testcase_67	AC	58 ms 16,128 KB
testcase_68	AC	66 ms 16,540 KB
testcase_69	AC	64 ms 16,448 KB
testcase_70	AC	72 ms 16,856 KB
testcase_71	AC	68 ms 16,596 KB
testcase_72	AC	61 ms 16,404 KB
testcase_73	AC	76 ms 16,896 KB
testcase_74	AC	78 ms 17,024 KB
testcase_75	AC	77 ms 16,812 KB
testcase_76	AC	76 ms 16,896 KB
testcase_77	AC	78 ms 16,896 KB
testcase_78	AC	77 ms 16,920 KB
testcase_79	AC	76 ms 16,908 KB
testcase_80	AC	75 ms 16,808 KB
testcase_81	AC	75 ms 16,896 KB
testcase_82	TLE	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

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

typedef complex<double> cpx;
const double PI = acos(-1);
vector<cpx> FFT(vector<cpx> P, bool inv = 0) {
  assert(__builtin_popcount(P.size()) == 1);
  int lg = 31 - __builtin_clz(P.size()), n = 1 << lg;  // == P.size();
  for (int j = 1, i = 0; j < n - 1; ++j) {
    for (int k = n >> 1; k > (i ^= k); k >>= 1)
      ;
    if (j < i) swap(P[i], P[j]);
  }                                                   // bit reverse
  auto w1 = exp((2 - 4 * inv) * PI / n * cpx(0, 1));  // order is 1<<lg
  for (int i = 1; i <= lg; ++i) {
    auto wn = pow(w1, 1 << (lg - i));  // order is 1<<i
    for (int k = 0; k < (1 << lg); k += 1 << i) {
      cpx base = 1;
      for (int j = 0; j < (1 << i - 1); ++j, base = base * wn) {
        auto t = base * P[k + j + (1 << i - 1)];
        auto u = P[k + j];
        P[k + j] = u + t;
        P[k + j + (1 << i - 1)] = u - t;
      }
    }
  }
  if (inv)
    for (int i = 0; i < n; ++i) P[i] /= n;
  return P;
}

const int M = 1e9 + 7;

int ipow(int v, int64_t n, int m = M) {
  int r = 1;
  while (n) {
    if (n & 1) r = 1LL * r * v % m;
    v = 1LL * v * v % m;
    n >>= 1;
  }
  return r;
}

int inv(int v, int m = M) {
  return ipow(v, m - 2, m);
}

int main() {
  ios::sync_with_stdio(false);

  int N;
  cin >> N;

  vector<int> A(N);
  for (int i = 0; i < N; ++i) {
    cin >> A[i];
  }

  pair<int, int> dnr(A[0], A[1]);
  for (int i = N - 2, minp = A[N - 1]; ~i; --i) {
    auto log_eval = [](int x, int y) {
      return log(x + y) + y * log(x);
    };
    if (log_eval(A[i], minp) < log_eval(dnr.first, dnr.second)) {
      dnr = make_pair(A[i], minp);
    }
    minp = min(minp, A[i]);
  }

  vector<int64_t> pa(N + 1);
  for (int i = 0; i < N; ++i) {
    pa[i + 1] = pa[i] + A[i];
  }

  int maxa = *max_element(A.begin(), A.end());
  int maxa2 = maxa * 2 + 1; while (__builtin_popcount(maxa2) != 1) ++maxa2;

  vector<cpx> c(maxa2);
  for (int a : A) c[a] += 1.0;
  vector<cpx> fc = FFT(c);
  for (int i = 0; i < fc.size(); ++i) {
    fc[i] *= fc[i];
  }
  vector<cpx> c2 = FFT(fc, true);

  int ans = 1;

  for (int s = 2; s < maxa * 2 + 1; ++s) {
    int64_t cs = int(real(c2[s]) + 0.5) / 2;
    if (~s & 1) cs -= int(real(c[s / 2]) + 0.5) / 2;
    ans = 1LL * ans * ipow(s, cs) % M;
  }

  for (int i = 0; i + 1 < N; ++i) {
    ans = 1LL * ans * ipow(A[i], pa[N] - pa[i + 1]) % M;
  }

  int idnr = inv(1LL * (dnr.first + dnr.second) % M * ipow(dnr.first, dnr.second) % M);
  ans = 1LL * ans * idnr % M;

  cout << ans << endl;

  return 0;
}

yukicoder

結果

テストケース

ソースコード