二叉搜索樹轉換成單鏈表

題目概述

二叉樹數據結構TreeNode可用來表示單向鏈表（其中left置空，right為下一個鏈表節點）。實現一個方法，把二叉搜索樹轉換為單向鏈表，要求值的順序保持不變，轉換操作應是原址的，也就是在原始的二叉搜索樹上直接修改。

思路

BST的半線性結構轉換成線性結構，其實就是中序遍歷而已。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

class Solution {
public:
    TreeNode* convertBiNode(TreeNode* root) {
      if (root == nullptr) return root;
      std::stack<TreeNode*> stack;  // 中序遍歷需要的棧
      while(root) {
        stack.push(root);
        root = root->left;
      }
      TreeNode* newHead = stack.top();  // 最左的那棵葉節點即為單鏈表頭結點
      while(!stack.empty()) {
        TreeNode* leftNode = stack.top(); stack.pop();
        TreeNode* node = leftNode->right;
        while(node) {
          stack.push(node);
          node = node->left;
        }
        leftNode->left = nullptr;
        leftNode->right = stack.empty() ? nullptr : stack.top();
      }
      return newHead;
    }
};