538. Convert BST to Greater Tree（把二叉搜索樹轉換爲累加樹）

538. 把二叉搜索樹轉換爲累加樹

給定一個二叉搜索樹（Binary Search Tree），把它轉換成爲累加樹（Greater Tree)，使得每個節點的值是原來的節點值加上所有大於它的節點值之和。

例如：

輸入: 二叉搜索樹:
              5
            /   \
           2     13

輸出: 轉換爲累加樹:
             18
            /   \
          20     13

解法一

//時間複雜度O(n), 空間複雜度O(1)
/**
 * 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:
    void getSumOrConvert(TreeNode* root, int& sum, bool isFirst) {
        if(!root) return;
        getSumOrConvert(root->left, sum, isFirst);
        if(isFirst) sum += root->val;
        else {
            sum -= root->val;
            root->val += sum;
        }
        getSumOrConvert(root->right, sum, isFirst);
    }
    TreeNode* convertBST(TreeNode* root) {
        int sum = 0;
        getSumOrConvert(root, sum, true);
        getSumOrConvert(root, sum, false);
        return root;
    }
};

解法二

//時間複雜度O(n), 空間複雜度O(1)
/**
 * 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:
    void convert(TreeNode* root, int& sum) {
        if(!root) return;
        convert(root->right, sum);
        sum += root->val;
        root->val = sum;
        convert(root->left, sum);
    }
    TreeNode* convertBST(TreeNode* root) {
        int sum = 0;
        convert(root, sum);
        return root;
    }
};

解法一

正常思種，用中序遍歷兩次，第一次求所有結點值的和，第二次爲每個結點更新值。使用標記isFirst判斷當前遍歷是第一次還是第二次。

解法二

同樣可以使用中序遍歷，但是是以“右子樹-根結點-左子樹”的順序，這樣可以只遍歷一次完成這個任務。

2019/05/31 15:54