結果

問題 No.1084 積の積
ユーザー AEnAEn
提出日時 2023-01-13 23:09:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 482 ms / 2,000 ms
コード長 3,816 bytes
コンパイル時間 233 ms
コンパイル使用メモリ 82,292 KB
実行使用メモリ 95,616 KB
最終ジャッジ日時 2024-06-07 00:01:45
合計ジャッジ時間 8,906 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 34 ms
52,352 KB
testcase_01 AC 35 ms
52,352 KB
testcase_02 AC 33 ms
52,352 KB
testcase_03 AC 33 ms
52,736 KB
testcase_04 AC 403 ms
93,356 KB
testcase_05 AC 49 ms
76,832 KB
testcase_06 AC 393 ms
93,432 KB
testcase_07 AC 35 ms
52,608 KB
testcase_08 AC 33 ms
52,608 KB
testcase_09 AC 55 ms
82,048 KB
testcase_10 AC 33 ms
52,608 KB
testcase_11 AC 96 ms
77,268 KB
testcase_12 AC 194 ms
84,044 KB
testcase_13 AC 321 ms
93,412 KB
testcase_14 AC 159 ms
80,512 KB
testcase_15 AC 157 ms
80,512 KB
testcase_16 AC 353 ms
95,616 KB
testcase_17 AC 179 ms
82,944 KB
testcase_18 AC 245 ms
87,808 KB
testcase_19 AC 318 ms
93,568 KB
testcase_20 AC 138 ms
78,720 KB
testcase_21 AC 163 ms
80,640 KB
testcase_22 AC 145 ms
79,544 KB
testcase_23 AC 217 ms
85,392 KB
testcase_24 AC 199 ms
84,344 KB
testcase_25 AC 324 ms
93,312 KB
testcase_26 AC 459 ms
93,824 KB
testcase_27 AC 475 ms
94,052 KB
testcase_28 AC 475 ms
93,696 KB
testcase_29 AC 479 ms
93,952 KB
testcase_30 AC 482 ms
93,696 KB
testcase_31 AC 458 ms
93,864 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class LazySegmentTree:
    """
    init(init_val, ide_ele): 配列init_valで初期化 O(N)
    add(l, r, x): 区間[l, r)にxを加算 O(logN)
    query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN)
    """
    def __init__(self, init_val, segfunc, ide_ele):
        """
        init_val: 配列の初期値
        segfunc: 区間にしたい操作
        ide_ele: 単位元
        num: n以上の最小の2のべき乗
        data: 値配列(1-index)
        lazy: 遅延配列(1-index)
        """
        if type(init_val) == int:
            n = init_val
        else:
            n = len(init_val)
        self.segfunc = segfunc
        self.ide_ele = ide_ele
        self.num = 1 << (n - 1).bit_length()
        self.data = [ide_ele] * 2 * self.num
        self.lazy = [0] * 2 * self.num
        if type(init_val) == list:
        # 配列の値を葉にセット
            for i in range(n):
                self.data[self.num + i] = init_val[i]
        # 構築していく
            for i in range(self.num - 1, 0, -1):
                self.data[i] = self.segfunc(self.data[2 * i], self.data[2 * i + 1])

    def gindex(self, l, r):
            """
            伝搬する対象の区間を求める
            lm: 伝搬する必要のある最大の左閉区間
            rm: 伝搬する必要のある最大の右開区間
            """
            l += self.num
            r += self.num
            lm = l >> (l & -l).bit_length()
            rm = r >> (r & -r).bit_length()

            while r > l:
                if l <= lm:
                    yield l
                if r <= rm:
                    yield r
                r >>= 1
                l >>= 1
            while l:
                yield l
                l >>= 1

    def propagates(self, *ids):
        """
        遅延伝搬処理
        ids: 伝搬する対象の区間 
        """
        for i in reversed(ids):
            v = self.lazy[i]
            if not v:
                continue
            self.lazy[2 * i] += v
            self.lazy[2 * i + 1] += v
            self.data[2 * i] += v
            self.data[2 * i + 1] += v
            self.lazy[i] = 0

    def add(self, l, r, x):
        """
        区間[l, r)の値にxを加算
        l, r: index(0-index)
        x: additional value
        """
        *ids, = self.gindex(l, r)
        l += self.num
        r += self.num
        while l < r:
            if l & 1:
                self.lazy[l] += x
                self.data[l] += x
                l += 1
            if r & 1:
                self.lazy[r - 1] += x
                self.data[r - 1] += x
            r >>= 1
            l >>= 1
        for i in ids:
            self.data[i] = self.segfunc(self.data[2 * i], self.data[2 * i + 1]) + self.lazy[i]


    def query(self, l, r):
        """
        [l, r)のsegfuncしたものを得る
        l: index(0-index)
        r: index(0-index)
        """
        *ids, = self.gindex(l, r)
        self.propagates(*ids)

        res = self.ide_ele

        l += self.num
        r += self.num
        while l < r:
            if l & 1:
                res = self.segfunc(res, self.data[l])
                l += 1
            if r & 1:
                res = self.segfunc(res, self.data[r - 1])
            l >>= 1
            r >>= 1
        return res

N = int(input())
A = list(map(int, input().split()))
mod = 10**9+7

if 0 in A:
    print(0)
    exit()

lst = LazySegmentTree([0]*(N+5),lambda x,y:x+y,0)
l,r = 0,0
sm = 1
while l<N:
    while r<N and sm*A[r]<10**9:
        sm *= A[r]
        lst.add(l,r+1,1)
        lst.add(r+1,r+2,-(r-l+1))
        r += 1
    sm //= A[l]
    l += 1

res = 1
s = 0
for i in range(N):
    s += lst.query(i,i+1)
    if A[i]!=1:
        res *= pow(A[i],s,mod)
        res %= mod
print(res)
0