結果

問題 No.2961 Shiny Monster Master
コンテスト
ユーザー のーと
提出日時 2024-11-16 16:20:57
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,422 bytes
コンパイル時間 317 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 141,312 KB
最終ジャッジ日時 2024-11-16 16:21:31
合計ジャッジ時間 12,720 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 64 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

R, N = map(int, input().split())
A = set(map(int, input().split()))

# R周期の各秒に色違いが出現するかどうかを示すリスト
is_shiny = [0] * R
for a in A:
    is_shiny[a] = 1

# 累積和リスト
from itertools import accumulate
shiny_cumsum = list(accumulate(is_shiny))

# 累積和の区間和を計算する関数
def range_sum(S, l, r):
    if l == 0:
        return S[r]
    else:
        return S[r] - S[l - 1]

# 次の R の倍数を計算する関数
def next_multiple(x, m):
    return ((x + m - 1) // m) * m

# 前の R の倍数を計算する関数
def prev_multiple(x, m):
    return (x // m) * m

# 全周期の色違い出現回数
total_shiny_per_cycle = sum(is_shiny)

# クエリ処理
Q = int(input())
for _ in range(Q):
    l, r = map(int, input().split())
    # 次の R の倍数以上の時刻
    next_g = next_multiple(l, R)
    # 前の R の倍数以下の時刻
    prev_f = prev_multiple(r, R)

    if prev_f < next_g:
        # 1周期以内に収まる場合
        print(range_sum(shiny_cumsum, l % R, r % R))
    else:
        # 1周期を超える場合
        # 部分的な計算
        left_part = range_sum(shiny_cumsum, l % R, R - 1)
        right_part = range_sum(shiny_cumsum, 0, r % R)
        # 完全な周期の回数分を加算
        full_cycles = (prev_f - next_g) // R
        print(left_part + right_part + full_cycles * total_shiny_per_cycle)
0