114. Flatten Binary Tree to Linked List

Given a binary tree, flatten it to a linked list in-place.
For example,Given

    1 
   / \  
  2   5 
 / \   \ 
3   4   6

The flattened tree should look like:

1 
 \
  2 
   \ 
    3 
     \ 
      4 
       \ 
        5
         \ 
          6

在深度優(yōu)先搜索這棵樹時做點手腳就好了,把pop出來的元素的右節(jié)點指向棧頂元素,左節(jié)點指空。

var flatten = function(root) {
    if (!root)
        return;
    var stack = [root];
    var last;
    while (stack.length!==0) {
        last = stack.pop();
        if (last.right)
            stack.push(last.right);
        if (last.left)
            stack.push(last.left);
        last.right = stack[stack.length-1];
        last.left = null;
    }
    
};
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容