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

Posted 2021-02-18 itdef

地址 https://leetcode-cn.com/contest/weekly-contest-190/problems/pseudo-palindromic-paths-in-a-binary-tree/

题目描述
给你一棵二叉树，每个节点的值为 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
二叉树的值只有0~9 而且符合回文要求也就是说一条从根到叶子的路径上最多只能有一个元素的数量为单数
那么 添加一个哈希表并且使用一个变量记录有几个元素的数量为单数就好了

如果一条路径上 元素数量为单数的 这些元素个数小于等于1 那就回文路径

/**
 * 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:
    int ans =0;


    void pseudoPalindromicPathsInner(TreeNode* root,int T[10],int isSingle){
        if(root==NULL){
            return;
        }

        if(T[root->val]%2 != 0){
            isSingle--;
        }else{
            isSingle++;
        }
        T[root->val]++;


        if(root->left == NULL && root->right == NULL){
            for(int i =0; i < 10;i++){
                cout<< T[i] << " ";
            }
            if(isSingle<=1) ans++; return;
        }
        int TT[10];
        memcpy(TT,T,sizeof(int)*10);
        if(root->left != NULL){
            pseudoPalindromicPathsInner(root->left,TT,isSingle);
        }

        if(root->right != NULL){
            pseudoPalindromicPathsInner(root->right,T,isSingle); 
        }

        return;
    }

    int pseudoPalindromicPaths (TreeNode* root) {
        if(root == NULL){
            return 0;
        }
        int T[10]={0};
        T[root->val]++;
        int isSingle =1;

        if(root->left == NULL && root->right == NULL){
            if(isSingle<=1) ans++; return ans;
        }
        int TT[10];
        memcpy(TT,T,sizeof(int)*10);
        pseudoPalindromicPathsInner(root->left,TT,isSingle);
        pseudoPalindromicPathsInner(root->right,T,isSingle);

        return ans;
    }
};