結果

問題 No.2501 Maximum Inversion Number
ユーザー chineristACchineristAC
提出日時 2023-10-13 21:36:07
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,107 bytes
コンパイル時間 257 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 121,704 KB
最終ジャッジ日時 2024-09-15 17:12:54
合計ジャッジ時間 6,911 ms
ジャッジサーバーID
(参考情報)
judge4 / judge6
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 4 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
from itertools import permutations
import heapq
from collections import deque
import random

input = lambda :sys.stdin.readline().rstrip()
mi = lambda :map(int,input().split())
li = lambda :list(mi())

def solve(n,m,L,R):
    if m < sum(L)  or sum(R) < m:
        return -1
    
    cnt = [L[i] for i in range(n)]
    rest = m - sum(cnt)

    def cond(k):
        check = 0
        for i in range(n):
            if L[i] <= k:
                check += min(k,R[i]) - L[i]
        return check <= rest
    
    ok = min(L)
    ng = max(R) + 1
    while ng-ok>1:
        mid = (ok+ng)//2
        if cond(mid):
            ok = mid
        else:
            ng = mid
    
    
    
    for i in range(n):
        cnt[i] = max(min(R[i],ok),L[i])
    
    
    rest = m - sum(cnt)
    idx = [i for i in range(n)]
    idx.sort(key=lambda i:cnt[i])
    for i in range(rest):
        cnt[i] += 1
    
    res = m * (m-1)//2
    for i in range(n):
        res -= cnt[i] * (cnt[i]-1) //2
    
    return res

for _ in range(int(input())):
    n,m = mi()
    L = li()
    R = li()

    print(solve(n,m,L,R))
0