add solution 146 [solution.java]

Author: zhng1456

solution/146.Lru Cache/Solution.java

@@ -0,0 +1,88 @@
+//双向链表的节点
+class Node{
+    public int key;
+    public int val;
+    public Node pre;//指向前面的指针
+    public Node next;//指向后面的指针
+    public Node(int key,int value){
+        this.val=value;
+        this.key=key;
+    }
+}
+class LRUCache {
+    int capacity;//容量
+    Node head;//双向链表的头，维护这个指针，因为set,get时需要在头部操作
+    Node end;//双向链表的尾，set时，要是满了，需要将链表的最后一个节点remove
+    HashMap<Integer,Node> map=new HashMap<Integer,Node>();//hash表
+    public LRUCache(int capacity) {
+        this.capacity=capacity;
+    }
+    //添加，删除尾部，插入头部的操作
+    public void remove(Node node){
+        Node cur=node;
+        Node pre=node.pre;
+        Node post=node.next;
+        if(pre==null){//说明cur是头部节点
+            head=post;
+        }
+        else pre.next=post;//更新指针，删除
+        if(post==null){//说明cur是最后的节点
+            end=pre;
+        }
+        else post.pre=pre;
+    }
+    public void setHead(Node node){
+        //直接插入
+        node.next=head;
+        node.pre=null;
+        if(head!=null) head.pre=node;//防止第一次插入时为空
+        head=node;
+        if(end==null) end=node;
+    }
+    public int get(int key) {
+        if(map.containsKey(key)){
+            //需要把对应的节点调整到头部
+            Node latest=map.get(key);
+            remove(latest);
+            setHead(latest);
+            //返回value
+            return latest.val;
+        }
+        else return -1;
+    }
+    
+    public void put(int key, int value) {
+        if(map.containsKey(key)){//这个key原来存在
+            //只需要把key对应的node提到最前面，更新value
+            Node oldNode=map.get(key);
+            oldNode.val=value;
+            remove(oldNode);
+            setHead(oldNode);
+        }
+        else{
+            //这个key原来不存在，需要重新new出来
+            Node newNode=new Node(key,value);
+            //接下来要考虑容量
+            if(map.size()<capacity){
+                setHead(newNode);
+                map.put(key,newNode);
+            }
+            else{
+                //容量不够，需要先将map中,最不常使用的那个删除了删除
+                map.remove(end.key);
+                //接下来更新双向链表
+                remove(end);
+                setHead(newNode);
+                //放入新的
+                map.put(key,newNode);
+            }
+        }
+    }
+}
+
+/**
+ * Your LRUCache object will be instantiated and called as such:
+ * LRUCache obj = new LRUCache(capacity);
+ * int param_1 = obj.get(key);
+ * obj.put(key,value);
+ */