primes_and_queries.py

# Copyright (c) 2021 kamyu. All rights reserved.
#
# Google Kick Start 2021 Round D - Problem D. Primes and Queries
# https://codingcompetitions.withgoogle.com/kickstart/round/00000000004361e3/000000000082bcf4
#
# Time:  O(N * (logN + log(MAX_A)) + Q * (logN + log(MAX_VAL) + log(MAX_S)))
# Space: O(N)
#

class BIT(object):  # 0-indexed.
    def __init__(self, n):
        self.__bit = [0]*(n+1)  # Extra one for dummy node.

    def add(self, i, val):
        i += 1  # Extra one for dummy node.
        while i < len(self.__bit):
            self.__bit[i] += val
            i += (i & -i)

    def query(self, i):
        i += 1  # Extra one for dummy node.
        ret = 0
        while i > 0:
            ret += self.__bit[i]
            i -= (i & -i)
        return ret

def vp(p, x):  # Time: O(logx)
    if x == 0:
        return 0
    result = 0
    while x%p == 0:
        x //= p
        result += 1
    return result

def add(p, bits, pos, val, sign):  # Time: O(logN + log(MAX_VAL))
    a, b = val, val%p
    if a == b:
        return  # V(a^s - b^s) is 0, just skip
    if b == 0:
        bits[0].add(pos, sign * vp(p, a))
    else:
        bits[1].add(pos, sign * 1)
        bits[2].add(pos, sign * vp(p, a-b))
        if p == 2:
            bits[3].add(pos, sign * (vp(p, a+b)-1))

# reference: https://en.wikipedia.org/wiki/Lifting-the-exponent_lemma#Statements
# given a is N+, p is prime, b = a%p,
# find V(a^s - b^s) => (a-b)%p = 0
# (1) if a = b => V(a^s - b^s) = 0
# (2) if a != b =>
#     (2.1) if b = 0 => V(a^s - b^s) = V(a^s) = s*V(a)
#     (2.2) if b != 0 => a%p != 0 and b%p != 0 =>
#           (2.2.1) if p != 2 or s%2 != 0 => V(a^s - b^s) = V(a-b) + V(s)
#           (2.2.2) if p = 2 and s%2 = 0  => V(a^s - b^s) = V(a-b) + V(a+b) + V(s) - 1
def query(p, bits, pos, s):  # Time: O(logN + log(MAX_S))
    # sum(s*vp(p, A[i]) for i in xrange(pos+1) if A[i] >= p and A[i]%p == 0) + \
    # sum(vp(p, s) + vp(p, A[i]-A[i]%p) for i in xrange(pos+1) if A[i] >= p and A[i]%p != 0) + \
    # (sum(vp(p, A[i]+A[i]%p)-1 for i in xrange(pos+1) if A[i] >= p and A[i]%p != 0) if p == 2 and s%2 == 0 else 0)
    return s*bits[0].query(pos) + \
           vp(p, s)*bits[1].query(pos) + bits[2].query(pos) + \
           (bits[3].query(pos) if p == 2 and s%2 == 0 else 0)

def primes_and_queries():
    N, Q, P = map(int, raw_input().strip().split())

    bits = [BIT(N) for _ in xrange(3 if P != 2 else 4)]
    A = [0]*N
    for i, val in enumerate(map(int, raw_input().strip().split())):
        A[i] = val
        add(P, bits, i, A[i], 1)  # Time: O(logN + log(MAX_A))
    result = []
    for ops in (map(int, raw_input().strip().split()) for _ in xrange(Q)):
        if len(ops) == 3:
            _, pos, val = ops
            i = pos-1
            add(P, bits, i, A[i], -1)  # Time: O(logN + log(MAX_VAL))
            A[i] = val
            add(P, bits, i, A[i], 1)  # Time: O(logN + log(MAX_VAL))
        else:
            _, S, L, R = ops
            result.append(query(P, bits, (R-1), S) - query(P, bits, (L-1)-1, S))  # Time: O(logN + log(MAX_S))
    return " ".join(map(str, result))

for case in xrange(input()):
    print 'Case #%d: %s' % (case+1, primes_and_queries())