给定节点在 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

更新时间:20-Apr-2020

245 次浏览

职业起步

完成课程并获得认证

开始使用
广告