给定一个大小为 n
的数组 nums
,返回其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋
的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
1、自然能想到一次遍历,让不同的元素碰到就消失,这样最后剩下的就是大于一半的元素。在C和C++中,利用问号表达式,非常巧妙的减少了代码的长度。
C
int majorityElement(int* nums, int numsSize) {
int p=1;
int some_num = nums[0];
int count = 1;
while (p<numsSize){
if (count == 0) some_num = nums[p];
count += nums[p] != some_num ? -1 : 1;
p++;
}
return some_num;
}
C++
int majorityElement(vector<int>& nums) {
int some_num,count = 0;
for (int num : nums){
if (count == 0) some_num =num;
count += some_num != num ? -1 : 1;
}
return some_num;
}
};
2、go语言中,利用返回定义使得代码中可以直接返回
go
func majorityElement(nums []int) (some_num int) {
count := 0
for _,num := range nums{
if count == 0{
some_num = num
}
if some_num != num {
count--
}else{
count++
}
}
return
}
3、rust语言利用fold方法,fold是一个从左往右的迭代,把计算所有元素最终输出一个值,fold类似reduce,区别是fold允许初始化返回变量,而reduce默认返回变量的初始化是第一个元素。
Rust
pub fn majority_element(nums: Vec<i32>) -> i32 {
nums.into_iter().fold((0,0),|(mut some_num,mut count),num|{
if count == 0 {
some_num = num;
}
count += if some_num != num {-1}else{1};
(some_num,count)
}).0
}
4、为了减少if判断,还能把判断转为数学计算
Rust
pub fn majority_element(nums: Vec<i32>) -> i32 {
nums.into_iter().fold((0,0),|(mut some_num,mut count),num|{
if count == 0 {
some_num = num;
}
count += 1 - 2 * (some_num != num) as i32;
(some_num,count)
}).0
}