[LEET CODE] 706. Design HashMap

# Problem

Design a HashMap without using any built-in hash table libraries. Implement the MyHashMap class: MyHashMap() initializes the object with an empty map. void put(int key, int value) inserts a (key, value) pair into the HashMap. If the key already exists in the map, update the corresponding value. int get(int key) returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key. void remove(key) removes the key and its corresponding value if the map contains the mapping for the key.   Constraints: 0 <= key, value <= 106 At most 104 calls will be made to put, get, and remove.

 

# My Answer

class MyHashMap:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.hashmap=[]

    def put(self, key: int, value: int) -> None:
        """
        value will always be non-negative.
        """
        for k in self.hashmap:
            if k[0]==key:
                k[1] = value
                return 
        self.hashmap.append([key,value])

    def get(self, key: int) -> int:
        """
        Returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key
        """
        for k in self.hashmap:
            if k[0] ==key:
                return k[1]
        return -1

    def remove(self, key: int) -> None:
        """
        Removes the mapping of the specified value key if this map contains a mapping for the key
        """
        for k in self.hashmap:
            if k[0] == key:
                self.hashmap.remove(k)
                return 

문제에서 키가 같은 값이 들어왔을 때 업데이트를 하면 되서 충돌시 따로 구현하지 않아도 문제가 풀림

예시 답안에서는 개별 체이닝 기법으로 구현

 

# Solution 1 - 개별 체이닝 방식을 이용한 해시 테이블 구현

# Definition for singly-linked list.
class ListNode:
    def __init__(self, key=None, value=None):
        self.key = key
        self.value = value
        self.next = None


class MyHashMap:
    # 초기화
    def __init__(self):
        self.size = 1000
        self.table = collections.defaultdict(ListNode)

    # 삽입
    def put(self, key: int, value: int) -> None:
        index = key % self.size
        # 인덱스에 노드가 없다면 삽입 후 종료
        if self.table[index].value is None:
            self.table[index] = ListNode(key, value)
            return

        # 인덱스에 노드가 존재하는 경우 연결 리스트 처리
        p = self.table[index]
        while p:
            if p.key == key:
                p.value = value
                return
            if p.next is None:
                break
            p = p.next
        p.next = ListNode(key, value)

    # 조회
    def get(self, key: int) -> int:
        index = key % self.size
        if self.table[index].value is None:
            return -1

        # 노드가 존재할때 일치하는 키 탐색
        p = self.table[index]
        while p:
            if p.key == key:
                return p.value
            p = p.next
        return -1

    # 삭제
    def remove(self, key: int) -> None:
        index = key % self.size
        if self.table[index].value is None:
            return

        # 인덱스의 첫 번째 노드일때 삭제 처리
        p = self.table[index]
        if p.key == key:
            self.table[index] = ListNode() if p.next is None else p.next
            return

        # 연결 리스트 노드 삭제
        prev = p
        while p:
            if p.key == key:
                prev.next = p.next
                return
            prev, p = p, p.next

collections.defaultdict(ListNode)는 존재하지 않는 인덱스로 조회를 시도할 경우 디폴트 객체를 생성하기 때문에

self.table[index] is None 이 아니라 self.table[index].value is None 으로 value 값 비교