結果

問題 No.2581 [Cherry Anniversary 3] 28輪の桜のブーケ
ユーザー 👑 rin204rin204
提出日時 2023-12-09 13:07:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 294 ms / 3,000 ms
コード長 1,065 bytes
コンパイル時間 356 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 86,400 KB
最終ジャッジ日時 2024-09-27 03:35:00
合計ジャッジ時間 9,027 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

from bisect import bisect_left

M = int(input())
n = 28
G = list(map(int, input().split()))
H = list(map(int, input().split()))

l = 18
L = [[] for _ in range(l + 1)]
for bit in range(1 << l):
    pc = 0
    tot = 0
    for i in range(l):
        if bit >> i & 1:
            pc += 1
            tot += G[i]
        else:
            tot += H[i]
    L[pc].append(tot % M)

for i in range(l + 1):
    L[i].sort()

R = []
for bit in range(1 << (n - l)):
    pc = 0
    tot = 0
    for i in range(n - l):
        if bit >> i & 1:
            pc += 1
            tot += G[i + l]
        else:
            tot += H[i + l]
    R.append((pc, tot % M))

Q = int(input())
for _ in range(Q):
    k, x = map(int, input().split())
    ans = 0
    for kk, vv in R:
        t = k - kk
        if t < 0 or l < t:
            continue
        low = (x - vv) % M
        upp = M - vv
        if low < upp:
            ans += bisect_left(L[t], upp) - bisect_left(L[t], low)
        else:
            ans += bisect_left(L[t], upp) + len(L[t]) - bisect_left(L[t], low)

    print(ans)
0