SkipList

Redis中使用跳表作为有序集合键(zset)的底层实现之一。当有序集合元素较多时，或有序集合中元素为较长字符串时，都会使用跳表作为底层结构。

Redis在两个地方用到了跳表，一种是有序集合键中，另一种是集群节点中做内部数据结构。

 

跳表

下面的结构是就是跳表：

 其中 -1 表示 INT_MIN， 链表的最小值，1 表示 INT_MAX，链表的最大值。

 

跳表具有如下性质：

(1) 由很多层结构组成

(2) 每一层都是一个有序的链表

(3) 最底层(Level 1)的链表包含所有元素 越接近顶层的链表，含有的节点则越少  （一般是做为索引层）

(4) 如果一个元素出现在 Level i 的链表中，则它在 Level i 之下的链表也都会出现。

(5) 每个节点包含4个指针，一个指向同一链表中的下一个元素，一个指向下面一层的元素。一个指向上面一层的元素，一个指向同一链表中的上一个元素

 

 

package com.high_structure;

import java.util.Random;

public class MySkipList<K extends Comparable<K>, V> {
	// 该类应该的有的属性 size head节点
	private int size;
	private Node<K, V> head;

	// 因为我们会通过随机数概率的方式去生成是否生成索引层
	private final Random random = new Random();
	//
	private final double DEFAULT_PROBABILITY = 0.5;

	public MySkipList() {
		super();
		this.size = 0;
		this.head = new Node<K, V>(null, null, 0);
	}

	// 提供几个基本方法 方便后期去操作链表 1 判断是否为空 2水平插入到某个元素的后面 3 垂直插入（上下关联）

	public boolean isEmpty() {
		return size == 0;
	}

	/**
	 * 吧b插入到a的后面<br>
	 * a->c->d <br>
	 * b
	 * 
	 * @param a
	 * @param b
	 */
	public void horizontalLink(Node<K, V> a, Node<K, V> b) {
		b.setPre(a);
		b.setNext(a.getNext());
		if (a.getNext() != null) // 建立从右到左的链接关系
		{
			// 在这里a.getnext == b。getnext的
			a.getNext().setPre(b);
		}
		a.setNext(b);
	}

	/**
	 * 吧b插入到a下面
	 * 
	 * @param a
	 * @param b
	 */
	public void vertLink(Node<K, V> a, Node<K, V> b) {
		a.setDown(b);
		b.setUp(a);
	}

	public static void main(String[] args) {
		MySkipList<Integer, String> skipList = new MySkipList<>();
		skipList.add(151, "1");
		skipList.add(12, "12");

		skipList.add(121, "121");
		skipList.add(112, "112");
		skipList.add(11, "11");
		skipList.remove(12);
		skipList.remove(121);
		skipList.remove(15);
		System.out.println(skipList);
	}

	/**
	 * 判断key1是否小于key2
	 * 
	 * @param key1
	 * @param key2
	 * @return
	 */
	public boolean lessthanOrEquals(K key1, K key2) {
		return key1.compareTo(key2) <= 0;
	}

	public Node<K, V> get(K key) {
		// 省略key的判断
		Node<K, V> node = searchNode(key);
		if (node.getKey().equals(key)) {
			return node;
		}
		return null;
	}

	/**
	 * 通过指定的key查找元素
	 * 
	 * @param key
	 * @return
	 */
	public Node<K, V> searchNode(K key) {
		// 该node表示最后返回的节点
		Node<K, V> node = head;
		// 该节点表示遍历会用到的节点
		Node<K, V> next = null;
		Node<K, V> down = null;
		// 先在从head开始查找

		while (true) {
			next = node.getNext();
			// 先遍历第一层节点
			while (next != null && lessthanOrEquals(next.getKey(), key))
			// 如果不为空 并且遍历的节点<=当前节点
			{
				node = next;
				next = next.getNext();
			}
			if (node.getKey() != null && node.getKey().compareTo(key) == 0) {
				break;
			}

			// 如果不相等 进入下一层级
			down = node.getDown();
			if (down == null) // 最后一层
			{
				break;
			} else {
				node = down;
			}
		}
		return node;
	}

	/**
	 * 添加元素 如果 key存在 就修改该node的value 如果不存在 就添加到他应该的位置
	 * 
	 * @param node
	 */
	public void add(K key, V val) {
		// 省略判断key是否合法
		Node<K, V> node = searchNode(key);
		K searchKey = node.getKey();
		if (searchKey != null && searchKey.compareTo(key) == 0) {
			node.setValue(val);
			return;
		}
		// 如果不相等 那么我们就把我们的节点插入到该节点的后面
		Node<K, V> newNode = new Node<K, V>(key, val, node.getLevel());
		horizontalLink(node, newNode);
		// 通过随机函数的的方式看是否需要生成索引层 【如果查找的数和我们 的head在同一层在同一级就需要生成 】
		int headLevel = head.getLevel();
		int currentLevel = node.getLevel();
		while (isNeedBuildLevel()) {
			if (currentLevel >= headLevel) { // 需要创建的心的索引层头结点
				Node<K, V> newHead = new Node<K, V>(null, null, ++headLevel);
				vertLink(newHead, head);
				head = newHead;
			}
//			结合跳表的图更好理解
//			如果当前节点（node）的up为空  我们就跳转到当前节点的head节点 然后 获取up
//			如果不为空直接获取up 然后在up后面添加当前节点作为索引层
			while (node.getUp() == null) {
				node = node.getPre();
			}
			node = node.getUp();
			Node<K, V> newTemp = new Node<K, V>(key, val, node.getLevel());
			horizontalLink(node, newTemp);
			vertLink(newTemp, newNode);

　　　　　　　　　　　　　　//因为我们生成索引层是随机的 有可能在生成了一个索引层以后又会生成一个
　　　　　　　　　　　　　　//比如 现在有2层
　　　　　　　　　　　　　　// leav1 0--> 1
　　　　　　　　　　　　　　//                  ||
　　　　　　　　　　　　　　// leav0 0--> 1  假如又生成一个新的索引层 我们其实是需要记录上一次索引层的位置 也就是 tmp的值 我们把他保存在newNode 下次使用

			newNode = newTemp;
			currentLevel++;
		}
		size++;
	}

	/**
	 *
	 * 删除的核心思路 先找到该节点 然后走到最下面 从下面开始删除 （从上往下删除 ）
	 *
	 * @param key
	 */
	public void remove(K key) {
		// 省略判断key是否合法
		Node<K, V> node = searchNode(key);
		// 删除
		if (node != null && node.getKey().equals(key)) {
			while (node.getDown() != null) {
				node = node.getDown();
			}

			Node<K, V> next = null;

			Node<K, V> pre = null;
			while (node != null) {
				pre = node.getPre();
				next = node.getNext();
				if (pre != null) {
					pre.setNext(next);
				}
				if (next != null) {
					next.setPre(pre);
				}
				node = node.getUp();
			}
			// 对顶层链表进行调整，去除无效的顶层链表 某些时候可能存在某一行索引就一个单独的头结点 该头结点的next没有值
			while (head.getNext() == null && head.getDown() != null) {
				head = head.getDown();
				head.setUp(null);
			}
		}
		size--;
	}

	@Override
	public String toString() {
		StringBuilder sb = new StringBuilder();
		Node<K, V> node = head;

		// 移动到最底层
		while (node.getDown() != null)
			node = node.getDown();

		while (node.getPre() != null)
			node = node.getPre();

		// 第一个节点是头部节点，没有任何意义，所以需要移动到后一个节点
		if (node.getNext() != null)
			node = node.getNext();

		// 遍历
		while (node != null) {
			sb.append(node.toString()).append("\n");
			node = node.getNext();
		}

		return sb.toString();
	}

	private boolean isNeedBuildLevel() {
		return random.nextDouble() < DEFAULT_PROBABILITY;
	}

	// 节点类型 采用内部类的方式
	static class Node<K, V> {
		private K key;
		private V value;
		// 还需要一个值来保存该节点所在的层级
		private int level;

		public int getLevel() {
			return level;
		}

		public void setLevel(int level) {
			this.level = level;
		}

		private Node<K, V> up, down, next, pre;

		public K getKey() {
			return key;
		}

		public void setKey(K key) {
			this.key = key;
		}

		public V getValue() {
			return value;
		}

		public void setValue(V value) {
			this.value = value;
		}

		public Node<K, V> getUp() {
			return up;
		}

		public void setUp(Node<K, V> up) {
			this.up = up;
		}

		public Node<K, V> getDown() {
			return down;
		}

		public void setDown(Node<K, V> down) {
			this.down = down;
		}

		public Node<K, V> getNext() {
			return next;
		}

		public void setNext(Node<K, V> next) {
			this.next = next;
		}

		public Node<K, V> getPre() {
			return pre;
		}

		public void setPre(Node<K, V> pre) {
			this.pre = pre;
		}

		public Node(K key, V value, int level) {
			super();
			this.key = key;
			this.value = value;
			this.level = level;
		}

		public Node() {

		}

		@Override
		public String toString() {
			return "Node [key=" + key + ", value=" + value + "]";
		}

	}
}

最后的测试结果如下