結果

問題 No.1084 積の積
ユーザー saxofone111saxofone111
提出日時 2021-03-14 12:36:44
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 140 ms / 2,000 ms
コード長 4,708 bytes
コンパイル時間 2,220 ms
コンパイル使用メモリ 207,928 KB
実行使用メモリ 7,296 KB
最終ジャッジ日時 2024-11-06 02:25:08
合計ジャッジ時間 6,032 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,820 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 71 ms
7,168 KB
testcase_05 AC 33 ms
7,296 KB
testcase_06 AC 36 ms
7,168 KB
testcase_07 AC 2 ms
6,816 KB
testcase_08 AC 2 ms
6,820 KB
testcase_09 AC 108 ms
6,820 KB
testcase_10 AC 2 ms
6,820 KB
testcase_11 AC 7 ms
6,816 KB
testcase_12 AC 59 ms
6,816 KB
testcase_13 AC 123 ms
6,820 KB
testcase_14 AC 46 ms
6,816 KB
testcase_15 AC 43 ms
6,820 KB
testcase_16 AC 140 ms
7,168 KB
testcase_17 AC 54 ms
6,820 KB
testcase_18 AC 90 ms
6,816 KB
testcase_19 AC 124 ms
6,816 KB
testcase_20 AC 29 ms
6,816 KB
testcase_21 AC 45 ms
6,816 KB
testcase_22 AC 35 ms
6,816 KB
testcase_23 AC 75 ms
6,816 KB
testcase_24 AC 67 ms
6,820 KB
testcase_25 AC 121 ms
6,820 KB
testcase_26 AC 103 ms
7,168 KB
testcase_27 AC 103 ms
7,168 KB
testcase_28 AC 104 ms
7,168 KB
testcase_29 AC 106 ms
7,296 KB
testcase_30 AC 105 ms
7,168 KB
testcase_31 AC 106 ms
7,168 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

#define MOD 1000000007
#define rep(i, n) for(ll i=0; i < (n); i++)
#define rrep(i, n) for(ll i=(n)-1; i >=0; i--)
#define ALL(v) v.begin(),v.end()
#define rALL(v) v.rbegin(),v.rend()
#define FOR(i, j, k) for(ll i=j;i<k;i++)
#define debug_print(var) cerr << #var << "=" << var <<endl;
#define DUMP(i, v)for(ll i=0;i<v.size();i++)cerr<<v[i]<<" "
#define fi first
#define se second

using namespace std;
typedef long long int ll;
typedef vector<ll> llvec;
typedef vector<double> dvec;
typedef pair<ll, ll> P;
typedef long double ld;
struct edge{ll x, c;};

ll mod(ll a, ll mod){
  ll res = a%mod;
  if(res<0)res=res + mod;
  return res;
}

ll modpow(ll a, ll n, ll mod){
  ll res=1;
  while(n>0){
    if(n&1) res=res*a%mod;
    a=a*a%mod;
    n>>=1;
  }
  return res;
}

ll modinv(ll a, ll mod){
  ll b=mod, u=1, v=0;
  while(b){
    ll t=a/b;
    a-=t*b; swap(a, b);
    u-=t*v; swap(u, v);
  }
  u%=mod;
  if(u<0)u+=mod;
  return u;
}

ll gcd(ll a, ll b){
  ll r = a%b;
  if(r==0) return b;
  else return gcd(b, a%b);
}

// @param b 1
// @return pair(g, x) s.t. g = gcd(a, b), xa = g (mod b), 0 <= x < b/g
std::pair<long long, long long> inv_gcd(long long a, long long b) {
  a = mod(a, b);
  if (a == 0) return {b, 0};

  // Contracts:
  // [1] s - m0 * a = 0 (mod b)
  // [2] t - m1 * a = 0 (mod b)
  // [3] s * |m1| + t * |m0| <= b
  long long s = b, t = a;
  long long m0 = 0, m1 = 1;

  while (t) {
    long long u = s / t;
    s -= t * u;
    m0 -= m1 * u;  // |m1 * u| <= |m1| * s <= b

    // [3]:
    // (s - t * u) * |m1| + t * |m0 - m1 * u|
    // <= s * |m1| - t * u * |m1| + t * (|m0| + |m1| * u)
    // = s * |m1| + t * |m0| <= b

    auto tmp = s;
    s = t;
    t = tmp;
    tmp = m0;
    m0 = m1;
    m1 = tmp;
  }
  // by [3]: |m0| <= b/g
  // by g != b: |m0| < b/g
  if (m0 < 0) m0 += b / s;
  return {s, m0};
}

// (rem, mod)
std::pair<long long, long long> crt(const std::vector<long long>& r,
                                    const std::vector<long long>& m) {
  assert(r.size() == m.size());
  int n = int(r.size());
  // Contracts: 0 <= r0 < m0
  long long r0 = 0, m0 = 1;
  for (int i = 0; i < n; i++) {
    assert(1 <= m[i]);
    long long r1 = mod(r[i], m[i]), m1 = m[i];
    if (m0 < m1) {
      std::swap(r0, r1);
      std::swap(m0, m1);
    }
    if (m0 % m1 == 0) {
      if (r0 % m1 != r1) return {0, 0};
      continue;
    }
    // assume: m0 > m1, lcm(m0, m1) >= 2 * max(m0, m1)

    // (r0, m0), (r1, m1) -> (r2, m2 = lcm(m0, m1));
    // r2 % m0 = r0
    // r2 % m1 = r1
    // -> (r0 + x*m0) % m1 = r1
    // -> x*u0*g % (u1*g) = (r1 - r0) (u0*g = m0, u1*g = m1)
    // -> x = (r1 - r0) / g * inv(u0) (mod u1)

    // im = inv(u0) (mod u1) (0 <= im < u1)
    long long g, im;
    std::tie(g, im) = inv_gcd(m0, m1);

    long long u1 = (m1 / g);
    // |r1 - r0| < (m0 + m1) <= lcm(m0, m1)
    if ((r1 - r0) % g) return {0, 0};

    // u1 * u1 <= m1 * m1 / g / g <= m0 * m1 / g = lcm(m0, m1)
    long long x = (r1 - r0) / g % u1 * im % u1;

    // |r0| + |m0 * x|
    // < m0 + m0 * (u1 - 1)
    // = m0 + m0 * m1 / g - m0
    // = lcm(m0, m1)
    r0 += x * m0;
    m0 *= u1;  // -> lcm(m0, m1)
    if (r0 < 0) r0 += m0;
  }
  return {r0, m0};
}


bool is_prime(ll n){
  ll i = 2;
  if(n==1)return false;
  if(n==2)return true;
  bool res = true;
  while(i*i <n){
    if(n%i==0){
      res = false;
    }
    i = i+1;
  }

  //if(i==1)res = false;
  if(n%i==0)res=false;
  return res;
}

#define fracMax  10000000
ll frac[fracMax];
ll ifrac[fracMax];

void gen(){
  frac[0] = 1;
  ifrac[0] = 1;
  rep(i, fracMax-1){
    frac[i+1] = mod((i+1)*frac[i], MOD);
    ifrac[i+1] = modinv(frac[i+1], MOD);   
  }
  return;
}

ll binom(ll n, ll k){
  if(k<0 or k>n){
    return 0;
  }else{
    return mod(mod(frac[n]*ifrac[n-k], MOD)*ifrac[k], MOD);
  }
}


/**************************************
** A main function starts from here  **
***************************************/
int main(){
  ll N;
  cin >> N;
  llvec A(N);
  vector<ll> S(N+1, 1);
  vector<ll> Sinv(N+1, 1);
  vector<ll> T(N+1, 1);
  vector<ll> Tinv(N+1, 1);
  rep(i, N){
    cin >> A[i];
    S[i+1] = mod(S[i]*A[i], MOD);
    Sinv[i+1] = modinv(S[i+1], MOD);
    T[i+1] = mod(T[i]*S[i+1], MOD);
    Tinv[i+1] = modinv(T[i+1], MOD);
  }

  //T[j][k] = T[k]*Tinv[j]*modpow(S[j], j);
  ll tmp = 1;
  ll ans = 1;
  for(ll l = 0, r=0;l<N;tmp = mod(tmp*modinv(A[l++], MOD), MOD)){
    while(r<N and tmp<1e9/A[r]){
      tmp *=A[r];
      r++;
    }
    //a_l ^(r-l) * a_(l+1)^(r-l-1)* ... * a_(r-1)^1
    ll s = mod(mod(T[r]*Tinv[l],MOD)*modpow(Sinv[l], r-l, MOD), MOD);
    //cerr << l << " " << r <<" " << s << endl;
    ans = mod(ans*s, MOD );
  }
  cout << ans << endl;
  return 0;
}
0