LeetCode 190周赛 T3. 二叉树中的伪回文路径

题目

给你一棵二叉树，每个节点的值为 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/pseudo-palindromic-paths-in-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

其实在读懂题意后就会发现这是一道暴力题，只需要将从根节点到叶节点处的所有值记录下来，然后判断是否为一个回文排列即可。由于数据范围在1-9，所以定义一个int a[10]即可。

class Solution {
public:
    bool check(vector<int>& arr) {
        int a[10]={};
        for (int n : arr) {
            a[n]++;
        }
        bool flag = false;
        for (int i = 1; i <= 9; i++) {
            if (a[i] % 2 == 1) {
                if (flag == true)
                    return false;
                flag = true;
            }
        }
        return true;
    }
    int ans = 0;
    void dfs(TreeNode* root,vector<int> tmp) {
    //将当前节点值记录下来
        tmp.push_back(root->val);
        //找到叶节点
        if (root->left == NULL && root->right == NULL) {
            if (check(tmp) == true)
                ans++;
            return;
        }
        
        if (root->left != NULL)
            dfs(root->left, tmp);
        if (root->right != NULL)
            dfs(root->right, tmp);
    }
    int pseudoPalindromicPaths(TreeNode* root) {
        if (root == NULL)
            return 0;
        vector<int> tmp;
        dfs(root, tmp);
        return ans;
    }
};