LinkedHashMap 学习（手写）

LinkedHashMap

我们先来总结一下HashMap的特点：

• HashMap是查询效率最高的数据结构（O(1)级别）
• HashMap存储元素是无序的

如果我们想要按照添加元素的顺序遍历，显然HashMap是达不到我们的要求的，TreeMap可以满足我们的要求，但是效率没有HashMap高

接下来介绍一种新的数据结构——LinkedHashMap

LinkedHashMap就是在HashMap的基础上维护元素的添加顺序，使得遍历的结果是遵从添加顺序的

1. 实现添加逻辑

其实我们实现也很简单，就是在添加元素的时候用指针将每个添加的元素串起来就好了
这里的HashMap是笔者之前手写的，可以在笔者之前的代码中拷贝一下

/**
 * <p>
 * LinkedHashMap,红黑树上面的所有结点都用线连接起来了
 * </p>
 *
 * @since: 2022/7/21 9:35
 * @author: Eureka
 */
public class LinkedHashMap<K, V> extends HashMap<K, V> {

    /* 需要记录链表的头尾结点 */
    private LinkedNode<K, V> first;
    private LinkedNode<K, V> last;

    /**
     * LinkedHashMap结点
     */
    protected static class LinkedNode<K, V> extends Node<K, V> {
        // 指向上一个结点的指针和指向下一个结点的指针
        LinkedNode<K, V> prev;
        LinkedNode<K, V> next;

        public LinkedNode(K key, V value, Node<K, V> parent) {
            super(key, value, parent);
        }
    }

    @Override
    public void forEach(BiConsumer<? super K, ? super V> action) {
        // 从头结点开始遍历
        LinkedNode<K, V> node = first;
        while (node != null) {
            action.accept(node.key, node.value);
            node = node.next;
        }
    }

    @Override
    public void traversal(Visitor<K, V> visitor) {
        if (visitor == null) return;
        LinkedNode<K, V> node = first;
        while (node != null) {
            if (visitor.visit(node.key, node.value)) return;
            node = node.next;
        }
    }

    /**
     * 我们需要在创建结点的时候用指针将其串起来
     */
    @Override
    protected Node<K, V> createNode(K k, V v, Node<K, V> parent) {
        LinkedNode<K, V> node = new LinkedNode<>(k, v, parent);
        // 添加头结点
        if (first == null) {
            first = last = node;
        } else {
            // 非头结点
            last.next = node;
            node.prev = last;
            last = node;
        }
        return node;
    }

    /**
     * 清空结点需要将first和last去掉，不然不会进行GC
     */
    @Override
    public void clear() {
        super.clear();
        first = null;
        last = null;
    }
}

测试一下：

@Test
public void test01() {
    Person p1 = new Person(18, 650, "张三");
    Person p2 = new Person(18, 65, "张三");
    Person p3 = new Person(23, 65, "李四");
    Map<Person, String> map = new LinkedHashMap<>();
    map.put(p1, "张三");
    map.put(p2, "李四");
    map.put(p3, "王五");
    map.forEach((k, v) -> System.out.println(v));
}

结果：

张三
李四
王五

可以发现，结果确实有序了😄

2. 删除逻辑

添加做完了，接下来我们来看删除的逻辑，这里的删除逻辑会复杂一些，因为在删除的时候我们不仅要处理红黑树的逻辑，还要处理链表的逻辑，我们可以根据之前的代码然后添加删除链表的逻辑

/**
 * 这里并不是删除结点后修复红黑树性质的代码，而是修复链表和红黑树结点指向关系的代码
 */
@Override
protected void afterRemove(Node<K, V> removedNode) {
    if (removedNode == null) return;
    LinkedNode<K, V> linkedNode = (LinkedNode<K, V>) removedNode;
    LinkedNode<K, V> prev = linkedNode.prev;
    LinkedNode<K, V> next = linkedNode.next;
    if (prev == null) {
        // 是头结点，那么下一个结点成为头结点
        first = next;
    } else {
        prev.next = next;
    }
    if (next == null) {
        // 删除的是尾结点
        last = prev;
    } else {
        next.prev = prev;
    }
}

但是上面移除链表结点的代码并不是完全正确，我们来看下面的这种删除情况，例如我们要删除结点83

我们会发现，站在红黑树的角度上来看，删除83结点其实是删除它的前驱或者后继结点，也就是删除结点95，但是站在链表的角度来看，删除结点83就是删除结点83

在我们删除度为2的结点的时候就会产生冲突，所以我们上面的链表结点删除代码中逻辑其实是有问题的

那么我们如果解决红黑树与链表之间的冲突的呢？

其实做法也很简单，我们只需要将红黑树的后继结点与要删除的结点在链表中的顺序交换即可（改变的是指针的指向）

我们来简单演示 一下这个过程，假设我现在要删除结点31

• 从红黑树的角度，其实需要交换结点31和结点37的值，然后将结点37从内存中抹去
• 在链表的角度，之前链表结构为：52 -> 37 -> 21 -> 31 -> 41，当我们交换两个结点在链表中的顺序后为：52 -> 31 -> 21 -> 37 -> 41，注意这里并没有交换结点的位置，只是交换了结点在链表中的指向，可以看到当我们的红黑树删除结点31的时候，链表的顺序并没有收到影响

那么到底如何实现呢？我们可以改造一下之前在HashMap的afterRemove函数，处理一下度为2的结点在红黑树中删除的逻辑

/**
 * 删除结点后的操作，注意这里不是修复红黑树的性质
 *
 * @param willNode    红黑树中想要删除的结点
 * @param removedNode 红黑树中实际要删除的结点
 */
protected void afterRemove(Node<K, V> willNode, Node<K, V> removedNode) {
}

在红黑树删除结点的时候，将刚刚传进来的结点当做willNode

/**
  * 根据结点删除该结点
  */
protected V remove(Node<K, V> node) {
    if (node == null) return null;
    // 开始想要删除的结点,此处是为了处理linkedHashMap的删除逻辑
    Node<K, V> willNode = node;
    ....其他代码
}

接着我们只需要在子类中去处理交换链表中结点的逻辑，交换链表中的两个结点的位置，这个应该是比较容易的

先处理所有指向要交换结点的指针

/**
 * 这里并不是删除结点后修复红黑树性质的代码，而是修复链表和红黑树结点指向关系的代码
 */
@Override
protected void afterRemove(Node<K, V> willNode, Node<K, V> removedN
    LinkedNode<K, V> linkedWillNode = (LinkedNode<K, V>) willNode;
    LinkedNode<K, V> linkedRemoveNode = (LinkedNode<K, V>) removedN
    // 如果想要删除的实际要删除的结点不一样，表示该结点为红黑树中度为2的结点，需要交换链表的指向
    if (linkedRemoveNode != linkedWillNode) {
        // 设置中间结点,处理prev
        LinkedNode<K, V> temp = linkedWillNode.prev;
        linkedWillNode.prev = linkedRemoveNode.prev;
        linkedRemoveNode.prev = temp;
        if (linkedWillNode.prev == null) {
            first = linkedWillNode;
        } else {
            linkedWillNode.prev.next = linkedWillNode;
        }
        if (linkedRemoveNode.prev == null) {
            first = linkedRemoveNode;
        } else {
            linkedRemoveNode.prev.next = linkedRemoveNode;
        }
        // 处理next
        temp = linkedWillNode.next;
        linkedWillNode.next = linkedRemoveNode.next;
        linkedRemoveNode.next = temp;
        if (linkedWillNode.prev == null) {
            last = linkedWillNode;
        } else {
            linkedWillNode.next.prev = linkedWillNode;
        }
        if (linkedRemoveNode.next == null) {
            last = linkedRemoveNode;
        } else {
            linkedRemoveNode.next.prev = linkedRemoveNode;
        }
    }
    // 后面的逻辑省略，和上面的是一样的
}

3. LinkedHashMap源码分析

我们通过LinkedHashMap的源码可以看到，基本上逻辑和我们的差不多，主要思路也就是用双向链表将添加的结点串起来