题目
1471 数组中的K个最强值
题目链接
解法思路: 双指针
代码记录
class Solution:
def getStrongest(self, arr: List[int], k: int) -> List[int]:
arr.sort()
m = arr[(len(arr)-1)//2]
res=[]
cnt=0
left=0
right=len(arr)-1
while cnt<k:
if abs(arr[left]-m)>abs(arr[right]-m):
res.append(arr[left])
cnt +=1
left+=1
else:
res.append(arr[right])
cnt +=1
right-=1
return res