题目描述
在选举中,第 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]
。
解题思路
读完本题的第一感觉很奇妙,感觉可以ac掉,但是又想不清如何组织数据。最后设置两个map,一个map<int,vector<int>>记录候选人得票情况,另一个map<int,int>记录时刻t获得票最多的候选人(相同票数按最近获得票数的优先)
map<int,vector<int>> vote;
map<int,int> que;
vector<int> vect;
TopVotedCandidate(vector<int> persons, vector<int> times) {
int len = times.size();
for(int i=0;i<len;i++){
int per = persons[i];
int tim = times[i];
vect.push_back(tim);
vote[per].push_back(tim);
int maxv = vote[per].size();
for(auto it=vote.begin();it != vote.end();it++){
int num = it->second.size();
if(num > maxv){
maxv = num;
per = it->first;
tim = it->second[num-1];
}else if(num == maxv && it->second[num-1] > tim){
per = it->first;
tim = it->second[num-1];
}
}
que[times[i]] = per;
}
}
int q(int t) {
int key = lower_bound(vect.begin(),vect.end(),t) - vect.begin();
if(vect[key] != t) key--;
return que[vect[key]];
}