剑指 Offer - 25:复杂链表的复制

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/AnselLyy/article/details/85048100

题目描述

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

题目链接:https://www.nowcoder.com/practice/f836b2c43afc4b35ad6adc41ec941dba

解题思路

1、复制每个节点,到源结点的 next 处;
2、重新遍历链表,调整 cloneNode 的 random 指向;
3、拆分链表,返回复制链表

public class Solution {
    public RandomListNode Clone(RandomListNode pHead) {
        if (pHead == null) return pHead;
        
        RandomListNode curNode = pHead;
        
        // 1. 复制每个节点,到源结点的 next 处
        while (curNode != null) {
            RandomListNode cloneNode = new RandomListNode(curNode.label);
            RandomListNode nextNode = curNode.next;
            curNode.next = cloneNode;
            cloneNode.next = nextNode;
            curNode = nextNode;
        }
        
        // 2. 重新遍历链表,调整 cloneNode 的 random 指向
        curNode = pHead;
        while (curNode != null) {
            curNode.next.random = (curNode.random == null ? null : curNode.random.next);
            curNode = curNode.next.next;
        }
        
        // 3. 拆分链表
        curNode = pHead;
        RandomListNode cloned = pHead.next;
        while (curNode != null) {
            RandomListNode cloneNode = curNode.next;
            curNode.next = cloneNode.next;
            cloneNode.next = (cloneNode.next == null ? null : cloneNode.next.next);
            curNode = curNode.next;
        }
        return cloned;
    }
}

猜你喜欢

转载自blog.csdn.net/AnselLyy/article/details/85048100