結果

問題 No.2501 Maximum Inversion Number
ユーザー なえしら
提出日時 2024-04-20 13:10:46
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 650 bytes
コンパイル時間 267 ms
コンパイル使用メモリ 81,908 KB
実行使用メモリ 112,736 KB
最終ジャッジ日時 2024-10-12 08:36:01
合計ジャッジ時間 7,457 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 2 WA * 14 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

def binary_search(ok, ng):
  while abs(ok - ng) > 1:
    md = (ng + ok) // 2
    S = sum(min(max(l, md), r) for l, r in zip(L, R))
    if S <= M: ok = md
    else: ng = md
  return ok

for _ in range(T := int(input())):
  N, M = map(int, input().split())
  L = list(map(int, input().split()))
  R = list(map(int, input().split()))
  
  if not sum(L) <= M <= sum(R):
    print(-1); continue

  K = [binary_search(min(L), max(R)+1)]*N
  for i in range(N):
    K[i] = min(max(L[i], K[i]), R[i])
  S = sum(K)
  for i in range(N):
    if S == M: break
    if R[i] <= K[i] < R[i]:
    	K[i] += 1; S += 1
  
  print(M*(M-1)//2 - sum(k*(k-1)//2 for k in K))
0