LeetCode 1457 Pseudo-Palindromic Paths in a Binary Tree

简单题，可以训练一下优化思想。

题意：给出一个二叉树，每个节点的值在1-9之间，若一条从根节点到叶节点的路径（path）包含的值可以在排列变换后形成回文串，则我们称这条路径是Pseudo-Palindromic Path。

求PPP的数量。

数据范围：1≤n≤1e5

很自然的想到自根向下的dfs维护每种数值的数量，只要做到这点已经能ac了。

最裸的dfs的话，时空复杂度均为O(n)，因为dfs过程中保存了9种value的状态。可以使用01进行优化，即使用数组来记录每个value的状态stat，每次更新一个value时，就把stat[value]^=1，然后根据stat[value]的值来维护NumberOfOdd的值，若NumberOfOdd<=1，且当前节点为叶节点，则认为搜索到一条合法路径。

同时，为避免传递参数，可以使用类似八皇后的加锁解锁方法，在子节点递归结束后将当前状态进行还原，实际上由于异或运算的特性，只要再调用一次对当前value的更新即可。

这样一来，空间复杂度就降到了常数。

代码如下：

/**
 * 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:
    struct stat{
        int v[10];
        stat() {
            for (int i = 0; i < 10; ++i)
                v[i] = 0;
        }//v[0] is used to record the number of odd numbers
        bool check(int x){
            v[x] ^= 1;
            if (v[x]) ++v[0];
            else v[0]--;
            return v[0] <= 1;
        }
    }now;
    int ans = 0;
    void dfs(TreeNode* p){
        if (now.check(p->val) && p->left == nullptr && p->right == nullptr)
            ++ans;
        if (p->left != nullptr)
            dfs(p->left);
        if (p->right != nullptr)
            dfs(p->right);
        now.check(p->val);
    }
    int pseudoPalindromicPaths (TreeNode* root) {
        stat rt;
        dfs(root);
        return ans;
    }
};

实际上我第一次写的时候在dfs的参数中是传递了stat类型的，不知道为什么改了空间更优的写法重新提交了一次之后，结果并没有太大的不同。

这道题可以进行很多变种，比如不限于从根节点出发，把value的取值范围增大等变形，应该能编出许多树上dp甚至树链剖分的题目。（由于671太菜了，就摸了）