在线选举

题目链接

给你两个整数数组 persons 和 times 。在选举中,第 i 张票是在时刻为 times[i] 时投给候选人 persons[i] 的。

对于发生在时刻 t 的每个查询,需要找出在 t 时刻在选举中领先的候选人的编号。

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

实现 TopVotedCandidate 类:

TopVotedCandidate(int[] persons, int[] times) 使用 persons 和 times 数组初始化对象。 int q(int 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]

解释:
TopVotedCandidate topVotedCandidate = new TopVotedCandidate([0, 1, 1, 0, 0, 1, 0], [0, 5, 10, 15, 20, 25, 30]);
topVotedCandidate.q(3); // 返回 0 ,在时刻 3 ,票数分布为 [0] ,编号为 0 的候选人领先。
topVotedCandidate.q(12); // 返回 1 ,在时刻 12 ,票数分布为 [0,1,1] ,编号为 1 的候选人领先。
topVotedCandidate.q(25); // 返回 1 ,在时刻 25 ,票数分布为 [0,1,1,0,0,1] ,编号为 1 的候选人领先。(在平局的情况下,1 是最近获得投票的候选人)。
topVotedCandidate.q(15); // 返回 0
topVotedCandidate.q(24); // 返回 0
topVotedCandidate.q(8); // 返回 1

思路

解法一:

二分法

  1. 先统计出不同时刻领先的候选人
  2. 使用二分法找出不大于t的对应候选人
class TopVotedCandidate:

    def __init__(self, persons: List[int], times: List[int]):
        result = []
        vote_counts = defaultdict() # 候选人的得票数
        vote_counts[-1] = -1
        top = -1 # 领先的候选人
        for person in persons:
            vote_counts[person] = vote_counts.get(person, 0) + 1
            if vote_counts[person] >= vote_counts[top]:
                top = person
            result.append(top) # 不同时刻领先的候选人

        self.result = result
        self.times = times

    def q(self, t: int) -> int:
        left, right = 0, len(self.times) - 1
        while left < right:  # 找到满足 times[left] <= t 的最大的 l
            mid = left + (right - left + 1) // 2
            if self.times[mid] <= t:
                left = mid
            else:
                right = mid - 1
        return self.result[left]