在选举中,第 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]。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/online-election 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class TopVotedCandidate {
vector<int> winner;
vector<int> T;
public:
TopVotedCandidate(vector<int>& persons, vector<int>& times) {
int maxVotes = 0, winId = -1;
unordered_map<int,int> votes;//人,得票数
T = times;
for(int i = 0; i < persons.size(); ++i)
{
votes[persons[i]]++;
if(votes[persons[i]] >= maxVotes)
{
winId = persons[i];
maxVotes = votes[persons[i]];
}
winner.push_back(winId);
}
}
int q(int t) {
int l = 0, r = T.size()-1, mid;
while(l <= r)//找小于等于我的最后一个时刻
{
mid = l +((r-l)>>1);
if(T[mid] > t)
r = mid-1;
else
{
if(mid == T.size()-1 || T[mid+1] > t)
return winner[mid];
else
l = mid+1;
}
}
return -1;
}
};
532 ms 91.2 MB