leetcode刷题日记-在线选举

• 题目描述：
• 给你两个整数数组 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

• 解析，乍一看，这个问题有点小复杂，因为他是根据时间来决定应该查询的投票情况，当出现平票的时候，最近获得投票的人获胜。先不考虑这些细枝末节，很容易想到，输入t，我们要用二分查找在times找到他应该插入的位置索引，这就能得到我们时间的投票情况，但是如果我们每次输入t，都进行重新查找，即找到t对应的插入位置索引，得到投票情况，统计不同候选人的投票数，判断是否存在平票，如果不存在，返回最大的，如果存在，从后往前遍历当前时间范围内的投票情况，返回第一个处于平票状态的候选人编号，这样理论上是肯定行得通的，但是时间复杂度很高，在leetcode时间上不能通过。所以，我们进一步考虑这个问题，既然查询条件是时间，而且times是递增，因此，我们就在初始化的时候维护一个数组tops，每一个时刻对应一个成功的候选人，我们通过二分查找找到查询时刻所对应的插入位置索引，直接在tops里面进行查找就可以了。
• 代码如下：

import bisect
class TopVotedCandidate:

    def __init__(self, persons: List[int], times: List[int]):
        """
        对self.tops和self.times进行初始化
        >>>TopVotedCandidate topVotedCandidate = new TopVotedCandidate([0, 1, 1, 0, 0, 1, 0], [0, 5, 10, 15, 20, 25, 30]);
        >>>topVotedCandidat.tops
        >>>[0, 1, 1, 0, 0, 1, 0]
        """
        top, tops=-1,  []
        votes = {}
        votes[-1] = -1
        for vote in persons:
            votes[vote] = votes.setdefault(vote, 0) + 1
            if votes[vote] >= votes[top]: 
                top = vote
            tops.append(top)
        self.tops = tops
        self.times = times 
    def q(self, t: int) -> int:
        """
        返回当前时刻获胜的候选人
        >>>topVotedCandidate(12)
        >>>1
        """
        return self.tops[bisect.bisect_right(self.times, t)-1]

• bisect.bisect_right(times, t)是包bisect的一个方法，基于二分查找的方法返回t在times应该插入的位置索引，当times中有和t一样的元素时，将t插入到该相同元素的右边。