Code360 powered by Coding Ninjas X Naukri.com. Code360 powered by Coding Ninjas X Naukri.com
Last Updated: 7 Nov, 2021

XOR Query

Hard
Asked in companies
AmazonProtiumTata 1mg

Problem statement

You are given a tree(root 0) with N vertex having N - 1 edges. You are also given an array ‘QUERY’ of size ‘Q’, where each query consists of an integer that denotes a node. You have to print the xor of all the values of nodes in the sub-tree of the given node.

Note:

Here XOR denotes the bitwise operator (^).
Input Format:
The first line of the input contains an integer, 'T,’ denoting the number of test cases.

The first line of each test case contains two space-separated integers, ‘N’ and 'Q', denoting the number of nodes and the total queries operation to be performed.

The next ‘N’-1 lines of each test case contain two integers representing an edge between the given indices.

The next line of each test case contains ‘Q’ space-separated integers denoting the nodes of the tree on which query has to be performed.
Output Format:
For each test case, print ‘Q’ space-separated integers denoting the result of the queries.

Print the output of each test case in a separate line.
Note :
You do not need to print anything, it has already been taken care of. Just implement the given function.
Constraints :
1 <= T <= 10
0 <= N, Q <= 10^5 
0 <= QUERY[i] < N

Time Limit: 1 sec

Approaches

01 Approach

The basic idea is to traverse the subtree for each given node in the query. We traverse the nodes using the DFS and compute the XOR value of the subtree while traversing the tree.

 

Here is the algorithm :

 

  1. Create a graph (say, ‘GRAPH’) using the ‘EDGES’ array in the form of an adjacency list. (An array of lists in which ‘arr[i]’ represents the lists of vertices adjacent to the ‘ith’ vertex).
  2. Create an array (say, ‘RES’) to store the result of queries.
  3. Run a loop from 0 to ‘Q’ (say, iterator ‘i’) to traverse the queries.
    • Call the DFS function to compute the XOR of the subtree and store the value returned by the function in the ‘RES’.
  4. Return ‘RES’.

 

DFS(‘GRAPH’, ‘CURR’, ‘PREV’)  (where ‘CURR’ is the current vertex and ‘PREV’ is the previous vertex traversed).

 

  1. Create a variable (say, ‘VAL’) to store the XOR value and initialize it with ‘CURR’.
  2. Traverse the adjacent vertices of the ‘CURR’ vertex (say, ‘V’)
    • Check if ‘V’ is not equal to ‘PREV’
      • Recursively call the DFS function to traverse the tree and update the ‘VAL’ by taking the XOR of it by the value returned by the function.
  3. Return ‘VAL’.