※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 125.229.37.69 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1731510632.A.C65.html
用一堆bisect function可以過
但其實insort是O(N) 所以這樣是O(N^2)
想當然是墊底
看答案
原來先sort+two pointer也行
我好笨
def countFairPairs(self, nums: List[int], lower: int, upper: int) -> int:
traveled = []
ans = 0
for num in nums:
if traveled:
idx_r = bisect_right(traveled, upper-num)
idx_l = bisect_left(traveled, lower-num)
ans += (idx_r-idx_l)
insort(traveled, num)
return ans
--