結果
| 問題 |
No.1641 Tree Xor Query
|
| コンテスト | |
| ユーザー |
👑 potato167
|
| 提出日時 | 2021-08-06 22:01:42 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 95 ms / 5,000 ms |
| コード長 | 2,939 bytes |
| コンパイル時間 | 1,991 ms |
| コンパイル使用メモリ | 202,348 KB |
| 最終ジャッジ日時 | 2025-01-23 15:21:47 |
|
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 18 |
ソースコード
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define _GLIBCXX_DEBUG
using namespace std;
using std::cout;
using std::cin;
using std::endl;
using ll=long long;
using ld=long double;
ll I=1167167167167167167;
ll mod=1e9+7;
#define rep(i,a) for (ll i=0;i<a;i++)
template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>;
template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;}
template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;}
template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());}
template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});}
void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";}
namespace po167{
template <class T,T (*op)(T,T),T(*e)()>
struct segment_tree{
int _n,size;
std::vector<T> seg;
int ceil_pow2(int a){
int b=1;
while(a>b){
b<<=1;
}
return b;
}
void update(int k){seg[k]=op(seg[k*2],seg[k*2+1]);};
segment_tree(int n) :_n(n){
size=ceil_pow2(n);
seg=std::vector<T>(size*2,e());
}
segment_tree(std::vector<T> &p) :_n((int) p.size()){
size=ceil_pow2(_n);
seg=std::vector<T>(size*2,e());
for(int i=0;i<_n;i++) seg[i+size]=p[i];
for(int i=size-1;i>0;i--) update(i);
}
void set(int ind,T val){
assert(0<=ind&&ind<_n);
ind+=size;
seg[ind]=val;
while(ind!=1){
ind>>=1;
update(ind);
}
}
T get(int ind){
assert(0<=ind&&ind<_n);
return seg[ind+size];
}
T query(int l,int r){
assert(0<=l&&l<_n);
assert(0<=r&&r<=_n);
T l_val=e();
T r_val=e();
l+=size,r+=size;
while(l<r){
if(l&1) l_val=op(l_val,seg[l]),l+=1;
if(r&1) r-=1,r_val=op(seg[r],r_val);
r>>=1;
l>>=1;
}
return op(l_val,r_val);
}
};
}
using po167::segment_tree;
int Xor(int a,int b){return a^b;}
int e(){return 0;}
int MAX_N=100100;
vector<int> order;
vector<int> seen(MAX_N);
vector<vector<int>> G(MAX_N);
void dfs(int a){
order.push_back(a);
seen[a]=1;
for(auto x:G[a]){
if(seen[x]==0){
seen[x]=1;
dfs(x);
}
}
order.push_back(a);
}
//おちこんだりもしたけれど、私はげんきです。
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int N,T;
cin>>N>>T;
vector<int> p(N);
rep(i,N) cin>>p[i];
rep(i,N-1){
int a,b;
cin>>a>>b;
a--,b--;
G[a].push_back(b);
G[b].push_back(a);
}
vector<int> start(N,-1),end(N,-1);
segment_tree<int,Xor,e> seg(2*N);
dfs(0);
rep(i,N*2){
if(start[order[i]]==-1) start[order[i]]=i;
else end[order[i]]=i;
}
rep(i,N){
seg.set(start[i],p[i]);
//cout<<start[i]<<" "<<end[i]<<endl;
}
rep(i,T){
int t,x,y;
cin>>t>>x>>y;
x--;
if(t==1){
seg.set(start[x],Xor(seg.get(start[x]),y));
}
else
{
cout<<seg.query(start[x],end[x])<<"\n";
}
}
}
potato167