#499041 (C++14) No.1084 積の積

提出ソース

結果

問題	No.1084 積の積
ユーザー	ミドリムシ
提出日時	2020-06-19 21:51:10
言語	C++14 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	243 ms / 2,000 ms
コード長	5,517 bytes
コンパイル時間	1,642 ms
コンパイル使用メモリ	174,820 KB
実行使用メモリ	9,736 KB
最終ジャッジ日時	2023-09-16 13:58:39
合計ジャッジ時間	6,483 ms
ジャッジサーバーID （参考情報）	judge15 / judge12

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 4,376 KB
testcase_01	AC	2 ms 4,376 KB
testcase_02	AC	1 ms 4,380 KB
testcase_03	AC	1 ms 4,376 KB
testcase_04	AC	96 ms 9,592 KB
testcase_05	AC	15 ms 4,376 KB
testcase_06	AC	98 ms 9,512 KB
testcase_07	AC	1 ms 4,376 KB
testcase_08	AC	1 ms 4,380 KB
testcase_09	AC	34 ms 4,380 KB
testcase_10	AC	1 ms 4,376 KB
testcase_11	AC	9 ms 4,376 KB
testcase_12	AC	94 ms 6,228 KB
testcase_13	AC	200 ms 9,644 KB
testcase_14	AC	68 ms 4,572 KB
testcase_15	AC	65 ms 4,580 KB
testcase_16	AC	243 ms 9,564 KB
testcase_17	AC	86 ms 6,168 KB
testcase_18	AC	149 ms 6,548 KB
testcase_19	AC	203 ms 9,600 KB
testcase_20	AC	44 ms 4,576 KB
testcase_21	AC	68 ms 4,644 KB
testcase_22	AC	54 ms 4,640 KB
testcase_23	AC	117 ms 6,168 KB
testcase_24	AC	106 ms 6,424 KB
testcase_25	AC	202 ms 9,680 KB
testcase_26	AC	215 ms 9,664 KB
testcase_27	AC	215 ms 9,644 KB
testcase_28	AC	214 ms 9,560 KB
testcase_29	AC	217 ms 9,736 KB
testcase_30	AC	217 ms 9,656 KB
testcase_31	AC	217 ms 9,568 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
using lint = long long;
constexpr lint mod = 1e9 + 7;
#define all(x) (x).begin(), (x).end()
#define bitcount(n) __builtin_popcountl((lint)(n))
#define fcout cout << fixed << setprecision(15)
#define highest(x) (63 - __builtin_clzl(x))
#define rep(i, n) for(int i = 0; i < int(n); i++)
#define rep2(i, l, r) for(int i = int(l); i < int(r); i++)
#define repr(i, n) for(int i = int(n) - 1; i >= 0; i--)
#define repr2(i, l, r) for(int i = int(r) - 1; i >= int(l); i--)
#define SZ(x) int(x.size())
constexpr int inf9 = 1e9; constexpr lint inf18 = 1e18;
inline void YES(bool condition){ if(condition) cout << "YES" << endl; else cout << "NO" << endl; }
inline void Yes(bool condition){ if(condition) cout << "Yes" << endl; else cout << "No" << endl; }
inline void assertNO(bool condition){ if(!condition){ cout << "NO" << endl; exit(0); } }
inline void assertNo(bool condition){ if(!condition){ cout << "No" << endl; exit(0); } }
inline void assertm1(bool condition){ if(!condition){ cout << -1 << endl; exit(0); } }
lint power(lint base, lint exponent, lint module){ if(exponent % 2){ return power(base, exponent - 1, module) * base % module; }else if(exponent){ lint root_ans = power(base, exponent / 2, module); return root_ans * root_ans % module; }else{ return 1; }}
struct position{ int y, x; }; position mv[4] = {{0, -1}, {1, 0}, {0, 1}, {-1, 0}}; double euclidean(position first, position second){ return sqrt((second.x - first.x) * (second.x - first.x) + (second.y - first.y) * (second.y - first.y)); }
template<class T, class U> string to_string(pair<T, U> x){ return to_string(x.first) + "," + to_string(x.second); } string to_string(string x){ return x; }
template<class T> string to_string(complex<T> x){ return to_string(make_pair(x.real(), x.imag())); }
template<class itr> void array_output(itr start, itr goal){ string ans; for(auto i = start; i != goal; i++) cout << (i == start ? "" : " ") << (*i); if(!ans.empty()) ans.pop_back(); cout << ans << endl; }
template<class itr> void cins(itr first, itr last){ for(auto i = first; i != last; i++){ cin >> (*i); } }
template<class T> T gcd(T a, T b){ if(b) return gcd(b, a % b); else return a; }
template<class T> T lcm(T a, T b){ return a / gcd(a, b) * b; }
struct combination{ vector<lint> fact, inv; combination(int sz) : fact(sz + 1), inv(sz + 1){ fact[0] = 1; for(int i = 1; i <= sz; i++){ fact[i] = fact[i - 1] * i % mod; } inv[sz] = power(fact[sz], mod - 2, mod); for(int i = sz - 1; i >= 0; i--){ inv[i] = inv[i + 1] * (i + 1) % mod; } } lint P(int n, int r){ if(r < 0 || n < r) return 0; return (fact[n] * inv[n - r] % mod); } lint C(int p, int q){ if(q < 0 || p < q) return 0; return (fact[p] * inv[q] % mod * inv[p - q] % mod); } };
template<class itr> bool next_sequence(itr first, itr last, int max_bound){ itr now = last; while(now != first){ now--; (*now)++; if((*now) == max_bound){ (*now) = 0; }else{ return true; } } return false; }
template<class itr, class itr2> bool next_sequence2(itr first, itr last, itr2 first2, itr2 last2){ itr now = last; itr2 now2 = last2; while(now != first){ now--, now2--; (*now)++; if((*now) == (*now2)){ (*now) = 0; }else{ return true; } } return false; }
template<class T> bool chmax(T &a, const T &b){ if(a < b){ a = b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b){ if(b < a){ a = b; return 1; } return 0; }
inline int at(lint i, int j){ return (i >> j) & 1; }
random_device rnd;
bool is_in_board(lint y, lint x, lint H, lint W){ return (0 <= y && y < H && 0 <= x && x < W); }

template< typename Monoid >
struct SegmentTree
{
    using F = function< Monoid(Monoid, Monoid) >;
    
    int sz;
    vector< Monoid > seg;
    
    const F f;
    const Monoid M1;
    
    SegmentTree(int n, const F f, const Monoid &M1) : f(f), M1(M1)
    {
        sz = 1;
        while(sz < n) sz <<= 1;
        seg.assign(2 * sz, M1);
    }
    
    void set(int k, const Monoid &x)
    {
        seg[k + sz] = x;
    }
    
    void build()
    {
        for(int k = sz - 1; k > 0; k--) {
            seg[k] = f(seg[2 * k + 0], seg[2 * k + 1]);
        }
    }
    
    void update(int k, const Monoid &x)
    {
        k += sz;
        seg[k] = x;
        while(k >>= 1) {
            seg[k] = f(seg[2 * k + 0], seg[2 * k + 1]);
        }
    }
    
    Monoid query(int a, int b)
    {
        Monoid L = M1, R = M1;
        for(a += sz, b += sz; a < b; a >>= 1, b >>= 1) {
            if(a & 1) L = f(L, seg[a++]);
            if(b & 1) R = f(seg[--b], R);
        }
        return f(L, R);
    }
    
    Monoid operator[](const int &k) const
    {
        return seg[k + sz];
    }
};

int main(){
    int N;
    cin >> N;
    vector<int> A(N);
    cins(all(A));
    if(count(all(A), 0)){
        cout << 0 << endl;
        return 0;
    }
    SegmentTree< lint > seg(N, [](lint a, lint b){ return a * b; }, 1);
    SegmentTree< lint > seg_mod(N, [](lint a, lint b){ return a * b % mod; }, 1);
    SegmentTree< lint > seg_stairly(N, [](lint a, lint b){ return a * b % mod; }, 1);
    rep(i, N){
        seg.set(i, A[i]);
        seg_mod.set(i, A[i]);
        seg_stairly.set(i, power(A[i], N - i, mod));
    }
    seg.build();
    seg_mod.build();
    seg_stairly.build();
    int r = 0;
    lint ans = 1;
    rep(l, N){
        while(r < N && seg.query(l, r + 1) < inf9){
            r++;
        }
        ans = ans * seg_stairly.query(l, r) % mod * power(seg_mod.query(l, r), (N - r) * (mod - 2), mod) % mod;
    }
    cout << ans << endl;
}

yukicoder

結果

テストケース

ソースコード