题目:
给你两个整数数组 persons 和 times 。在选举中,第 i 张票是在时刻为 times[i] 时投给候选人 persons[i] 的。
对于发生在时刻 t 的每个查询,需要找出在 t 时刻在选举中领先的候选人的编号。
在 t 时刻投出的选票也将被计入我们的查询之中。在平局的情况下,最近获得投票的候选人将会获胜。
实现 TopVotedCandidate 类:
- TopVotedCandidate(int[] persons, int[] times) 使用 persons 和 times 数组初始化对象。
- int q(int t) 根据前面描述的规则,返回在时刻 t 在选举中领先的候选人的编号。
解答:
class TopVotedCandidate:
def __init__(self, persons: List[int], times: List[int]):
tops=[]
#使用vote来记录候选人的选票情况{候选人:所得票数}
vote=defaultdict(int)
#使用top来记录领先候选人
top=-1
vote[-1]=-1
for p in persons:
vote[p]+=1
if vote[p]>=vote[top]:
top=p
tops.append(top)
self.tops=tops
self.times=times
self.n=len(times)
#使用二分查找来确定领先候选人
#找到仅次于当前时间t的时间所对应的领先候选人
def q(self, t: int) -> int:
left,right=0,self.n-1
while left<right:
#当数组长度为偶数时,选取靠右的那个数
#取靠左的数,有可能会陷入死循环
mid=left+(right-left+1)//2
if self.times[mid]<=t:
left=mid
else:
right=mid-1
return self.tops[left]
# Your TopVotedCandidate object will be instantiated and called as such:
# obj = TopVotedCandidate(persons, times)
# param_1 = obj.q(t)