-
Notifications
You must be signed in to change notification settings - Fork 0
/
Pseudo_Palindromic_Paths_In_A_Binary_Tree.py
92 lines (65 loc) · 2.62 KB
/
Pseudo_Palindromic_Paths_In_A_Binary_Tree.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
Given a binary tree where node values are digits from 1 to 9.
A path in the binary tree is said to be pseudo-palindromic if
at least one permutation of the node values in the path is a palindrome.
Return the number of pseudo-palindromic paths going from the root node to leaf nodes.
Example 1:
Input: root = [2,3,1,3,1,null,1]
Output: 2
Explanation: The figure above represents the given binary tree.
There are three paths going from the root node to leaf nodes:
the red path [2,3,3], the green path [2,1,1], and the path [2,3,1].
Among these paths only red path and green path are pseudo-palindromic p
aths since the red path [2,3,3] can be rearranged in [3,2,3] (palindrome)
and the green path [2,1,1] can be rearranged in [1,2,1] (palindrome).
Example 2:
Input: root = [2,1,1,1,3,null,null,null,null,null,1]
Output: 1
Explanation: The figure above represents the given binary tree.
There are three paths going from the root node to leaf nodes:
the green path [2,1,1], the path [2,1,3,1], and the path [2,1].
Among these paths only the green path is pseudo-palindromic
since [2,1,1] can be rearranged in [1,2,1] (palindrome).
Example 3:
Input: root = [9]
Output: 1
Constraints:
The given binary tree will have between 1 and 10^5 nodes.
Node values are digits from 1 to 9.
Hint #1
Note that the node values of a path form a palindrome if
at most one digit has an odd frequency (parity).
Hint #2
Use a Depth First Search (DFS) keeping the frequency (parity) of the digits.
Once you are in a leaf node check if at most one digit has an odd frequency (parity).
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
# O(n) Time and O(h) Space
class Solution:
def pseudoPalindromicPaths (self, root: TreeNode) -> int:
self.result = 0
self.digits = [0 for i in range(10)]
self.dfs(root)
return self.result
def isPalindrome(self):
is_odd = 0
for i in range(10):
if self.digits[i] % 2 != 0:
is_odd += 1
if is_odd > 1:
return False
return True
def dfs(self, root):
if root is None:
return
self.digits[root.val] += 1
if not root.right and not root.left:
if self.isPalindrome():
self.result += 1
else:
self.dfs(root.left)
self.dfs(root.right)
self.digits[root.val] -= 1