138. 复制带随机指针的链表

Node* copyRandomList(Node* head) {

        std::map<Node *, int> node_map;

        std::vector<Node *> node_vec;

        Node *ptr = head;

        int i = 0;

        while (ptr){

            node_vec.push_back(new Node(ptr->val));

            node_map[ptr] = i;

            ptr = ptr->next;

            i++;

        }

        node_vec.push_back(0);

        ptr = head;

        i = 0;

        while(ptr){

            node_vec[i]->next = node_vec[i+1];

            if (ptr->random){

                int id = node_map[ptr->random];

                node_vec[i]->random = node_vec[id];

            }

            else

                node_vec[i]->random = NULL;

            ptr = ptr->next;

            i++;

        }

        return node_vec[0];

    }