870. 优势洗牌
解法:双指针+贪心
类似田忌赛马,对于每个状态,比较A和B中的最大值max1、max2,若max1>max2,即当前状态上等马能够胜利,则将max1放在max2所在索引。否则放弃此轮比较,拿A最小值放在max2所在索引,即用下等马对抗对方上等马。需要注意的是不能使用之前状态已经出战过的马匹。
具体可以参考:田忌赛马背后的算法决策
class Solution:
def advantageCount(self, nums1: List[int], nums2: List[int]) -> List[int]:
maxpq = []
for i, num in enumerate(nums2):
heapq.heappush(maxpq, (-1*num, i))
nums1.sort()
left, right = 0, len(nums1) - 1
result = [0] * len(nums1)
# print(maxpq)
while maxpq:
next_item = heapq.heappop(maxpq)
i, value = next_item[1], -1*next_item[0]
if nums1[right] > value:
result[i] = nums1[right]
right -= 1
else:
result[i] = nums1[left]
left += 1
return result
1287. 有序数组中出现次数超过25%的元素
解法:遍历
遍历,统计元素的出现次数即可
class Solution:
def findSpecialInteger(self, arr: List[int]) -> int:
if len(arr) == 1:
return arr[0]
threshold = len(arr) // 4
count = 1
for i in range(len(arr) - 1):
if arr[i] == arr[i+1]:
count += 1
if count > threshold:
return arr[i]
else:
if count > threshold:
return arr[i]
else:
count = 1