复杂链表的复制

2018-12-07  本文已影响0人  twilight_mao

题目描述

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

思路

1.复制链表,得new新节点,值等于旧链表节点的值。
2.新的下一个节点及任意节点可能已经存在了。

代码

/*
public class RandomListNode {
    int label;
    RandomListNode next = null;
    RandomListNode random = null;

    RandomListNode(int label) {
        this.label = label;
    }
}
*/
public class Solution {
   public RandomListNode Clone(RandomListNode pHead) {
        if (pHead == null) {
            return pHead;
        }
        RandomListNode qHead = new RandomListNode(pHead.label);
        fun(pHead, qHead, pHead, qHead);
        return qHead;
    }

    public void fun(RandomListNode pHead, RandomListNode qHead, RandomListNode p, RandomListNode q) {
        if (p == null || q == null) {
            return;
        }
        if (p.next != null) {
            RandomListNode node1 = find(qHead, p.next.label);
            q.next = node1;
        }
        if (p.random != null) {
            RandomListNode node2 = find(qHead, p.random.label);
            q.random = node2;
        }
        fun(pHead, qHead, p.next, q.next);
    }

    private RandomListNode find(RandomListNode head, int val) {
        RandomListNode node = head;
        while (node != null) {
            if (node.label == val) {
                return node;
            }
            if (node.random != null && node.random.label == val) {
                return node.random;
            }
            node = node.next;
        }
        return new RandomListNode(val);
    }
}
上一篇下一篇

猜你喜欢

热点阅读