結果

問題 No.2758 RDQ
ユーザー NPNP
提出日時 2024-05-17 21:37:11
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 841 bytes
コンパイル時間 252 ms
コンパイル使用メモリ 82,364 KB
実行使用メモリ 849,720 KB
最終ジャッジ日時 2024-05-17 21:37:18
合計ジャッジ時間 7,088 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
54,620 KB
testcase_01 AC 43 ms
54,616 KB
testcase_02 AC 40 ms
54,296 KB
testcase_03 AC 41 ms
54,504 KB
testcase_04 AC 44 ms
54,700 KB
testcase_05 AC 507 ms
186,432 KB
testcase_06 AC 429 ms
186,204 KB
testcase_07 AC 452 ms
186,248 KB
testcase_08 AC 422 ms
186,332 KB
testcase_09 AC 431 ms
186,452 KB
testcase_10 MLE -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
権限があれば一括ダウンロードができます

ソースコード

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