主元素 III
题目
给定一个整型数组,找到主元素,它在数组中的出现次数严格大于数组元素个数的1/k。
注意事项
数组中只有唯一的主元素说明
给出数组 [3,1,2,3,2,3,3,4,4,4] ,和 k = 3,返回 3
挑战
要求时间复杂度为O(n),空间复杂度为O(k)
题解
沿用前两题的思路,对不同的K个数进行消除,这里选用HashMap对Candidates进行处理。在遍历过程中如有K个不同的数则全部消除,反之记录其出现次数,并在HashMap长度大于K时随时删除长度为1的记录。这样可以保证主元素在遍历完成后一定会出现在HashMap中,在将HashMap中所有的数进行一次遍历并记录出现的次数,次数最大的就是主元素。
public class Solution
{
/**
* @param nums:
* A list of integers
* @param k:
* As described
* @return: The majority number
*/
public int majorityNumber(ArrayList<Integer> nums, int k)
{
HashMap<Integer, Integer> counters = new HashMap<Integer, Integer>();
for (Integer i : nums)
{
counters.put(i, !counters.containsKey(i)?1:counters.get(i) + 1);
if (counters.size() >= k)
{
removeKey(counters);
}
}
for (Integer i : counters.keySet())
{
counters.put(i, 0);
}
for (Integer i : nums)
{
if (counters.containsKey(i))
{
counters.put(i, counters.get(i) + 1);
}
}
int maxCounter = 0, maxKey = 0;
for (Integer i : counters.keySet())
{
if (counters.get(i) > maxCounter)
{
maxCounter = counters.get(i);
maxKey = i;
}
}
return maxKey;
}
private void removeKey(HashMap<Integer, Integer> counters)
{
List<Integer> removeList = new ArrayList<>();
for (Integer key : counters.keySet())
{
if (counters.get(key) == 1)
{
removeList.add(key);
}
}
for (Integer key : removeList)
{
counters.remove(key);
}
}
}
Last Update 2016.9.30