本质上是考查一个数字在已经排好序的序列中,排到第几位。原来的序列没有排序,所以需要sort一下。
class Solution:
def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
res = []
newnums = nums[::]
newnums.sort()
for idex, num in enumerate(nums):
ans = bisect.bisect_left(newnums, num)
res.append(ans)
return res
一般来说貌似sort可以直接用。不过面试fb家的面试时,被问到了一个问题,本质是要求写一个QS的函数出来。还是记一个模版吧。万一被问到啥的。。。
def QS(q, l, r):
if l >= r: return
mid = q[l+(r-l)//2]
i = l - 1
j = r + 1
while i < j:
while True:
i += 1
if q[i] >= mid:break
while True:
j -= 1
if q[j] <= mid:break
if i < j:
q[i],q[j] = q[j],q[i]
QS(q, l, j)
QS(q, j+1,r)
l = 0
r = len(q) - 1