-
Notifications
You must be signed in to change notification settings - Fork 29
/
0169-Majority-Element.py
60 lines (48 loc) · 1.36 KB
/
0169-Majority-Element.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
'''
Given an array of size n, find the majority element. The majority element is the element that appears more than ⌊ n/2 ⌋ times.
You may assume that the array is non-empty and the majority element always exist in the array.
Example 1:
Input: [3,2,3]
Output: 3
Example 2:
Input: [2,2,1,1,1,2,2]
Output: 2
'''
from typing import List
# O(1) Space Complexity Solution 1
class Solution:
def majorityElement(self, nums: List[int]) -> int:
res = count = 0
for num in nums:
if not count:
res = num
count += 1
elif num == res:
count += 1
else:
count -= 1
return res
# O(1) Space Complexity Solution 2
class Solution:
def majorityElement(self, nums: List[int]) -> int:
res, count = nums[0], 1
for i in range(1, len(nums)):
count += 1 if nums[i] == res else -1
if not count:
res = nums[i]
count += 1
return res
# Using Hash Table
class Solution:
def majorityElement(self, nums: List[int]) -> int:
d = {}
for num in nums:
if num in d:
d[num] += 1
else:
d[num] = 1
return max(d, key = d.get)
# Custom Input Check
s = Solution()
answer = s.majorityElement([1, 1, 1, 5]) # 1
print(answer)