1. 程式人生 > 其它 >leetcode題解記錄-1471 陣列中的K個最強值(python3)

leetcode題解記錄-1471 陣列中的K個最強值(python3)

技術標籤:leetcodeleetcode

題目

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