JAVA程序设计：在线选举（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 <= persons.length = times.length <= 5000
0 <= persons[i] <= persons.length
times 是严格递增的数组，所有元素都在 [0, 10^9] 范围中。
每个测试用例最多调用 10000 次 TopVotedCandidate.q。
TopVotedCandidate.q(int t) 被调用时总是满足 t >= times[0]。

方法一：前缀和+二分。（超时）

我们考虑求每个人关于时间的前缀和，arri][j]表示第i个人在j秒前（包括j秒）获得的票数，mx[i]表示前i秒最多票数呢个人得票的数量，last[i]为第i个人最后一次得票的时间。qq[i]表示第i秒得票最多的是哪个人，然后遍历一波预处理就好啦，对于time，因为数字很大，我们考虑离散化，然后二分找值即可。

不解：5000*5000+10000log(5000)的复杂度为啥超时？

class TopVotedCandidate {
	
	private int n;
	int[] qq;
	int[] mx;
	int[] last;
	int[] nowTime;
	private int[][] arr;
	
    public TopVotedCandidate(int[] persons, int[] times) {
    	
    	n=persons.length;
    	qq=new int[n];
    	mx=new int[n];
    	last=new int[n+1];
    	nowTime=new int[n];
    	arr=new int[n+1][n];
    	
    	for(int i=0;i<n;i++) {
    		nowTime[i]=times[i];
    		for(int j=0;j<=n;j++) {
    			if(persons[i]==j) {
    				if(i==0) arr[j][i]=1;
    				else arr[j][i]=arr[j][i-1]+1;
    				last[j]=i;
    			}
    			else {
    				if(i==0) arr[j][i]=0;
    				else arr[j][i]=arr[j][i-1];
    			}
    			if(arr[j][i]>=mx[i]) {
    				if(arr[j][i]==mx[i] && last[j]>=last[qq[i]] || arr[j][i]>mx[i])
    					qq[i]=j;
    				mx[i]=arr[j][i];
    			}
    		}
    	}
    }
    
    public int q(int t) {
    	
    	int index=find(t);
    	
    	return qq[index];
    }
    
    private int find(int x) {
    	
    	int p=0;
    	int l=0,r=n-1;
    	while(l<=r) {
    		int mid=(l+r)/2;
    		if(nowTime[mid]<=x) {
    			p=mid;
    			l=mid+1;
    		}
    		else
    			r=mid-1;
    	}
    	
    	return p;
    }
}

方法二：列表+二分

我么可以把选票存储在选票列表的列表 A 中。每个投票都有一个人和一个时间戳，A[count] 是一个列表，记录当前人获得的第 count 张选票。然后，A[i][0] 和 A[i] 单调增加，所以我们可以利用二分搜索根据时间找到最近的选票。

class TopVotedCandidate {
	
	class vote{
		int person,time;
		public vote(int person,int time) {
			this.person=person;
			this.time=time;
		}
	}
	
	List<List<vote>> list;
	
    public TopVotedCandidate(int[] persons, int[] times) {
    	
    	list=new ArrayList<>();
    	Map<Integer,Integer> map=new HashMap<>();
    	
    	for(int i=0;i<persons.length;i++) {
    		int p=persons[i],t=times[i];
    		int c=map.getOrDefault(p, 0)+1;
    		
    		map.put(p, c);
    		while(list.size()<=c)
    			list.add(new ArrayList<>());
    		list.get(c).add(new vote(p,t));
    	}
    }
    
    public int q(int t) {
    	
    	int l=1,r=list.size();
    	while(l<r) {
    		int mid=(l+r)/2;
    		if(list.get(mid).get(0).time<=t)
    			l=mid+1;
    		else
    			r=mid;
    	}
    	
    	int i=l-1;
    	
    	l=0; r=list.get(i).size();
    	while(l<r) {
    		int mid=(l+r)/2;
    		if(list.get(i).get(mid).time<=t)
    			l=mid+1;
    		else
    			r=mid;
    	}
    	
    	int j=Math.max(l-1, 0);
    	
    	return list.get(i).get(j).person;
    }
}