leetcode題解記錄-1471 陣列中的K個最強值(python3)
阿新 • • 發佈:2021-02-19
題目
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