复杂链表的复制(c++&&go)

/*
输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，
另一个特殊指针random指向一个随机节点），请对此链表进行深拷贝，并返回拷
贝后的头结点。（注意，输出结果中请不要返回参数中的节点引用，否则判题程
序会直接返回空）。 下图是一个含有5个结点的复杂链表。图中实线箭头表示
next指针，虚线箭头表示random指针。为简单起见，指向null的指针没有画出。
*/


struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }
};

class Solution {
public:
    RandomListNode* Clone(RandomListNode* pHead) {
        if (pHead == nullptr)
        {
            return nullptr;
        }
        
        unordered_map<RandomListNode*, RandomListNode*> mp;
        //单链表的复制
        RandomListNode* pNewHead = new  RandomListNode(0);
        RandomListNode* curNode = pHead;
        RandomListNode* newCurNode = pNewHead;
        while (curNode)
        {
            RandomListNode* newNode = new RandomListNode(curNode->label);
            newCurNode->next = newNode;
            mp[curNode] = newNode;
            curNode = curNode->next;
            newCurNode = newCurNode->next;
        }

        for(auto var : mp)
        {
            var.second->random =  var.first->random == nullptr ? nullptr : mp[var.first->random];
        }

        return pNewHead->next;
    }
};

package main

type RandomListNode struct {
	Label  int
	Next   *RandomListNode
	Random *RandomListNode
}

/**
 *
 * @param pHead RandomListNode类
 * @return RandomListNode类
 */
func Clone(head *RandomListNode) *RandomListNode {
	if head == nil {
		return nil
	}

	mp := make(map[*RandomListNode]*RandomListNode, 0)
	newHead := new(RandomListNode)
	newCurNode := newHead
	curNode := head
	for curNode != nil {
		newNode := new(RandomListNode)
		newNode.Label = curNode.Label
		newCurNode.Next = newNode
		mp[curNode] = newNode
		curNode = curNode.Next
		newCurNode = newCurNode.Next
	}

	for key, value := range mp {
		if key.Random != nil {
			value.Random = mp[key.Random]
		} else {
			value.Random = nil
		}

	}

	return newHead.Next
}