cs: add “872. Leaf-Similar Trees”
Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
parent
fe58c04720
commit
1ce3f94173
1 changed files with 52 additions and 0 deletions
52
cs/leaf-similar-trees.cs
Normal file
52
cs/leaf-similar-trees.cs
Normal file
|
@ -0,0 +1,52 @@
|
||||||
|
/**
|
||||||
|
* Definition for a binary tree node.
|
||||||
|
* public class TreeNode {
|
||||||
|
* public int val;
|
||||||
|
* public TreeNode left;
|
||||||
|
* public TreeNode right;
|
||||||
|
* public TreeNode(int val=0, TreeNode left=null, TreeNode right=null) {
|
||||||
|
* this.val = val;
|
||||||
|
* this.left = left;
|
||||||
|
* this.right = right;
|
||||||
|
* }
|
||||||
|
* }
|
||||||
|
*/
|
||||||
|
public class Solution {
|
||||||
|
private static IEnumerable<int> Leafs(TreeNode node) {
|
||||||
|
if (node == null) {
|
||||||
|
yield break;
|
||||||
|
}
|
||||||
|
|
||||||
|
var stack = new Stack<TreeNode>();
|
||||||
|
stack.Push(node);
|
||||||
|
|
||||||
|
while (stack.Count > 0) {
|
||||||
|
var n = stack.Pop();
|
||||||
|
|
||||||
|
if (n.left == null && n.right == null) {
|
||||||
|
yield return n.val;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (n.right != null) {
|
||||||
|
stack.Push(n.right);
|
||||||
|
}
|
||||||
|
if (n.left != null) {
|
||||||
|
stack.Push(n.left);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public bool LeafSimilar(TreeNode root1, TreeNode root2) {
|
||||||
|
var xs = Leafs(root1).GetEnumerator();
|
||||||
|
var ys = Leafs(root2).GetEnumerator();
|
||||||
|
|
||||||
|
bool hasX = xs.MoveNext(), hasY = ys.MoveNext();
|
||||||
|
for (; hasX && hasY; hasX = xs.MoveNext(), hasY = ys.MoveNext()) {
|
||||||
|
if (xs.Current != ys.Current) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return hasX == hasY;
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue