JZ-025-複雜連結串列的複製

雄獅虎豹發表於2021-12-20

複雜連結串列的複製

題目描述

輸入一個複雜連結串列（每個節點中有節點值，以及兩個指標，一個指向下一個節點，另一個特殊指標random指向一個隨機節點），請對此連結串列進行深拷貝，

  • 並返回拷貝後的頭結點。（注意，輸出結果中請不要返回引數中的節點引用，否則判題程式會直接返回空）

題目連結: 複雜連結串列的複製

程式碼

/**
 * 標題：複雜連結串列的複製
 * 題目描述
 * 輸入一個複雜連結串列（每個節點中有節點值，以及兩個指標，一個指向下一個節點，另一個特殊指標random指向一個隨機節點），請對此連結串列進行深拷貝，
 * 並返回拷貝後的頭結點。（注意，輸出結果中請不要返回引數中的節點引用，否則判題程式會直接返回空）
 * 題目連結：
 * https://www.nowcoder.com/practice/f836b2c43afc4b35ad6adc41ec941dba?tpId=13&&tqId=11178&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
 */
public class Jz25 {

    /**
     * 方法：
     * 第一步，在每個節點的後面插入複製的節點。
     * 第二步，對複製節點的 random 連結進行賦值。
     * 第三步，拆分。
     *
     * @param pHead
     * @return
     */
    public RandomListNode clone(RandomListNode pHead) {
        if (pHead == null) {
            return null;
        }
        // 插入新節點
        RandomListNode cur = pHead;
        while (cur != null) {
            RandomListNode clone = new RandomListNode(cur.label);
            clone.next = cur.next;
            cur.next = clone;
            cur = clone.next;
        }
        // 建立 random 連結
        cur = pHead;
        while (cur != null) {
            RandomListNode clone = cur.next;
            if (cur.random != null) {
                clone.random = cur.random.next;
            }
            cur = clone.next;
        }
        // 拆分
        cur = pHead;
        RandomListNode pCloneHead = pHead.next;
        while (cur.next != null) {
            RandomListNode next = cur.next;
            cur.next = next.next;
            cur = next;
        }
        return pCloneHead;
    }
}
【每日寄語】 腳步不停，初心不變。

相關文章