結果

問題 No.1084 積の積
コンテスト
ユーザー hipotaf
提出日時 2020-06-19 23:14:46
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,204 bytes
コンパイル時間 729 ms
コンパイル使用メモリ 79,588 KB
実行使用メモリ 13,760 KB
最終ジャッジ日時 2024-07-03 15:36:07
合計ジャッジ時間 5,546 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4 -- * 1
other WA * 1 TLE * 1 -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <cmath>
#include <numeric>

using namespace std;

using ll = long long;

#define REP(i,n) for(ll i=0;i<(ll)(n);i++)
#define REPD(i,n) for(ll i=n-1;i>=0;i--)
#define FOR(i,a,b) for(ll i=a;i<=(ll)(b);i++)
#define FORD(i,a,b) for(ll i=a;i>=(ll)(b);i--)

#define input(...) __VA_ARGS__; in(__VA_ARGS__)

template<class T>
void print(vector<T> a) {
  cout << "[ ";
  REP(i, a.size()) cout << a[i] << " ";
  cout << "]" << endl;
}

void print() {
  std::cout << std::endl;
}

template <class Head, class... Tail>
void print(Head&& head, Tail&&... tail) {
  std::cout << head << " ";
  print(std::forward<Tail>(tail)...);
}

void in() { }

template <class Head, class... Tail>
void in(Head&& head, Tail&&... tail) {
  cin >> head;
  in(std::forward<Tail>(tail)...);
}

ll MOD = 1'000'000'007;
ll powmod(ll v, ll x) {
  if (x == 0) return 1;
  if (x % 2 == 0) {
    ll tmp = powmod(v, x / 2);
    return (tmp * tmp) % MOD;
  } else {
    return (powmod(v, x - 1) * v) % MOD;
  }
}

ll prod(ll left, ll right, vector<ll> &cumprod) {
  // return cumprod[right] / cumprod[left];
  return (cumprod[right] * powmod(cumprod[left], MOD - 2)) % MOD;
}

int main() {
  ll input(n);
  vector<ll> a(n);
  REP(i, n) cin >> a[i];

  vector<ll> cumprod(a.size() + 1, 1);
  partial_sum(a.begin(), a.end(), cumprod.begin() + 1, [](ll v1, ll v2){
      return (v1 * v2) % MOD;
      });
  // print(cumprod);
  // print(a);
  vector<int> one(n);

  ll ones = 0;
  REP(i, n) {
    if (a[n - i - 1] == 1) {
      one[n - i - 1] = ++ones;
    } else {
      ones = 0;
    }
  }
  // print(one);

  ll left = 0;
  ll p = a[0];
  ll ans = 1;
  FOR(right, 1, n) {
    while (p >= 1000000000) {
      p /= a[left++];
    }
    // print("range", left, right, p);

    ll prv = prod(0, right, cumprod);
    // print("---");
    FOR(i, left, right - 1) {
      // print(i);
      if (a[i] == 1) {
        prv = powmod(prv, one[i]);
        i += (one[i] - 1);
      } else prv = prod(i, right, cumprod);
      // prv = prod(i, right, cumprod);
      // print(i, prv);
      (ans *= prv) %= MOD;
    }
    if (right < n) p *= a[right];
  }
  print(ans);
}
0