来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/online-election
题目描述
给你两个整数数组 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 <= persons.length <= 5000 times.length == persons.length 0 <= persons[i] < persons.length 0 <= times[i] <= 109 times 是一个严格递增的有序数组 times[0] <= t <= 109 每个测试用例最多调用 104 次 q
解题思路
本题采用预处理+二分法来求解,观察题目提示,times数组和persons数组大小仅为5000,可以作为突破口,将每个时间段的领先候选人记录下来,然后判断查询时候的时间处于哪个时间段内,直接将领先候选人的编号返回。利用计数的方法将所有数据预处理,将领先的候选人存入tops中,tops[i]代表在tims[i]到time[i+1]内领先候选人的编号,查询时候,利用二分库函数upper_bound找到t所在的时间段time[t],直接返回tops[t]。
源码展示
class TopVotedCandidate { public: vector<int> mTops; vector<int> mTimes; TopVotedCandidate(vector<int>& persons, vector<int>& times) { mTops.resize(times.size()); mTimes = times; unordered_map<int, int> mapiiCount; int iMax = 0, iPerson = -1; for(int i = 0; i < persons.size(); i++) { mapiiCount[persons[i]]++; if(iMax <= mapiiCount[persons[i]]) { iMax = mapiiCount[persons[i]]; iPerson = persons[i]; } mTops[i] = iPerson; } } int q(int t) { int iPos = upper_bound(mTimes.begin(), mTimes.end(), t) - mTimes.begin() - 1; return mTops[iPos]; } }; /** * Your TopVotedCandidate object will be instantiated and called as such: * TopVotedCandidate* obj = new TopVotedCandidate(persons, times); * int param_1 = obj->q(t); */
运行结果