工作生活

复杂链表的拷贝

2019-07-03  本文已影响0人  小明今晚加班

题目描述

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


思路:本题是对一个复杂链表进行深拷贝,关键是结点中有random指针,如果按照正常思维(有些类似于深度遍历)去拷贝的话,复杂度太高,为O(n^2)。因此这里我们这里的拷贝分两步,第一步是拷贝next结点;第二步再考虑random指针结点。我们使用hashMap来存储每一个结点,避免random指针拷贝的时候出现重复拷贝的现象。

具体代码实现如下:

import java.util.HashMap;
/*
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 null;
        //p结点指向原始链表
        RandomListNode p = pHead;
        RandomListNode head = new RandomListNode(-1);
        //q结点用来操作新链表
        RandomListNode q = head;
        HashMap<RandomListNode,RandomListNode> hash = new HashMap<>();
        //第一步:只拷贝next结点
        while(p!=null){
            RandomListNode tmp = new RandomListNode(p.label);
            hash.put(p, tmp);
            q.next = tmp;
            q = q.next;
            p = p.next;
        }
        //第二步:考虑random结点
        p = pHead;
        while(p!=null){
            if(p.random!=null){
                RandomListNode value = hash.get(p);
                RandomListNode random = hash.get(p.random);
                value.random = random;
            }
            p  = p.next;
        }
        return head.next;
    }
}
上一篇下一篇

猜你喜欢

热点阅读