Untitled

mail@pastecode.io avatar
unknown
c_cpp
a month ago
6.3 kB
3
Indexable
Never
/*
           _                   _         _       __  __  _____ 
     /\   | |            /\   | |       | |     |  \/  |/ ____|
    /  \  | |__   ___   /  \  | |__   __| | ___ | \  / | |     
   / /\ \ | '_ \ / _ \ / /\ \ | '_ \ / _` |/ _ \| |\/| | |     
  / ____ \| |_) | (_) / ____ \| |_) | (_| | (_) | |  | | |____ 
 /_/    \_\_.__/ \___/_/    \_\_.__/ \__,_|\___/|_|  |_|\_____|
*/

// build command:
// g++ -std=gnu++17 -O3 -DDEBUG -g -fsanitize=signed-integer-overflow -fsanitize=bounds-strict -fsanitize=address -fsanitize=integer-divide-by-zero -fsanitize=float-divide-by-zero -fsanitize=pointer-overflow -fsanitize=shift-exponent -fsplit-stack -Wshadow -Wall -fconcepts -Wno-unused-result

// REMEMBER:
// BS, Offline, Persistent SegTree, SQRT, Treap, MaxFlow, FFT, Matrices

#include <bits/stdc++.h>
using namespace std;

#pragma GCC optimize("O3,unroll-loops")

// DEBUG STUFF
template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << '(' << p.first << ", " << p.second << ')'; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { os << '{'; string sep; for (const T &x : v) os << sep << x, sep = ", "; return os << '}'; }
void dbg_out() { cerr << endl; }
template<typename Head, typename... Tail> void dbg_out(Head H, Tail... T) { cerr << ' ' << H; dbg_out(T...); }
#ifdef DEBUG
#define dbg(...) cerr << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
#else
#define dbg(...)
#endif


#define F first
#define S second
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define sz(x) ((int)(x).size())

typedef long long ll;
typedef long double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

// RANDOM NUMBER GENERATOR
mt19937 RNG(chrono::steady_clock::now().time_since_epoch().count());  
#define SHUF(v) shuffle(all(v), RNG); 
// Use mt19937_64 for 64 bit random numbers.

int getrand(int l,int r) {
	return uniform_int_distribution<int>(l, r)(RNG);
}

const ld eps = 1e-9;
const int mod = 1e9 + 7;
const int oo = 1e9 + 7;
const ll lloo = 1e18 + 7;
const int N = 1e6 + 7;

void solve(int tc) {
	int n,q;
	cin >> n >> q;
	vector<int> a(n+1);
	vector<vector<int>> g(n+1);
	int SQ = sqrt(n)*3;
	for(int i = 1 ; i <= n ; i++) cin >> a[i];
	
	for(int i = 0 ; i < n-1 ; i++) {
		int u,v;
		cin >> u >> v;
		g[u].pb(v);
		g[v].pb(u);
	}
	
	vector<vector<int>> nodes_by_dpth(n+1);
	vector<int> dpth(n+1),par(n+1);
	function<void(int,int)> dfs = [&](int u,int p) {
		nodes_by_dpth[dpth[u]].pb(u);
		for(int v:g[u]) {
			if (v == p) continue;
			par[v] = u;
			a[v] ^= a[u];
			dpth[v] = dpth[u]+1;
			dfs(v,u);
		}
	};
	dfs(1,-1);

	reverse(all(nodes_by_dpth));
	vector<int> SZ(n+1);
	int timer = n;
	vector<vector<int>> G(n+1);
	vector<int> block_dpth;
	int mxsz = 0;
	for(auto nodes:nodes_by_dpth) {
		for(auto u:nodes) {
			SZ[u] = 1;
			for(int v:g[u]) {
				if (v == par[u]) continue;
				SZ[u] += SZ[v];
			}
			
			int sum = 0;
			vector<int> last_nodes;
			vector<int> block_nodes;
			for(int v:g[u]) {
				if (v == par[u]) continue;
				sum += SZ[v];
				last_nodes.pb(v);
				if (sum > SQ) {
					mxsz = max(mxsz,sum);
					int id = ++timer;
					G.pb(last_nodes);
					last_nodes.clear();
					SZ[u] -= sum;
					block_dpth.pb(dpth[u]);
					block_nodes.pb(id);
					sum = 0;
				}
			}
			mxsz = max(mxsz,SZ[u]);
			G[u] = last_nodes;
			for(auto id:block_nodes) G[u].pb(id);
		}
	}
	
	timer++;
	block_dpth.pb(0);
	G.pb({1});
	
	vector<vector<vector<vector<int>>>> cnt(timer-n+1,vector<vector<vector<int>>>(mxsz+1,vector<vector<int>>(20,vector<int>(2))));
	vector<vector<vector<int>>> total(timer-n+1,vector<vector<int>>(20,vector<int>(2)));

	vector<vector<int>> blockg(timer-n+1);

	function<void(int,int)> build_blockg = [&](int u,int id) {
		for(int v:G[u]) {
			if (v > n) {
				blockg[id].pb(v-n-1);
				build_blockg(v,v-n-1);
			}
			else build_blockg(v,id);
		}
	};
	
	build_blockg(timer,timer-n-1);

	function<void(int,int)> build_block = [&](int u,int id) {
		if (u <= n) {
			for(int i = 0 ; i < 20 ; i++) {
				cnt[id][dpth[u]-block_dpth[id]][i][(a[u]>>i)&1]++;
				total[id][i][(a[u]>>i)&1]++;
			}
		}
		for(int v:G[u]) {
			if (v > n) continue;
			build_block(v,id);
		}
	};
	
	for(int id = n+1 ; id <= timer ; id++) {
		build_block(id,id-n-1);
		//dbg(cnt[id-n-1]);
	}
	
	vector<int> xordpth(n+1);
	
	function<void(int,int,int,int)> update_block = [&](int u,int id,int lvl,int x) {
		if (lvl-block_dpth[id] < 0 || lvl-block_dpth[id] > mxsz) return;
		for(int i = 0 ; i < 20 ; i++) {
			if ((x>>i)&1) {
				total[id][i][0] -= cnt[id][lvl-block_dpth[id]][i][0];
				total[id][i][1] -= cnt[id][lvl-block_dpth[id]][i][1];
				swap(cnt[id][lvl-block_dpth[id]][i][0],cnt[id][lvl-block_dpth[id]][i][1]);
				total[id][i][0] += cnt[id][lvl-block_dpth[id]][i][0];
				total[id][i][1] += cnt[id][lvl-block_dpth[id]][i][1];
			}
		}
	};
	
	function<ll(int,int)> query_block = [&](int id,int x) {
		ll ret = 0;
		for(int i = 0 ; i < 20 ; i++) {
			ret += (1<<i)*(total[id][i][1^((x>>i)&1)]);
		}
		for(int v:blockg[id]) ret += query_block(v,x);
		dbg(id,ret);
		return ret;
	};
	
	function<ll(int,int)> query = [&](int u,int x) {
		ll ret = a[u]^xordpth[dpth[u]]^x;
		
		for(int v:G[u]) {
			if (v > n) {
				ret += query_block(v-n-1,x);
			}
			else ret += query(v,x);
		}
		dbg(u,G[u]);
		return ret;
	};
	
	for(int i = 0 ; i < q ; i++) {
		int ty;
		cin >> ty;
		if (ty == 1) {
			int l,r,x;
			cin >> l >> r >> x;
			xordpth[l] ^= x;
			for(int id = n+1 ; id <= timer ; id++) {
				update_block(id,id-n-1,l,x);
			}
		} else if (ty == 2) {
			int u;
			cin >> u;
			int x = 0;
			if (par[u] > 0) {
				x = a[par[u]]^xordpth[dpth[par[u]]];
			}
			cout << query(u,x) << '\n';
		}
	}
	
}

int32_t main() {
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	// freopen("in","r",stdin);
	// freopen("out","w",stdout);
	int T = 1;
	cin >> T;
	for(int i = 0 ; i < T ; i++) solve(i+1);
	return 0;
}
Leave a Comment