从二叉树一个节点到另一个节点每一步的方向

题目描述

给你一棵 二叉树 的根节点 root ，这棵二叉树总共有 n 个节点。每个节点的值为 1 到 n 中的一个整数，且互不相同。给你一个整数 startValue ，表示起点节点 s 的值，和另一个不同的整数 destValue ，表示终点节点 t 的值。

请找到从节点 s 到节点 t 的 最短路径 ，并以字符串的形式返回每一步的方向。每一步用 大写 字母 'L' ，'R' 和 'U' 分别表示一种方向：

'L' 表示从一个节点前往它的 左孩子 节点。
'R' 表示从一个节点前往它的 右孩子 节点。
'U' 表示从一个节点前往它的 父 节点。
请你返回从 s 到 t 最短路径 每一步的方向。

思路分析

1. 注意回溯的问题：无论是路径的方向还是路径的打印值，都要考虑回溯pop_back
2. 注意flag剪枝，降低时间复杂度
3. 递归时，避免复杂的return,直接return void即可

Source Code

class Solution {
public:
    vector<int> v,v1,v2,v22;
    string str="",s;
    string getDirections(TreeNode*& root, int startValue, int destValue) {
        int flag=0;
        function<void(TreeNode*&,int)>dfs=[&](TreeNode* &t,int num){
            if(flag) return;  //剪枝操作
            if(t==nullptr) return;
            if(num==t->val){
                flag=1;
                v.push_back(num);
                v1=v;
                return;
            }
            v.push_back(t->val);
            dfs(t->left,num);
            dfs(t->right,num);
            v.pop_back();
        };
        function<void(TreeNode*&,int)>dfs2=[&](TreeNode* &t,int num){
            if(flag) return;
            if(t==nullptr) return;
            if(num==t->val){
                flag=1;
                v2.push_back(num);
                v22=v2;
                str=s;
            }
            v2.push_back(t->val);

            s.push_back('L');
            dfs2(t->left,num);
            s.pop_back();

            s.push_back('R');
            dfs2(t->right,num);
            s.pop_back();

            v2.pop_back();
            return;  //两者都为空，则返回空串
        };
        dfs(root,startValue);
        flag=0;
        dfs2(root,destValue);
        int cross;
        int len=v1.size()>v22.size()?v1.size():v22.size();
        int i=0;
        for(;i<len;i++){
            if(i==v1.size()||i==v22.size()||v1[i]!=v22[i]){
                cross=i-1;
                break;
            }
        }
        str.erase(0,cross);
        int Up=v1.size()-cross-1;
        str.insert(0,Up,'U');
        //对v和v2进行分析
        return str;
    }
};