题目描述:给定一个整型数组,找出主元素,它在数组中的出现次数严格大于数组元素个数的二分之一。
样例:给出数组[1,1,1,1,2,2,2],返回 1
两种方法可以用来解决这个问题:
1. 中位数,因为主元素出现次数严格大于数组元素个数的二分之一,所以肯定是数组的中位数。而求取中位数的算法曾经讲过(详见:点击打开链接)
2. 贪心法:因为元素个数大于数组的二分之一,所以,我们可以采用消除的方法,消去不是诸元素的,最终只剩下主元素。这种方法是我在这道题中要讲的方法。
其实,贪心算法不是某中具体的算法,而是一种思路,这种思路要求在解决一个问题时,每一步都选择当时情况下的最优解,寄希望于最终得到整个问题的最优解。这个东西一般和动态规划联系起来,里面涉及的内容非常多,所以,我在此并不展开讲解,只是大概介绍。
好了,看题。既然主元素有这样的性质,那么,可以考虑建立一个哈希表,键为元素,值为元素的个数,元素个数的变化满足以下规律:
1. 遍历的下一个元素如果和哈希表中存储的元素相等,个数加1
2. 如果不等,个数减1;如果个数小于或等于0了,改变键所代表的元素
这样一来,由于主元素的个数满足的条件,最后,键一定是主元素,且个数大于0
写出代码就行:
class Solution: """ @param nums: A list of integers @return: The majority number """ def majorityNumber(self, nums): ele = nums[0] count = 0 for i in nums: if ele == i: count += 1 else: count -= 1 if count <= 0: ele = i return ele # write your code here