当Rust遇上LeetCode #169. 多数元素 [简单]

2020/3/15

题目描述

给定一个大小为 n 的数组，找到其中的多数元素。多数元素是指在数组中出现次数大于 ⌊ n/2 ⌋ 的元素。

你可以假设数组是非空的，并且给定的数组总是存在多数元素。

示例

示例 1:
输入: [3,2,3]
输出: 3

示例 2:
输入: [2,2,1,1,1,2,2]
输出: 2

相关标签

位运算
数组
分治算法

解题思路

• 算法：
  摩尔投票法
  候选人(cand_num)初始化为nums[0]，票数count初始化为1。
  当遇到与cand_num相同的数，则票数count = count + 1，否则票数count = count - 1。
  当票数count为0时，更换候选人，并将票数count重置为1。
  遍历完数组后，cand_num即为最终答案。
  为何这行得通呢？
  投票法是遇到相同的则票数 + 1，遇到不同的则票数 - 1。
  且“多数元素”的个数> ⌊ n/2 ⌋，其余元素的个数总和<= ⌊ n/2 ⌋。
  因此“多数元素”的个数 - 其余元素的个数总和 的结果 肯定 >= 1。
  这就相当于每个“多数元素”和其他元素 两两相互抵消，抵消到最后肯定还剩余至少1个“多数元素”。
  无论数组是1 2 1 2 1，亦或是1 2 2 1 1，总能得到正确的候选人。

• 复杂度分析：
  时间复杂度：O(n)
  空间复杂度：O(1)

源代码

impl Solution {
    pub fn majority_element(nums: Vec) -> i32 {
        let (mut cnt, len, mut res) = (0, nums.len(), nums[0]);
        for i in 0..len {
            cnt += if res == nums[i] { 1 } else { -1 };
            if cnt < 0 {
                res = nums[i];
                cnt = 1;
            }
        }
        res
    }
}

执行用时 : 0 ms, 在所有 Rust 提交中击败了100.00%的用户
内存消耗 : 2.1 MB, 在所有 Rust 提交中击败了100.00%的用户

总结

进阶问题： 229. 求众数 II