146. LRU Cache
基本想法就是用一個(gè)雙向鏈表加上一個(gè)hash(key, node)魂毁， 節(jié)點(diǎn)的val等于相應(yīng)key的value玻佩。當(dāng)get一個(gè)key的時(shí)候，把相應(yīng)的節(jié)點(diǎn)刪除席楚，并且放到鏈表的最后去咬崔。當(dāng)put一個(gè)key，value的時(shí)候烦秩，如果key在hash里就先刪除key和相應(yīng)的node垮斯，然后創(chuàng)建新的key和node并放到鏈表的最后。如果此時(shí)鏈表的大小超過了capacity只祠，就把鏈表的第一個(gè)值pop出去兜蠕。雖然在python里用ordereddict做更加快捷，但是下面的解法更是體現(xiàn)了這個(gè)思想铆农。
class Node:
def __init__(self, k, v):
self.key = k
self.val = v
self.prev = None
self.next = None
class LRUCache:
def __init__(self, capacity):
self.capacity = capacity
self.dic = dict()
self.head = Node(0, 0)
self.tail = Node(0, 0)
self.head.next = self.tail
self.tail.prev = self.head
def get(self, key):
if key in self.dic:
n = self.dic[key]
self._remove(n)
self._add(n)
return n.val
return -1
def set(self, key, value):
if key in self.dic:
self._remove(self.dic[key])
n = Node(key, value)
self._add(n)
self.dic[key] = n
if len(self.dic) > self.capacity:
n = self.head.next
self._remove(n)
del self.dic[n.key]
def _remove(self, node):
p = node.prev
n = node.next
p.next = n
n.prev = p
def _add(self, node):
p = self.tail.prev
p.next = node
self.tail.prev = node
node.prev = p
node.next = self.tail
