forked from sachuverma/DataStructures-Algorithms
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Check Mirror in N-ary tree.cpp
62 lines (54 loc) · 1.43 KB
/
Check Mirror in N-ary tree.cpp
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
/*
Check Mirror in N-ary tree
===========================
Given two n-ary trees. Check if they are mirror images of each other or not. You are also given e denoting the number of edges in both trees, and two arrays, A[] and B[]. Each array has 2*e space separated values u,v denoting an edge from u to v for the both trees.
Example 1:
Input:
n = 3, e = 2
A[] = {1, 2, 1, 3}
B[] = {1, 3, 1, 2}
Output:
1
Explanation:
1 1
/ \ / \
2 3 3 2
As we can clearly see, the second tree
is mirror image of the first.
Example 2:
Input:
n = 3, e = 2
A[] = {1, 2, 1, 3}
B[] = {1, 2, 1, 3}
Output:
1
Explanation:
1 1
/ \ / \
2 3 2 3
As we can clearly see, the second tree
isn't mirror image of the first.
Your Task:
You don't need to read input or print anything. Your task is to complete the function checkMirrorTree() which takes 2 Integers n, and e; and two arrays A[] and B[] of size 2*e as input and returns 1 if the trees are mirror images of each other and 0 if not.
Expected Time Complexity: O(n)
Expected Auxiliary Space: O(n)
Constraints:
1 <= n,e <= 105
*/
int checkMirrorTree(int n, int e, int A[], int B[])
{
unordered_map<int, stack<int>> t1;
for (int i = 0; i < 2 * e; i += 2)
{
int u = A[i], v = A[i + 1];
t1[u].push(v);
}
for (int i = 0; i < 2 * e; i += 2)
{
int u = B[i], v = B[i + 1];
if (t1[u].top() != v)
return false;
t1[u].pop();
}
return true;
}