結果

問題 No.2581 [Cherry Anniversary 3] 28輪の桜のブーケ
ユーザー rlangevin
提出日時 2023-12-09 00:16:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 910 ms / 3,000 ms
コード長 986 bytes
コンパイル時間 336 ms
コンパイル使用メモリ 82,964 KB
実行使用メモリ 78,552 KB
最終ジャッジ日時 2024-09-27 03:20:00
合計ジャッジ時間 15,031 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

from bisect import *

M = int(input())
G = list(map(int, input().split()))
H = list(map(int, input().split()))
G1, H1 = G[:14], H[:14]
G2, H2 = G[14:], H[14:]
D1, D2 = [[] for i in range(15)], [[] for i in range(15)]
for s in range(1 << 14):
    v1, v2 = 0, 0
    cnt = 0
    for i in range(14):
        if (s >> i) & 1:
            v1 += G1[i]
            v2 += G2[i]
            cnt += 1
        else:
            v1 += H1[i]
            v2 += H2[i]
    D1[cnt].append(v1%M)
    D2[cnt].append(v2%M)
    
for i in range(15):
    D1[i].sort()
    D2[i].sort()
    
Q = int(input())
for _ in range(Q):
    K, X = map(int, input().split())
    ans = 0
    for i in range(15):
        if K - i < 0 or K - i > 14:
            continue
        for d in D1[i]:
            ind1 = bisect_left(D2[K-i], X - d)
            ind2 = bisect_right(D2[K-i], M - d - 1)
            ind3 = bisect_left(D2[K-i], X + M - d)
            ans += ind2 - ind1 + len(D2[K-i]) - ind3
            
    print(ans)
0