剑指offer35----复制复杂链表
题目:
请实现一个cloneNode方法,复制一个复杂链表。
在复杂链表中,每个结点除了有一个next指针指向下一个结点之外,还有一个random指向链表中的任意结点或者NULL。
结点的定义如下:
public static class ListNode { int val; ListNode next; ListNode random; ListNode(int x) { val = x; } }
思路:
方法1:
复制原始链表上的每一个结点,并通过next连接起来;然后再设置每个结点的random指针。
假设原始链表中某个结点N的random指针指向结点S,那么就需要从头到尾遍历查找结点S,如果从原始链表的头指针开始,经过m步之后达到结点S,那么在复制链表中的结点N'的random指针指向的结点也是距离复制链表s步的结点。通过这种办法就可以为复制链表上的每个结点设置random指针。
时间复杂度:O(N^2)
方法2:
方法1是通过链表查找来得到random指针所指向的结点,实际上我们可以通过空间换取时间,将原始链表和复制链表的结点通过哈希表对应起来,这样查找的时间就从O(N)变为O(1)。具体如下:
复制原始链表上的每个结点N创建N',然后把这些创建出来的结点用pNext连接起来。同时把<N,N'>的配对信息方法一个哈希表中;然后设置复制链表中的每个结点的random指针,如果原始链表中结点N的random指向结点S,那么在复制链表中,对应的N'应该指向S'。
时间复杂度:O(N)
根据方法2,我们可以写出代码:
private static ListNode cloneNode(ListNode head2) { if (head2 == null) { return null; } Map<ListNode, ListNode> map1 = new HashMap<>(); ListNode newHead = new ListNode(head2.val); map1.put(head2, newHead); ListNode head = newHead; // 复制节点和next指针 for (ListNode cur = head2.next; cur != null; cur = cur.next) { head.next = new ListNode(cur.val); head = head.next; map1.put(cur, head); } head = newHead; // 复制random节点 ListNode ranNode = null; for (ListNode cur = head2; cur != null; cur = cur.next) { ranNode = map1.get(cur.random); head.random = ranNode; head = head.next; } return newHead; }
方法3:
在不使用辅助空间的情况下实现O(N)的时间效率。
第一步:根据原始链表的每个结点N创建对应的N',然后将N‘通过next接到N的后面;
第二步:设置复制出来的结点的random。假设原始链表上的N的random指向结点S,那么其对应复制出来的N'是N->pNext指向的结点,同样S'也是结点S->next指向的结点。
第三步:把长链表拆分成两个链表,把奇数位置的结点用next连接起来的就是原始链表,把偶数位置的结点通过next连接起来的就是复制链表。
如图:
根据方法3,我们可以写出代码:
// 不要额外空间的时间复杂度为O(N) private static ListNode cloneNode2(ListNode head2) { if (head2 == null) { return null; } // 复制节点和next指针 ListNode cur = head2; while (cur != null) { ListNode prev = cur.next; ListNode clone = new ListNode(cur.val); cur.next = clone; cur.next.next = prev; cur = prev; } ListNode old = head2; ListNode clone = old.next; while (clone != null) { if (old.random != null) { clone.random = old.random.next; } old = clone.next; if (old != null) { clone = old.next; } else { break; } } ListNode res = oddEvenList(head2); return res; }
上面的代码可以得到图4.10的结构,而这个时候我们希望把链表分开,老的节点和复制的节点各分为一边,于是乎,出现了奇偶节点的变换(LeetCode328题):
public static ListNode oddEvenList(ListNode head) { if (head == null || head.next == null) { return head; } ListNode oddNode = head; ListNode evenNode = head.next; ListNode cur = evenNode.next; ListNode evenTemp = evenNode; int index = 3; while (cur != null) { if (index % 2 == 0) { evenNode.next = cur; evenNode = evenNode.next; } else { oddNode.next = cur; oddNode = oddNode.next; } cur = cur.next; index++; } evenNode.next = null; return evenTemp; }
返回的evenTemp节点就是我们复制链表的第一个节点!