Leetcode 138. 复制带随机指针的链表 c语言

Leetcode 138. 复制带随机指针的链表 c语言

问题描述

给定一个链表，每个节点包含一个额外增加的随机指针，该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的 深拷贝。

我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示：

• val：一个表示 Node.val 的整数。
• random_index：随机指针指向的节点索引（范围从 0 到 n-1）；如果不指向任何节点，则为 null 。

示例一

输入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出：[[7,null],[13,0],[11,4],[10,2],[1,0]]

示例二：

输入：head = [[1,1],[2,1]]
输出：[[1,1],[2,1]]

示例三：

输入：head = [[3,null],[3,0],[3,null]]
输出：[[3,null],[3,0],[3,null]]

示例四：

输入：head = [ ]
输出：[ ]
解释：给定的链表为空（空指针），因此返回 null。

提示：

• 10000 <= Node.val <= 10000
• Node.random 为空（null）或指向链表中的节点。
• 节点数目不超过 1000 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/copy-list-with-random-pointer

简要思路

先顺序复制链表，再复制随机指针。
代码如下：

/**
 * Definition for a Node.
 * struct Node {
 *     int val;
 *     struct TreeNode *next;
 *     struct TreeNode *random; 
 * };
 */

struct Node* copyRandomList(struct Node* head) 
{
	if(head == NULL)
    {
        return NULL;
    }
    
    struct Node *q = head, *rq = NULL, *tool = head;
    int i = 0;
    struct Node* p = (struct Node*)malloc(sizeof(struct Node));
    p->next = NULL;
    p->random = NULL;
    p->val = q->val;
    struct Node* cat = p;
    
    while(q->next != NULL)
    {
        q = q->next;        
        p->next = (struct Node*)malloc(sizeof(struct Node));
        p = p->next;
        p->next = NULL;
        p->random = NULL;
        p->val = q->val;
    }
    
    q = head;
    p = cat;
    while(q != NULL)
    {
        tool = head;
        rq = q->random;
        if(rq != NULL)
        {
            while(tool != rq)
            {
                i++;
                tool = tool->next;
            }
            tool = cat;
            while(i > 0)
            {
                tool = tool->next;
                i--;
            }
            p->random = tool;
        }
        q = q->next;
        p = p->next;
    }
    
    return cat;
}