给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋ 的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
这种方式超时了,但是使用partition函数的思路可以借鉴:
由于多数元素一定出现在数组的中间位置,那么使用快排的partition函数能够帮助定位。
解题思路:
- partition函数的作用是:int partition(vector<int>& nums,int beg,int end)对于给定的nums,开始和结束位置为beg 和end,也就是partition的操作对象为一个子数组。将子数组以某一个元素为界分为两部分,返回该分界点元素的下标。
- 循环:当pos不等于mid时进行循环,如果pos比mid小,那么可以收缩左边界,如果pos比mid大,那么可以收缩右边界。
class Solution {
public:
int majorityElement(vector<int>& nums) {
int n = nums.size();
int pos = partition(nums,0,n-1);
int start=0;int end=n-1;
int mid = n/2;
while(pos!=mid){
if(pos<mid){
start = pos+1;
pos = partition(nums,start,end);
}
else if(pos>mid){
end = pos-1;
pos = partition(nums,start,end);
}
}
return nums[pos];
}
int partition(vector<int>& nums,int beg,int end){
int i=beg;int j=end;
int pivot = nums[beg];
while(i<j){
while(i<j&&nums[j]>=pivot){
j--;
}
while(i<j&&nums[i]<=pivot){
i++;
}
if(i<j){
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
}
nums[beg] = nums[i];
nums[i] = pivot;
return i;
}
};