LeetCode：1457. 二叉树中的伪回文路径（DFS C++ Java）

news2024/11/25 11:48:39

1457. 二叉树中的伪回文路径

题目描述：

原理思路：

1457. 二叉树中的伪回文路径

题目描述：

给你一棵二叉树，每个节点的值为 1 到 9 。我们称二叉树中的一条路径是「伪回文」的，当它满足：路径经过的所有节点值的排列中，存在一个回文序列。

请你返回从根到叶子节点的所有路径中 伪回文 路径的数目。

示例 1：

输入：root = [2,3,1,3,1,null,1]
输出：2 
解释：上图为给定的二叉树。总共有 3 条从根到叶子的路径：红色路径 [2,3,3] ，绿色路径 [2,1,1] 和路径 [2,3,1] 。
     在这些路径中，只有红色和绿色的路径是伪回文路径，因为红色路径 [2,3,3] 存在回文排列 [3,2,3] ，绿色路径 [2,1,1] 存在回文排列 [1,2,1] 。

示例 2：

输入：root = [2,1,1,1,3,null,null,null,null,null,1]
输出：1 
解释：上图为给定二叉树。总共有 3 条从根到叶子的路径：绿色路径 [2,1,1] ，路径 [2,1,3,1] 和路径 [2,1] 。
     这些路径中只有绿色路径是伪回文路径，因为 [2,1,1] 存在回文排列 [1,2,1] 。

示例 3：

输入：root = [9]
输出：1

提示：

给定二叉树的节点数目在范围 [1, 105] 内
1 <= Node.val <= 9

实现代码与解析：

dfs

C++

/**
 * 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:
    vector<int> cnt = vector<int>(11, 0);
    
    int res = 0;
    bool check() {
        int flag = 1; // 只能有一个奇数
        for (int i = 1; i <= 9; i++) {
            if (cnt[i] % 2 == 0) continue;
            else {
                if (flag == 0) return false;
                else flag--;
            }
        }
        return true;
    }
    void dfs(TreeNode* cur) {
        if (cur == 0) return;

        cnt[cur->val]++;
        if (!cur->left && !cur->right) {
            if (check()) res++;
        }
        if (cur->left) dfs(cur->left);
        if (cur->right) dfs(cur->right);
        cnt[cur->val]--;

        return;
    }
    int pseudoPalindromicPaths (TreeNode* root) {

        dfs(root);
        return res;
    }
};

Java

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */


class Solution {
    int[] cnt = new int[11];
    int res = 0;
    public boolean check() {
        int flag = 1;
        for (int i = 1; i <= 9; i++) {
            if (cnt[i] % 2 == 0) continue;
            else {
                if (flag == 0) return false;
                else flag--;
            }
        }
        return true;
    }

    public void dfs(TreeNode cur) {
        if (cur == null) return;

        cnt[cur.val]++;
        if (cur.left == null && cur.right == null) {
            if (check()) res++;
        }
        if (cur.left != null) dfs(cur.left);
        if (cur.right != null) dfs(cur.right);
        cnt[cur.val]--;

        return;
    }

    public int pseudoPalindromicPaths (TreeNode root) {
        dfs(root);
        return res;
    }
}