結果

問題 No.2273 一点乗除区間積
コンテスト
ユーザー ecottea
提出日時 2023-03-09 03:46:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,842 ms / 5,000 ms
コード長 2,625 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 82,476 KB
実行使用メモリ 192,860 KB
最終ジャッジ日時 2024-09-18 02:44:14
合計ジャッジ時間 12,171 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

def factor_integer(n):
    ps = []
    es = []

    i = 2
    while (i * i <= n):
        e = 0
        while n % i == 0:
            e += 1
            n //= i
        if e > 0:
            ps += [i]
            es += [e]
        i += 1
    
    if n > 1:
        ps += [n]
        es += [1]
    
    return ps, es


class SegmentTree:
    def __init__(self, ini, op, e):
        self.n = len(ini)
        self.op = op
        self.e = e
        self.v = [e] * (2 * self.n)
    
        for i in range(self.n):
            self.v[self.n + i] = ini[i]
  
        for i in range(self.n - 1, 0, -1):
            self.v[i] = op(self.v[2 * i], self.v[2 * i + 1])

    def set(self, i, x):
        i += self.n
        self.v[i] = x
        while i > 1:
            i >>= 1
            self.v[i] = self.op(self.v[2 * i], self.v[2 * i + 1])

    def prod(self, l, r):
        resl = self.e
        resr = self.e

        l += self.n
        r += self.n
        while l < r:
            if l & 1:
                resl = self.op(resl, self.v[l])
                l += 1
            if r & 1:
                resr = self.op(self.v[r - 1], resr)
            l >>= 1
            r >>= 1
        return self.op(resl, resr)
    
    def get(self, i):
        return self.v[i + self.n]
    
    def dump(self):
        print(self.v[self.n:])


N, B, Q = map(int, input().split())

ps, es = factor_integer(B)
K = len(ps)

def op(x, y):
    res = [0] * (K + 1)
    for k in range(K):
        res[k] = x[k] + y[k]
    res[K] = x[K] * y[K] % B
    return res

e = ([0] * K) + [1]

A = list(map(int, input().split()))

ini = []
for n in range(N):
    x = [0] * (K + 1)
    if A[n] != 0:
        for k in range(K):
            while A[n] % ps[k] == 0:
                A[n] //= ps[k]
                x[k] += 1
    x[K] = A[n] % B
    ini += [x]

seg = SegmentTree(ini, op, e)

for _ in range(Q):
    j, m, l, r = map(int, input().split())

    x = seg.get(j)

    divisible = False
    if m == B:
        divisible = True
        if x[K] != 0:
            for k in range(K):
                if x[k] < es[k]:
                    divisible = False
                    break
            if divisible:
                for k in range(K):
                    x[k] -= es[k]

    if not divisible:
        if m != 0:
            for k in range(K):
                while m % ps[k] == 0:
                    m //= ps[k]
                    x[k] += 1
        
        x[K] = x[K] * m % B
    
    seg.set(j, x)
   
    x = seg.prod(l, r + 1)

    if x[K] != 0:
        for k in range(K):
            x[K] = x[K] * pow(ps[k], x[k], B) % B
    
    print(x[K])
0