简单题,可以训练一下优化思想。
题意:给出一个二叉树,每个节点的值在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太菜了,就摸了)