时间Vs杀手

LeetCode--430. 扁平化多级双向链表
您将获得一个双向链表,除了下一个和前一个指针之外,它还有一个子指针,可能指向单独的双向链表。这些子列表可能有一个或...
扫描右侧二维码阅读全文
17
2018/12

LeetCode--430. 扁平化多级双向链表

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

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

示例:

输入:

 1---2---3---4---5---6--NULL
         |
         7---8---9---10--NULL
             |
             11--12--NULL

输出:
1-2-3-7-8-11-12-9-10-4-5-6-NULL

思路:在遍历的过程中,将child链表插在p和p->next之间,并修改相应指针

/*
// Definition for a Node.
class Node {
public:
    int val = NULL;
    Node* prev = NULL;
    Node* next = NULL;
    Node* child = NULL;

    Node() {}

    Node(int _val, Node* _prev, Node* _next, Node* _child) {
        val = _val;
        prev = _prev;
        next = _next;
        child = _child;
    }
};
*/
class Solution {
public:
    Node* flatten(Node* head) {
        Node* p = head;
        while(p!=NULL){
            if(p->child!=NULL){
                Node* pp = p->child;
                while(pp->next!=NULL){
                    pp = pp->next;
                }
                Node* p_next = p->next;
                p->next = p->child;
                if(p->next!=NULL){
                    p->next->prev = p;   
                }
                
                pp->next = p_next;
                if(pp->next!=NULL){
                    pp->next->prev = pp;   
                }
                
                p->child = NULL;
            }
            p = p->next;
        }
        return head;
    }
};
Last modification:December 17th, 2018 at 08:59 pm
If you think my article is useful to you, please feel free to appreciate

Leave a Comment