leetcode 911：在线选举

题目描述

在选举中，第 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. 1 <= persons.length = times.length <= 5000
2. 0 <= persons[i] <= persons.length
3. times 是严格递增的数组，所有元素都在 [0, 10^9] 范围中。
4. 每个测试用例最多调用 10000 次 TopVotedCandidate.q。
5. 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]];
    }