数据结构-树 刷题总结

路径总和III

题目

思路分析：
关于这题我得第一思路是分别对左节点与右节点做两种不同类型得搜索
见下：

        solution(sum_n,root->left);
        solution(0,root->left);
        solution(sum_n,root->right);
        solution(0,root->right);

这个真是一种很傻得写法，对同一个节点进行了两次搜索，这样到下一步再进行搜索得时候肯定会发生重复的鸭！！
要记住对一个树的左节点或者右节点只能进行一次搜索！

下面的代码中首先对书进行了后序遍历找到了每个节点为根节点的树，之后对每个树进行DFS，这个树从节点开始路径都算进去。这样就避免重复啦！
代码：

/**
 * 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:
    int pathSum(TreeNode* root, int sum) {
        int res = 0;
        if(!root) return res;
        if(root->val == sum) res++;
        
        //和root节点无关的
        res += pathSum(root->left,sum);
        res += pathSum(root->right,sum);

        //和root有关的，这里不能直接调用pathSum,否则递归以后造成“断开的路径”
        res += pathSum_nroot(root->left,  sum - root->val);
        res += pathSum_nroot(root->right, sum - root->val);


        return res;
    }

    int pathSum_nroot(TreeNode* root, int sum) {   //这个函数只找包含root的路径
        if(!root) return 0;
        int res = 0;
        if(root->val == sum) res++;
        res += pathSum_nroot(root->left,sum - root->val); //自递归
        res += pathSum_nroot(root->right,sum - root->val);
        return res;
    }
};

恢复二叉搜索树

题目

思路分析：
知识点

1. 对于二叉搜素树而言，中序遍历之后是一个升序！！！
2. 指针变量也是一个变量，如果想在函数中获得修改个指针值，必须采用指针引用的方式传参
   如果不采用指针引用的话，在传入参数之后会发生拷贝，那么修改的就不是主函数里的参数了！！！
   void DFS(TreeNode* root, TreeNode*&p1 ,TreeNode*&p2,TreeNode*&pre)

代码：
代码中巧妙地传入了一个参数pre，可以记录中序遍历前一个节点的值，注意中序遍历是在遍历完左节点之后才遍历当前节点，因而在遍历左节点之后才使得 pre = root

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:

    void DFS(TreeNode* root,TreeNode*&p1,TreeNode*&p2,TreeNode*&pre)
        {
           if(root==nullptr)
           {
               return;
           }
           if(root->left)
           {
               DFS(root->left,p1,p2,pre);
           }
           if(pre&&(pre->val>root->val))
            {
               if(p1)
               {
                   p2 = root;
               }
               else
               {
                   p1 = pre;
                   p2 = root; 
               }
            }
           pre = root;
           if(root->right)
           DFS(root->right,p1,p2,pre);
        }
    void recoverTree(TreeNode* root) {
       TreeNode * p1 = nullptr;
       TreeNode * p2 = nullptr;
       TreeNode * pre = nullptr;
       DFS(root,p1,p2,pre);
       int t = 0;
       t = p1->val;
       p1->val = p2->val;
       p2->val = t;
    }
};