图解 LeetCode 算法汇总——链表

本文首发公众号：小码A梦

一般数据主要存储的形式主要有两种，一种是数组，一种是链表。数组是用来存储固定大小的同类型元素，存储在内存中是一片连续的空间。而链表就不同于数组。链表中的元素不是存储在内存中可以是不连续的空间。

链表主要有两个元素：结点和指针。结点是存储数据，指针是指向下一个结点数据。

链表的有几个特点：

• 不必按顺序存储数据，使用指针关联各个结点的数据。
• 插入数据时间复杂度O(1)。
• 单链表查询时间复杂度为O(n)。

JDK 链表源码

JDK 很多地方用到了链表，这里举例两个：HashMap 和 LinkedList。

HashMap

HashMap 基于数组 + 链表/红黑树的数据结构，其中链表用 Node 表示。主要存储数据和指向下一个结点的指针，源码如下所示。

//HashMap Node 部分源码
static class Node<K,V> implements Map.Entry<K,V> {
    final int hash;
    final K key;
    V value;
    Node<K,V> next;

    Node(int hash, K key, V value, Node<K,V> next) {
        this.hash = hash;
        this.key = key;
        this.value = value;
        this.next = next;
    }
}

以上蓝色的 Node 结点表示一个链表结点。

LinkedList

LinkedList 本身就是一个双链表，有前置指针和后置指针。

//LinkedList 部分源码
private static class Node<E> {
    E item;
    Node<E> next;
    Node<E> prev;

    Node(Node<E> prev, E element, Node<E> next) {
        this.item = element;
        this.next = next;
        this.prev = prev;
    }
}

LeetCode 题解

LeetCode 有很多链表的题目，选取了几种比较典型的链表，面试的时候也经常会问到。

206.反转链表(简单)

题目描述

解法

链表结点不变，将指针方向反转。如果只是在一个链表的上操作，每次都需要遍历结点，比较繁琐。所以添加一个空链表 pre。pre 表示将指针向后，改成向前。比如 1 -> 2 变成 2 -> 1，先将 1 放在新的链表上，然后再将 2 放入新的链表，并指向 1。如下图所示：

总结一下解决步骤：

• 设置一个空链表pre
• 遍历链表，每次先将链表的next，赋值给一个新的链表。
• 链表的next指向pre。
• 当前链表赋值给pre。
• 继续遍历next链表

整理好解题思路后，根据解题写出以下代码：

class Solution {
    public ListNode reverseList(ListNode head) {
        ListNode cur = head;
        ListNode pre = null;
        while(cur != null) {
            ListNode next = cur.next;
            cur.next = pre;
            pre = cur;
            cur = next;
        }
        return pre;

    }
}

19.删除链表的倒数第 N 个结点（中等）

题目描述

解法

类似链表删除结点，都是修改指针指向的结点，比如删除 4 结点，就需要将 4 节点之前的结点的指针，指向 4 的下一个结点。

解题的关键点就是找到结点的位置，并修改指针的指向，使用 node.next = node.next.next 即可修改指针的指向。找到结点的位置需要找到链表的长度，然后遍历链表，找到对应结点并修改指针。如果移除的是首结点，就找不到上一个结点，所以需要设置一个伪结点。总结一下解题思路：

• 创建伪结点，next 指向链表。首结点就是一个为伪结点。
• 遍历链表获取链表长度。
• 获取结点位置就是 L-n+1。
• 遍历链表，找到前一个结点的位置，也就是 L-n的结点的 next 指向 next.next.返回伪结点的 next 结点。

有了思路，写代码就快了：

class Solution {
    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode cur = head;
		int length = 0;
		while(cur != null) {
			cur = cur.next;
			length++;
		}
		int index = 1;
		int sub = length - n + 1;
		ListNode dummy = new ListNode(1,head);
		cur = dummy;
		while(cur != null) {
			if(index == sub) {
				cur.next = cur.next.next;
				break;
			}
			cur = cur.next;
			index++;
		}
		return dummy.next;
    }
}

141.环形链表（简单）

题目描述

解法

环形链表表示链表的某个结点可以通过连续 next 指针再次被遍历到。所以需要做个查重的功能，如果存在重复的结点，就说明链表是一个环形链表。而查重使用 hash 表即可，汇总一下解题思路：

• 遍历结点，将结点添加到 hash 表中。
• 如果在添加数据时发现结点已存在，说明链表是环形链表。

public class Solution {
    public boolean hasCycle(ListNode head) {
        Set<ListNode> set = new HashSet<>();
		while (head != null) {
			if (!set.add(head)) {
				return true;
			}
			head = head.next;
		}
        return false;
    }
}

set 的 add 的方法判断是否存在相同的元素，其实是使用 HashMap 的 put 方法，返回之前的数据，如果之前数据不存在就返回 null。

21.合并两个有序链表（简单）

题目描述

解法

单链表查找结点是需要一个一个往后指针找结点，合并链表的，如果在原来的基础上添加结点就需要不断的遍历结点，合并的链表使用一个新的链表存储。首先找到最小的数据，然后结点存储到新链表，同时断开结点的 next。如下图所示。

先找到最小的结点，然后新链表（蓝色部分）指向小的结点，然后断开 next 指针。之后的重复类似的步骤。

需要考虑到如果某个链表遍历结束了，但是其他链表还有值，就需要遍历非空链表的数据了，最终总结题解的步骤：

• 定义一个新的链表 node，存储合并的链表。
• 找到最小的结点，新的链表的next 指向新的结点，并将找到的链表的 next 断开。
• 重复遍历链表，找到最小的节点，直到某个链表为空。
• 一个链表为空，另外不为空，就将链表直接赋值个新链表。

代码如下：

class Solution {
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode node = new ListNode(0);
		ListNode last = node;
		while (l1 != null && l2 != null) {
			if (l1.val < l2.val) {
				last.next = l1;
				l1 = l1.next;
			} else {
				last.next = l2;
				l2 = l2.next;
			}
			last = last.next;
		}
		if (l1 != null) {
			last.next = l1;
		}
		if (l2 != null) {
			last.next = l2;
		}
		return node.next;
    }
}

总结

链表相对数组来说，在定义的时候不需要确定好内存大小，因为链表的通过指针可以存储非连续空间的数据。链表的查询需要通过指针一个个找到对应的结点，而不能像数组一样通过索引直接找到对应的数据。在 JDK 中也会使用双链表，双链表更占内存，因为多了一个指针，带来的好处是查询数据可以从前往后查，也可以从后往前查，用空间换时间。

主要介绍了几种常见的链表算法：

• 反转链表
  • 反转指针方向，结点数据不会改变。 拆分一下数据，1—>2,先将 1 取出来，将 1 作为新链表的头结点，然后再将 2 取出，作为头结点，并指向新的 1，此时就完成 2 -> 1 的过程，后续结点以此类推。
• 删除链表的倒数第 N 个结点
  • 关键点事要找到要删除的位置，倒数第 N 的节点，也就是链表长度 L-n+1,再修改指针，将自己上一个结点的指针指向自己节点的下一个结点。
• 环形链表
  • 环形链表表示存在相同的链表结点，遍历链表，存储在 hash 表，如果存在相同的结点就是环形链表。
• 合并两个有序链表
  • 先找到两个链表最小的结点，新的链表指向最小结点，对应的链表去掉指向的结点，以此类推，直到遍历完某个链表。此时如果别的链表不为空，直接指向该链表。