题目
给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数大于 ⌊ n/2 ⌋ 的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/majority-element
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
示例 1:
输入:[3,2,3]
输出:3
示例 2:
输入:[2,2,1,1,1,2,2]
输出:2
进阶
尝试设计时间复杂度为 O(n)、空间复杂度为 O(1) 的算法解决此问题。
解答
class Solution:
def majorityElement(self, nums):
len_list = len(nums)
for i in nums:
if nums.count(i) > len_list/2:
return i
if __name__ == '__main__':
li = [3, 2, 3]
res = Solution().majorityElement(li)
print(res) # 3
开始是这样做的,最笨的办法,遍历数组,判断当前元素出现次数是否大于数组长度的一半,如果大于,就return当前元素。但是执行时超出时间限制了。
于是考虑,将数组去重,这样遍历数组时就不用考虑已经判断过的元素。果然这样一改测试用例就通过了。
class Solution:
def majorityElement(self, nums):
l = len(nums) / 2
set_list = set(nums)
for i in set_list:
if nums.count(i) > l:
return i
if __name__ == '__main__':
li = [3, 2, 3]
res = Solution().majorityElement(li)
print(res) # 3
但是想达到进阶中要求的时间复杂度为 O(n)、空间复杂度为 O(1) 还需要再优化。
想到一种解法,既然要返回超过半数的值,那将列表排序,中间的那个值一定是满足要求的。
class Solution:
def majorityElement(self, nums):
nums.sort()
return nums[len(nums)//2]
if __name__ == '__main__':
li = [3, 2, 3]
res = Solution().majorityElement(li)
print(res) # 3