9.27 百度笔试 算法卷AC

from collections import defaultdict

n,k = tuple(map(int ,input().split()))
nums = list(map(int , input().split()))

diffk = defaultdict(int)
res = 0
for num in nums:
    res += diffk.get(num , 0)
    diffk[num + k] += 1
    diffk[num - k] += 1
print(res)

#5 3
#3 2 5 4 6
#1 2 1 3 0
#2

#1 2
#3
#100
#-1

import heapq

n,k0 = tuple(map(int , input().split()))
h = list(map(int , input().split()))
a = list(map(int , input().split()))

prev = []
heapq.heapify(prev)
res = 0
k = k0
flag = 1
for i in range(n):
    if k >= h[i]:
        heapq.heappush(prev,-a[i])
    else:
        while k < h[i] and prev != []:
            k -= heapq.heappop(prev)
            res += 1
        heapq.heappush(prev,-a[i])
        
    if k < h[i]:
        flag = 0
        break

if flag == 0:
    print(-1)
else:
    print(res)