結果

問題 No.2581 [Cherry Anniversary 3] 28輪の桜のブーケ
ユーザー 👑 p-adicp-adic
提出日時 2023-12-05 01:40:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 990 ms / 3,000 ms
コード長 800 bytes
コンパイル時間 294 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 79,360 KB
最終ジャッジ日時 2024-09-27 03:16:33
合計ジャッジ時間 15,399 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

I,F=input,range
J=lambda:map(int,I().split())
M=int(I())
G=[list(J())for i in F(2)]
C=[0]*15
L=[{}for i in F(15)]
R=[{}for i in F(15)]
for i in F(1<<14):
	l=r=m=0
	for d in F(14):
		b=(i>>d)&1
		m,l,r=m+1-b,l+G[b][d],r+G[b][14+d]
	if l%M in L[m]:L[m][l%M]+=1
	else:L[m][l%M]=1
	if r%M in R[m]:R[m][r%M]+=1
	else:R[m][r%M]=1
T=[]
S=[{}for i in F(15)]
for e in F(15):
	l=0
	for i in sorted(R[e]):S[e][i],l=l,l+R[e][i]
	S[e][M]=l
	T+=[sorted(S[e])]
Q=int(I())
for q in F(Q):
	K,X=J()
	a=0
	for c in F(max(0,K-14),min(15,K+1)):
		e=K-c
		for i in L[c]:
			b,l,r=i>X,-1,len(T[e])-1
			while l+1<r:
				m=(l+r)//2
				if T[e][m]<M-i:l=m
				else:r=m
			l,s=-1,len(T[e])-1
			while l+1<s:
				m=(l+s)//2
				if T[e][m]<M*b+X-i:l=m
				else:s=m
			a+=L[c][i]*(S[e][T[e][r]]-S[e][T[e][s]]+S[e][M]*b)
	print(a)
0