mirror of
https://github.com/ShusenTang/LeetCode.git
synced 2024-09-02 14:20:01 +00:00
62 lines
2.4 KiB
Markdown
62 lines
2.4 KiB
Markdown
# [169. Majority Element](https://leetcode.com/problems/majority-element/description/)
|
||
# 思路
|
||
题目要求就是求数组主元素,主元素就是在数组中出现次数超过元素个数一半的元素,题目保证主元素一定存在。
|
||
## 思路一
|
||
若对数组nums进行排序,则nums[n/2]就是主元素,即题目转换成求数组中第n/2小的元素。
|
||
求数组第k小的元素最优的思想应该就是采用类似快排划分的思想:
|
||
从数组中随机选择(一般选择第一个)一个元素作为pivot任何进行和快排划分一样的操作后,数组被pivot划分为两部分nums[0...m-1]和nums[m+1...n-1], nums[m]=pivot。
|
||
讨论m和k的关系:
|
||
(1) 若m=k, 即找到了,直接返回pivot;
|
||
(2) 若m<k, 对nums[m+1...n-1]递归地查找第k-m小的元素(对整个原数组来说还是第k小的元素);
|
||
(3) 若m>k, 对nums[0...m-1]递归地查找第k小的元素;
|
||
该算法平均情况下时间复杂度为O(n), 空间复杂度取决于划分的方法。
|
||
## 思路二
|
||
本题其实是思路一能求解的题目的一个特例。下面介绍更快的方法:
|
||
# C++
|
||
## 思路一
|
||
```
|
||
// 提交结果为900ms,相对于思路二可以说是很慢了
|
||
class Solution {
|
||
public:
|
||
// 定义递归函数
|
||
int kth_elem(vector<int>& nums, int low, int high, int k){
|
||
int pivot=nums[low];
|
||
int low_bk=low, high_bk = high; // 后面要修改low和high,所以先备份
|
||
while(low < high){
|
||
while(low < high && nums[high] >= pivot) high--;
|
||
nums[low] = nums[high];
|
||
while(low < high && nums[low] <= pivot) low++;
|
||
nums[high] = nums[low];
|
||
} // low==high 退出循环
|
||
nums[low]=pivot;
|
||
// 以上为快排划分思想
|
||
|
||
if(low == k) return pivot;
|
||
else if(low > k) return kth_elem(nums, low_bk, low-1, k);
|
||
else return kth_elem(nums, low+1, high_bk, k);
|
||
}
|
||
|
||
int majorityElement(vector<int>& nums) {
|
||
int k = nums.size() / 2;
|
||
return kth_elem(nums, 0, nums.size() - 1, k);
|
||
|
||
}
|
||
};
|
||
```
|
||
## 思路二
|
||
```
|
||
// 提交结果为12ms,较思路一有大幅提升
|
||
class Solution {
|
||
public:
|
||
int majorityElement(vector<int>& nums) {
|
||
int major = nums[0], count = 0;
|
||
for(int num : nums){ // 范围for语句
|
||
if(major == num) count++;
|
||
else if(count == 1) major = num;
|
||
else count--;
|
||
}
|
||
return major;
|
||
}
|
||
};
|
||
```
|