其他分享
首页 > 其他分享> > LeetCode 971. 翻转二叉树以匹配先序遍历

LeetCode 971. 翻转二叉树以匹配先序遍历

作者:互联网

给定一个有 N 个节点的二叉树,每个节点都有一个不同于其他节点且处于 {1, …, N} 中的值。

通过交换节点的左子节点和右子节点,可以翻转该二叉树中的节点。

考虑从根节点开始的先序遍历报告的 N 值序列。将这一 N 值序列称为树的行程。

(回想一下,节点的先序遍历意味着我们报告当前节点的值,然后先序遍历左子节点,再先序遍历右子节点。)

我们的目标是翻转最少的树中节点,以便树的行程与给定的行程 voyage 相匹配。

如果可以,则返回翻转的所有节点的值的列表。你可以按任何顺序返回答案。

如果不能,则返回列表 [-1]。

示例 1:

输入:root = [1,2], voyage = [2,1]
输出:[-1]
示例 2:

输入:root = [1,2,3], voyage = [1,3,2]
输出:[1]
示例 3:

输入:root = [1,2,3], voyage = [1,2,3]
输出:[]

提示:

1 <= N <= 100

思路:
递归一下就好了

/**
 * 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:
    vector<int>ans;
    bool dfs(TreeNode* root,vector<int>&voyage,int&id) {
        if(root == NULL) return true;
        if(root -> val != voyage[id]) {
            return false;
        }
        id++;
        int ID = id;
        if(dfs(root -> left,voyage,id)) {
            if(dfs(root -> right,voyage,id)) return true;
        } else {
            id = ID;
            ans.push_back(root -> val);
            if(dfs(root -> right,voyage,id) && dfs(root -> left,voyage,id)) return true;
        }
        return false;
    }
    vector<int> flipMatchVoyage(TreeNode* root, vector<int>& voyage) {
        int id = 0;
        if(dfs(root,voyage,id)) return ans;
        return vector<int>(1,-1);
    }
};

标签:voyage,return,971,dfs,LeetCode,二叉树,节点,root,id
来源: https://blog.csdn.net/tomjobs/article/details/113553001