Go语言实现:【剑指offer】复杂链表的复制

该题目来源于牛客网《剑指offer》专题。

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

Go语言实现:

type RandomListNode struct {
	Val        int
	Next       *RandomListNode
	RandomNext *RandomListNode
}

func clone(head *RandomListNode) *RandomListNode {
   if head == nil {
      return nil
   }

   //从A开始
   currentNode := head
   //1.复制结点A得到A1,A1放到A后面
   for currentNode != nil {
      //先存一下,nextNode现在为B
      nextNode := currentNode.Next
      cloneNode := new(RandomListNode)
      cloneNode.Val = currentNode.Val
      cloneNode.Next = nextNode
      currentNode.Next = cloneNode
      //A处理完,处理B
      currentNode = nextNode
   }

   //从A开始
   currentNode = head
   //2.复制Random,A到C,A1到C1
   for currentNode != nil {
      if currentNode.RandomNext == nil {
         currentNode.Next.RandomNext = nil
      } else {
         currentNode.Next.RandomNext = currentNode.RandomNext.Next
      }
      //A处理完,处理B
      currentNode = currentNode.Next.Next
   }

   //从A开始
   currentNode = head
   pCloneHead := head.Next
   //3.拆分,ABC,A1B1C1
   for currentNode != nil {
      //A1
      cloneNode := currentNode.Next
      currentNode.Next = cloneNode.Next
      if cloneNode.Next != nil {
         cloneNode.Next = cloneNode.Next.Next
      }
      //A处理完,处理B
      currentNode = currentNode.Next
   }
   return pCloneHead
}
posted @ 2019-12-03 16:44  南方有嘉木1993  阅读(161)  评论(0编辑  收藏  举报