Find the majority element

majority_element.py

class Solution(object):

    def solution(self, a):
        count = 0
        ele = None

        for n in a:
            if ele is None:
                count = 1
                ele = n
            else:
                if ele == n:
                    count += 1
                else:
                    if count != 0:
                        count -= 1

                    if count == 0:
                        ele = n
                        count = 1

        return ele


if __name__ == "__main__":
    a = [1, 1, 2, 2, 4]
    ans = Solution().solution(a)
    print(ans)