結果

問題 No.2581 [Cherry Anniversary 3] 28輪の桜のブーケ
コンテスト
ユーザー rin204
提出日時 2023-12-09 13:06:15
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,066 bytes
コンパイル時間 296 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 86,272 KB
最終ジャッジ日時 2024-09-27 03:34:51
合計ジャッジ時間 9,161 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 31 WA * 1
権限があれば一括ダウンロードができます

ソースコード

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