跳转至

LeetCode: 911. 在线选举

1、题目描述

在选举中,第 i 张票是在时间为 times[i] 时投给 persons[i] 的。

现在,我们想要实现下面的查询函数: TopVotedCandidate.q(int t) 将返回在 t 时刻主导选举的候选人的编号。

t 时刻投出的选票也将被计入我们的查询之中。在平局的情况下,最近获得投票的候选人将会获胜。

示例:

输入:["TopVotedCandidate","q","q","q","q","q","q"], [[[0,1,1,0,0,1,0],[0,5,10,15,20,25,30]],[3],[12],[25],[15],[24],[8]]
输出:[null,0,1,1,0,0,1]
解释:
时间为 3,票数分布情况是 [0],编号为 0 的候选人领先。
时间为 12,票数分布情况是 [0,1,1],编号为 1 的候选人领先。
时间为 25,票数分布情况是 [0,1,1,0,0,1],编号为 1 的候选人领先(因为最近的投票结果是平局)。
在时间 15、24 和 8 处继续执行 3 个查询。

提示:

  • 1 <= persons.length = times.length <= 5000
  • 0 <= persons[i] <= persons.length
  • times 是严格递增的数组,所有元素都在 [0, 10^9] 范围中。
  • 每个测试用例最多调用 10000 次 TopVotedCandidate.q。
  • TopVotedCandidate.q(int t) 被调用时总是满足 t >= times[0]。

2、解题思路

  • 首先统计出不同时刻的胜利候选人
  • 然后采用二分查找,找到对应的时间点对应的候选人返回即可
from collections import defaultdict
from bisect import bisect


class TopVotedCandidate:

    def __init__(self, persons: List[int], times: List[int]):
        self.time = times
        self.winner = {}
        tickets = defaultdict(int)

        pre_winner = [-1, -1]
        for t, p in zip(times, persons):

            tickets[p] += 1
            if tickets[p] >= pre_winner[0]:
                self.winner[t] = p
                pre_winner = [tickets[p], p]
            else:
                self.winner[t] = pre_winner[1]

    def q(self, t: int) -> int:
        pos = bisect(self.time, t)
        return self.winner[self.time[pos - 1]]

# Your TopVotedCandidate object will be instantiated and called as such:
# obj = TopVotedCandidate(persons, times)
# param_1 = obj.q(t)