[LeetCode] 705 | Design HashSet | Python

---

문제 링크: https://leetcode.com/problems/design-hashset/

---

문제

Design a HashSet without using any built-in hash table libraries.

Implement MyHashSet class:

• void add(key) Inserts the value key into the HashSet.
• bool contains(key) Returns whether the value key exists in the HashSet or not.
• void remove(key) Removes the value key in the HashSet. If key does not exist in the HashSet, do nothing.
•  

---

조건

• 0 <= key <= $10^6$
• At most $10^4$ calls will be made to add, remove, and contains.

---

내 풀이

리스트로 hashmap을 만든다. 0이면 없는 것, 1이면 있다는 뜻이다.

add함수는 self.hash[key]의 값을 1로 해서 있다고 저장하고, remove는 0으로 해서 없앤다.

contains함수는 self.hash[key]의 값이 0이면 False, 1이면 True를 리턴한다.

---

코드

class MyHashSet:

    def __init__(self):
        self.hash = [0 for _ in range(1000001)]

    def add(self, key: int) -> None:
        self.hash[key] = 1

    def remove(self, key: int) -> None:
        self.hash[key] = 0

    def contains(self, key: int) -> bool:
        return True if self.hash[key] else False


# Your MyHashSet object will be instantiated and called as such:
# obj = MyHashSet()
# obj.add(key)
# obj.remove(key)
# param_3 = obj.contains(key)