给定节点在 C++ 子树中所有节点的 XOR
对于这个问题,我们给定了一棵 n 树,其中一些查询是树的节点。我们的任务是打印由给定节点形成的子树的所有节点的 XOR。
我们举个例子来理解这个问题,
查询 − {1, 6, 5}
输出 −
0 0 5
说明 −
1^6^3^2^4^7^5 6^2^4 5
为了解决这个问题,我们将通过一次遍历树来计算子树所有节点的异或,并将其存储起来。现在,我们将计算所有子树的子节点的所有节点的异或,然后为所有给定的子树计算。存储结果可以节省时间。
示例
展示我们解决方案实现的程序,
#include <bits/stdc++.h> using namespace std; vector<vector<int> > graph; vector<int> values, xorValues; int computeXorValues(int i, int prev){ int x = values[i]; for (int j = 0; j < graph[i].size(); j++) if (graph[i][j] != prev) { x ^= computeXorValues(graph[i][j], i); } xorValues[i] = x; return x; } int solveQuerry(int u){ return xorValues[u]; } int main(){ int n = 7; graph.resize(n); xorValues.resize(n); graph[0].push_back(1); graph[0].push_back(2); graph[1].push_back(3); graph[1].push_back(4); graph[2].push_back(5); graph[2].push_back(6); values.push_back(1); values.push_back(2); values.push_back(3); values.push_back(4); values.push_back(5); values.push_back(6); values.push_back(7); computeXorValues(0, -1); int queries[] = { 0, 2, 4, 6 }; int q = sizeof(queries) / sizeof(queries[0]); for (int i = 0; i < q; i++) cout<<"Solution for querry "<<(i+1)<<": "<<solveQuerry(queries[i])<<endl; return 0; }
输出
Solution for querry 1: 0 Solution for querry 2: 2 Solution for querry 3: 5 Solution for querry 4: 7
广告