复杂链表的复制

题目描述

输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)

分析

本题的Node多了一个random指针,比较复杂,可以分三步走:

  1. 在原链表中复制节点(在旧链表每个节点后插入一个跟它值相同的节点。如原来是A->B-C,复制完后是A->A’->B->B’->C->C’。复制的节点的random域暂不设置)
  2. 设置复制节点的random域(复制节点的random域为原节点的random域的next)
  3. 拆分链表(将复制的节点从原链表中拆出来)

代码:

/*
public class RandomListNode {
int label;
RandomListNode next = null;
RandomListNode random = null;
RandomListNode(int label) {
this.label = label;
}
}
*/
public class Solution {
public RandomListNode Clone(RandomListNode pHead) {
CloneNodes(pHead);
ConnectRandomNodes(pHead);
return rebuildNodes(pHead);
}
public void CloneNodes(RandomListNode pHead) {
RandomListNode p = pHead;
while (p != null) {
RandomListNode q = new RandomListNode(p.label);
q.next = p.next;
p.next = q;
p = q.next;
}
}
public void ConnectRandomNodes(RandomListNode pHead) {
RandomListNode p = pHead;
while(p != null) {
if (p.random != null)
p.next.random = p.random.next;
p = p.next.next;
}
}
public RandomListNode rebuildNodes(RandomListNode pHead) {
RandomListNode p = pHead;
RandomListNode q1 = null;
RandomListNode q2 = null;
if (p != null) {
q1 = p.next;
q2 = p.next;
}
while(p != null) {
p.next = p.next.next;
if(q2.next != null)
q2.next = q2.next.next;
p = p.next;
q2 = q2.next;
}
return q1;
}
}

欢迎关注公众号: FullStackPlan 获取更多干货

Copyright © 2016 - 2017 LBD's Blog All Rights Reserved.

访客数 : | 访问量 :