扁平化多级双向链表 --java记录

您将获得一个双向链表,除了下一个和前一个指针之外,它还有一个子指针,可能指向单独的双向链表。这些子列表可能有一个或多个自己的子项,依此类推,生成多级数据结构,如下面的示例所示。

扁平化列表,使所有结点出现在单级双链表中。您将获得列表第一级的头部。

扁平化多级双向链表 --java记录_第1张图片
扁平化多级双向链表 --java记录_第2张图片
在这里插入图片描述
主要思想。
遍历节点,遇到有孩子节点的node,直接将整条孩子链插入到node后面。
在进行下个节点判断

class Solution {
    	public Node flatten(Node head) {
		
	        Node cur = head;
	         
	        while(cur != null) {
	        	 if(cur.child != null) {
	        		 Node child = cur.child;
	        		 Node temp = cur.child;  //记录cur的孩子节点
	        		 while(child.next != null) {  //寻找到孩子末尾节点
	        			 child = child.next;
	        		 }
	        		 // 将孩子链 插入 cur节点之后,拼接分两种情况.
	        		 if(cur.next != null) {//一种cur.next存在,需要将cur.next与child末端做连接
	        			 child.next = cur.next;
	        			 cur.next.prev = child;
	        			 cur.next = temp;
	        			 temp.prev = cur;
	        		 }else {//
	        			 cur.next = temp;
	        			 temp.prev = cur;
	        		 }
	        		 cur.child = null;	 
	        	 }
	        	 
	        	 cur = cur.next;
	        }
	      return head;  
	    }
}

你可能感兴趣的:(Leecode)