結果

問題 No.2758 RDQ
ユーザー NP
提出日時 2024-05-17 21:37:11
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 841 bytes
コンパイル時間 381 ms
コンパイル使用メモリ 82,580 KB
実行使用メモリ 1,035,940 KB
最終ジャッジ日時 2024-12-20 13:24:43
合計ジャッジ時間 38,506 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 MLE * 1
other AC * 6 TLE * 10 MLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

def preprocess_prefix_sums(A, K):
    N = len(A)
    prefix = [0] * (N + 1)
    for i in range(1, N + 1):
        prefix[i] = prefix[i - 1] + (1 if A[i - 1] % K == 0 else 0)
    return prefix
import sys
input = sys.stdin.read
data = input().split()
N = int(data[0])
Q = int(data[1])
A = list(map(int, data[2:N+2]))
queries = data[N+2:]
unique_ks = set()
query_data = []
for i in range(Q):
    L = int(queries[3 * i]) - 1
    R = int(queries[3 * i + 1])
    K = int(queries[3 * i + 2])
    unique_ks.add(K)
    query_data.append((L, R, K))
prefix_sums = {}
for K in unique_ks:
    prefix_sums[K] = preprocess_prefix_sums(A, K)
results = []
for L, R, K in query_data:
    prefix = prefix_sums[K]
    result = prefix[R] - prefix[L]
    results.append(str(result))
sys.stdout.write('\n'.join(results) + '\n')
0