LeetCode 430. Flatten a Multilevel Doubly Linked List

Posted 2022-11-03 hankunyan

 

DFS题目，先访问child再访问next。遍历的时候顺便创建doubly linked list即可。

 

Helper Function

可以利用全局变量，加上dfs的helper function。这样的好处是非常好写，不容易出错。

class Solution 
public:
    Node *dummy=new Node(0,NULL,NULL,NULL), *p=dummy, *prev=NULL;
    
    Node* flatten(Node* head) 
        dfs(head);
        return dummy->next;
    
    
    void dfs(Node *head)
        if (head==NULL) return;
        p->next = new Node(head->val,prev,NULL,NULL);
        prev = p = p->next;
        if (head->child) dfs(head->child);
        if (head->next) dfs(head->next);
    
;

 

Self Recursion

。。。