LeetCode-911 在线选举

来源：力扣（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] <= 10⁹
times 是一个严格递增的有序数组
times[0] <= t <= 10⁹
每个测试用例最多调用 10⁴ 次 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);
 */

 

 

 

运行结果