码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode_138_ Copy List with Random Pointer

时间:2015-02-16 11:45:05      阅读:191      评论:0      收藏:0      [点我收藏+]

标签:linked_list   c++   leetcode   

欢迎大家阅读参考,如有错误或疑问请留言纠正,谢谢技术分享


 Copy List with Random Pointer

A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.

Return a deep copy of the list.

算法分为三步:
1. copy and insert
2. set random pointer of copied node
3. split these two list


class Solution {
public:
    RandomListNode *copyRandomList(RandomListNode *head) {
        // Note: The Solution object is instantiated only once and is reused by each test case.
        if(head == NULL) return NULL;
        //1. copy and insert
        RandomListNode* pCurNode = head;
        while(pCurNode != NULL)
        {
            //copy
            RandomListNode* tmp = new RandomListNode(pCurNode->label);
            //insert
            RandomListNode* pNextNode = pCurNode->next;
            pCurNode->next = tmp;
            tmp->next = pNextNode;
            pCurNode = pNextNode;
        }
        //2. set random pointer of copied node
        pCurNode = head;
        while(pCurNode != NULL)
        {
            if(pCurNode->random != NULL)
                pCurNode->next->random = pCurNode->random->next;
            else
                pCurNode->next->random = NULL;
            pCurNode = pCurNode->next->next;
        }
        //3. split these two list
        RandomListNode* newHead = head->next;
        pCurNode = head;
        while(pCurNode != NULL)
        {
            RandomListNode* pCopiedNode = pCurNode->next;
            pCurNode->next = pCurNode->next->next;
            if(pCurNode->next != NULL)
                pCopiedNode->next = pCurNode->next->next;
            else 
                pCopiedNode->next = NULL;
            pCurNode = pCurNode->next;
        }
        
        return newHead;
    }
};


leetcode_138_ Copy List with Random Pointer

标签:linked_list   c++   leetcode   

原文地址:http://blog.csdn.net/keyyuanxin/article/details/43850917

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!