Copy List with Random Pointer


/*
 *  A linked list is given
 *  such that each node contains an additional random pointer
 *  which could point to any node in the list or null.
 *  
 *  Return a deep copy of the list. 
 */
public class CopyListRandomPointer {

	public RandomListNode copyRandomList(RandomListNode head) {
		// Note: The Solution object is instantiated only once and is reused
		// by each test case.
		if (head == null)
			return head;
		duplicateList(head);
		return adjustList(head);
	}

	public void duplicateList(RandomListNode head) {
		RandomListNode node = head;
		// copy each node and append it to the original one
		while (node != null) {
			RandomListNode temp = new RandomListNode(node.label);
			temp.next = node.next;
			node.next = temp;
			node = temp.next;
		}
	}

	public RandomListNode adjustList(RandomListNode head) {
		RandomListNode newHead = head.next;
		RandomListNode node = head;
		RandomListNode newNode = newHead;
		// adjust the random pointer of the copied nodes
		while (node != null) {
			if (node.random == null)
				node.next.random = null;
			else
				node.next.random = node.random.next;
			node = node.next.next;
		}
		node = head;
		newNode = newHead;
		// separate the list to two lists
		while (node != null) {
			if (newNode.next != null) {
				node.next = newNode.next;
				newNode.next = newNode.next.next;
				node = node.next;
				newNode = newNode.next;
			} else {
				node.next = null;
				newNode.next = null;
				node = node.next;
				newNode = newNode.next;
			}
		}
		return newHead;
	}

	class RandomListNode {
		int label;
		RandomListNode next, random;

		RandomListNode(int x) {
			this.label = x;
		}
	}
}


發佈了127 篇原創文章 · 獲贊 0 · 訪問量 9萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章