結果
問題 | No.1641 Tree Xor Query |
ユーザー | kiyoshi0205 |
提出日時 | 2021-08-06 21:52:23 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 99 ms / 5,000 ms |
コード長 | 8,054 bytes |
コンパイル時間 | 3,332 ms |
コンパイル使用メモリ | 201,120 KB |
実行使用メモリ | 17,200 KB |
最終ジャッジ日時 | 2024-09-17 01:48:03 |
合計ジャッジ時間 | 4,674 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 2 ms
5,376 KB |
testcase_03 | AC | 2 ms
5,376 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 2 ms
5,376 KB |
testcase_06 | AC | 2 ms
5,376 KB |
testcase_07 | AC | 2 ms
5,376 KB |
testcase_08 | AC | 2 ms
5,376 KB |
testcase_09 | AC | 2 ms
5,376 KB |
testcase_10 | AC | 2 ms
5,376 KB |
testcase_11 | AC | 2 ms
5,376 KB |
testcase_12 | AC | 2 ms
5,376 KB |
testcase_13 | AC | 99 ms
17,200 KB |
testcase_14 | AC | 95 ms
17,068 KB |
testcase_15 | AC | 4 ms
5,376 KB |
testcase_16 | AC | 8 ms
5,376 KB |
testcase_17 | AC | 6 ms
5,376 KB |
testcase_18 | AC | 6 ms
5,376 KB |
testcase_19 | AC | 4 ms
5,376 KB |
testcase_20 | AC | 71 ms
13,956 KB |
ソースコード
#pragma GCC target("avx") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include<bits/stdc++.h> // #include<ext/pb_ds/assoc_container.hpp> // #include<ext/pb_ds/tree_policy.hpp> // #include<ext/pb_ds/tag_and_trait.hpp> // using namespace __gnu_pbds; // #include<boost/multiprecision/cpp_int.hpp> // namespace multiprecisioninteger = boost::multiprecision; // using cint=multiprecisioninteger::cpp_int; using namespace std; using ll=long long; using datas=pair<ll,ll>; using ddatas=pair<long double,long double>; using tdata=pair<ll,datas>; using vec=vector<ll>; using mat=vector<vec>; using pvec=vector<datas>; using pmat=vector<pvec>; // using llset=tree<ll,null_type,less<ll>,rb_tree_tag,tree_order_statistics_node_update>; #define For(i,a,b) for(i=a;i<(ll)b;++i) #define bFor(i,b,a) for(i=b,--i;i>=(ll)a;--i) #define rep(i,N) For(i,0,N) #define rep1(i,N) For(i,1,N) #define brep(i,N) bFor(i,N,0) #define brep1(i,N) bFor(i,N,1) #define all(v) (v).begin(),(v).end() #define allr(v) (v).rbegin(),(v).rend() #define vsort(v) sort(all(v)) #define vrsort(v) sort(allr(v)) #define uniq(v) vsort(v);(v).erase(unique(all(v)),(v).end()) #define endl "\n" #define popcount __builtin_popcountll #define eb emplace_back #define print(x) cout<<x<<endl #define printyes print("Yes") #define printno print("No") #define printYES print("YES") #define printNO print("NO") #define output(v) do{bool f=0;for(auto outi:v){cout<<(f?" ":"")<<outi;f=1;}cout<<endl;}while(0) #define matoutput(v) do{for(auto outimat:v)output(outimat);}while(0) constexpr ll mod=1000000007; // constexpr ll mod=998244353; constexpr ll inf=1LL<<60; constexpr long double eps=1e-9; const long double PI=acosl(-1); template<class T,class E> ostream& operator<<(ostream& os,const pair<T,E>& p){return os<<"("<<p.first<<","<<p.second<<")";} template<class T> ostream& operator<<(ostream& os,const vector<T>& v){ os<<"{";bool f=false; for(auto& x:v){if(f)os<<",";os<<x;f=true;} os<<"}"; return os; } template<class T> ostream& operator<<(ostream& os,const set<T>& v){ os<<"{";bool f=false; for(auto& x:v){if(f)os<<",";os<<x;f=true;} os<<"}"; return os; } template<class T> ostream& operator<<(ostream& os,const multiset<T>& v){ os<<"{";bool f=false; for(auto& x:v){if(f)os<<",";os<<x;f=true;} os<<"}"; return os; } template<class T,class E> ostream& operator<<(ostream& os,const map<T,E>& v){ os<<"{";bool f=false; for(auto& x:v){if(f)os<<",";os<<x;f=true;} os<<"}"; return os; } template<class T> inline bool chmax(T& a,const T b){bool x=a<b;if(x)a=b;return x;} template<class T> inline bool chmin(T& a,const T b){bool x=a>b;if(x)a=b;return x;} #ifdef DEBUG void debugg(){cout<<endl;} template<class T,class... Args>void debugg(const T& x,const Args&... args){cout<<" "<<x;debugg(args...);} #define debug(...) cout<<__LINE__<<" ["<<#__VA_ARGS__<<"]:",debugg(__VA_ARGS__) #else #define debug(...) (void(0)) #endif inline void startupcpp(void) noexcept{ cin.tie(0); ios::sync_with_stdio(false); cout<<fixed<<setprecision(15); } ll modinv(ll a,const ll m=mod) noexcept{ ll b=m,u=1,v=0,t; while(b){ t=a/b; a-=t*b; swap(a,b); u-=t*v; swap(u,v); } return (u+m)%m; } ll moddevide(const ll a,const ll b) noexcept{return (a*modinv(b))%mod;} vec modncrlistp,modncrlistm; ll modncr(const ll n,const ll r) noexcept{ if(n<r)return 0; ll i,size=modncrlistp.size(); if(size<=n){ modncrlistp.resize(n+1); modncrlistm.resize(n+1); if(!size){ modncrlistp[0]=modncrlistm[0]=1; size++; } For(i,size,n+1)modncrlistp[i]=modncrlistp[i-1]*i%mod; modncrlistm[n]=modinv(modncrlistp[n]); for(i=n;i>size;--i)modncrlistm[i-1]=modncrlistm[i]*i%mod; } return modncrlistp[n]*modncrlistm[r]%mod*modncrlistm[n-r]%mod; } ll modpow(ll a,ll n,const ll m=mod){ if(n<0)return 0; ll res=1; while(n>0){ if(n&1)res=res*a%m; a=a*a%m; n>>=1; } return res; } constexpr ll gcd(const ll a,const ll b) noexcept{return (!b)?abs(a):(a%b==0)?abs(b):gcd(b,a%b);} constexpr ll lcm(const ll a,const ll b) noexcept{return a/gcd(a,b)*b;} struct unionfind{ private: int maxN; vector<int> par,treesize; public:unionfind(int N) :maxN(N),par(N),treesize(N,1){ for(int i=0;i<maxN;++i)par[i]=i; } int root(int x){ while(par[x]!=x){ x=par[x]=par[par[x]]; } return x; } bool unite(int x,int y){ x=root(x); y=root(y); if(x==y)return false; if(treesize[x]>treesize[y])swap(x,y); par[x]=y; treesize[y]+=treesize[x]; return true; } bool unite(pair<int,int> v){ return unite(v.first,v.second); } bool parcheck(int x,int y){ return root(x)==root(y); } bool parcheck(pair<int,int> v){ return parcheck(v.first,v.second); } int size(int x){ return treesize[root(x)]; } void clear(){ treesize.assign(maxN,1); for(int i=0;i<maxN;++i)par[i]=i; } vector<vector<int>> groups(){ vector<vector<int>> res(maxN); for(int i=0;i<maxN;++i)res[root(i)].eb(i); // return res; vector<vector<int>> res2; for(vector<int> x:res){ if(x.size())res2.eb(x); } return res2; } }; template<typename T> struct segtree{ private: int N=1; T E; using func=function<T(T,T)>; func f,g; vector<T> tree; public: segtree(){} segtree(vector<T>& v,T e,func Merge,func Act):E(e),f(Merge),g(Act){ int i,K=v.size(); while(N<K)N<<=1; tree.resize(N<<1); For(i,N,N+K)tree[i]=v[i-N]; For(i,N+K,N<<1)tree[i]=E; brep1(i,N)tree[i]=f(tree[i<<1],tree[(i<<1)|1]); } segtree(vector<T>& v,T e,func Merge){*this=segtree(v,e,Merge,Merge);}; segtree(int N,T e,func F){ vector<T> res(N,e); *this=segtree(res,e,F); } void action(int i,T a){ if((i<0)|(i>=N))return; i|=N; tree[i]=g(tree[i],a); while(i>1){ i>>=1; tree[i]=f(tree[i<<1],tree[(i<<1)|1]); } } void input(int i,T a){ if((i<0)|(i>=N))return; tree[i|=N]=a; while(i>1){ i>>=1; tree[i]=f(tree[i<<1],tree[(i<<1)|1]); } } T prod(int a,int b){ if((a>=b)|(a<0)|(b<=0)|(a>=N)|(b>N))return E; T l(E),r(E); for(a|=N,b+=N;a<b;a>>=1,b>>=1){ if(a&1)l=f(l,tree[a++]); if(b&1)r=f(tree[--b],r); } return f(l,r); } }; struct HLD{ private: //無向木であること! vector<vector<int>> g; int fdfs(int now,int p){ int d=0; par[now]=p; for(size_t i=0;i+1<g[now].size();++i){ if(g[now][i]==p)swap(g[now][i],g[now].back()); if(chmax(d,fdfs(g[now][i],now)))swap(g[now][0],g[now][i]); } g[now].pop_back(); return d+1; } void sdfs(int now){ in[now]=ETlist.size(); ETlist.emplace_back(now); for(size_t i=0;i<g[now].size();++i){ nxt[g[now][i]]=i?g[now][i]:nxt[now]; sdfs(g[now][i]); } out[now]=ETlist.size(); } public: // nxt[i]:iのheavy root // Heavy edge i = [in[nxt[i]],in[i]] // sub tree i = [in[i],out[i]) vector<int> par,nxt,in,out,ETlist; HLD(vector<vector<int>> G,int first=0):g(G),par(G.size()),nxt(G.size()),in(G.size()),out(G.size()){ ETlist.reserve(G.size()); init(first); }; HLD(int N):g(N),par(N),nxt(N),in(N),out(N){ETlist.reserve(N);}; void add_edge(int a,int b){ g[a].emplace_back(b); g[b].emplace_back(a); } void init(int first=0){ { int d=0;par[first]=-1; for(auto& x:g[first])if(chmax(d,fdfs(x,first)))swap(x,g[first][0]); } nxt[first]=first; sdfs(first); } int lca(int a,int b){ if(in[a]>in[b])swap(a,b); if(nxt[a]==nxt[b])return a; return lca(a,par[nxt[b]]); } }; int main(){ startupcpp(); ll i,a,b,N,Q; cin>>N>>Q; vec c(N); rep(i,N)cin>>c[i]; HLD tree(N); rep1(i,N){ cin>>a>>b; tree.add_edge(--a,--b); } tree.init(); { auto& ch=tree.ETlist; debug(ch); vec sc(N); rep(i,N)sc[i]=c[ch[i]]; swap(c,sc); } segtree<ll> seg(c,0LL,[](ll a,ll b){return a^b;}); while(Q--){ cin>>i>>a>>b; --a; if(i==1)seg.action(tree.in[a],b); else print(seg.prod(tree.in[a],tree.out[a])); } }