题目:多数元素
给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数大于 ⌊ n/2 ⌋ 的元素。 你可以假设数组是非空的,并且给定的数组总是存在多数元素。这里主要是以分治算法来求解这道题,关于分治算法的思路,可以click here, 有关很多的其它解法可看leetcode官方解法。
这里解决的问题很简单,就是跟你一个数组,找出其中的众数。
分解 ==> 求解 ==> 合并
首先就是要将给定的数组分成两部分【分解】,分别求解这两部分的众数【求解】,然后比较这两部分的众数,最终得出结果【合并】。
注意,我们要设定边界条件。
假设,lo、hi分别是选取数组的左边索引,右边索引(包括lo和hi),mid是切分的位置 给定一个数组a[lo, hi]
分解: mid = 索引的中间位置, (hi - lo) // 2 + lo 则左边数组为 left = a[lo, mid] 右边数组为 right = a[mid + 1, hi]
求解: 对于分解的两部分数据进行递归求解
合并: 统计左右两边的众数出现的次数,返回结果
class Solution: def majorityElement(self, nums: List[int]) -> int: def majority_element_rec(lo, hi): # 边界条件,或递归终止条件 if lo == hi: return nums[lo] # 计算lo、hi的中间索引 mid = (hi - lo) // 2 + lo # 分解、求解左边数组的众数和右边数组的众数 left = majority_element_rec(lo, mid) right = majority_element_rec(mid + 1, hi) if left == right: return left # 归并 left_count = sum(1 for i in range(lo, hi + 1) if nums[i] == left) right_count = sum(1 for i in range(lo, hi + 1) if nums[i] == right) return left if left_count > right_count else right return majority_element_rec(0, len(nums)-1)