Submission #637351

#TimeUsernameProblemLanguageResultExecution timeMemory
637351Cyber_WolfDeblo (COCI18_deblo)C++14
45 / 90
22 ms8328 KiB
//Problem: Deblo; //Contest: COCI18 Contest 2 Problem 3 //Link: https://oj.uz/problem/view/COCI18_deblo #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> #pragma GCC optimize("Ofast") using namespace std; using namespace __gnu_pbds; #define lg long long #define ordered_set tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> #define fastio ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0); #define error(args...) { string _s = #args; replace(_s.begin(), _s.end(), ',', ' '); stringstream _ss(_s); istream_iterator<string> _it(_ss); err(_it, args); #define endl \n #define lbound(x, y) lower_bound(x.begin(), x.end(), y) #define ubound(x, y) upper_bound(x.begin(), x.end(), y) #define sortasc(v) sort(v.begin(), v.end()) #define sortdesc(v) sort(v.rbegin(), v.rend()) #define custom_array(a,l, r) int _##a[r-l+1]; int*a=_##a-l; mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); const lg MOD = 1e9+7, N = 1e3+5, M = 1e7+1, SZ = 1e3+1; /* bitset<N> primes; lg pwrs[N], inv[N]; lg fast_power(lg n, lg k) { if(!k) return 1; if(k&1) return (fast_power(n, k-1)*n)%MOD; lg x = fast_power(n, k/2)%MOD; return (x*x)%MOD; } void sieve() { primes.set(); primes[0] = primes[1] = 0; for(lg i = 2; i < N; i++) { if(!primes[i]) continue; for(lg j = i*i; j < N; j += i) { primes[j] = 0; } } return; } struct matrix { vector<vector<lg>> con = vector<vector<lg>> (SZ, vector<lg> (SZ)); matrix operator *(const matrix& a) { matrix product; for(int i = 0; i < (lg)con.size(); i++) { for(int j = 0; j < (lg)a.con[0].size(); j++) { for(int k = 0; k < (lg)a.con.size(); k++) { product.con[i][j] = (product.con[i][j]+(con[i][k]*a.con[k][j])%MOD)%MOD; } } } return product; } }; void preprocess(lg x) { inv[2e5] = fast_power(x, MOD-2); for(int i = 2e5-1; i > 1; i--) { inv[i] = (inv[i+1]*i)%MOD; } pwrs[0] = 1; for(int i = 1; i <= 2e5; i++) { pwrs[i] = (pwrs[i]*i)%MOD; } return; } */ vector<lg> v, val(N), adj[N]; lg vis[N][N], x; lg dfs(lg src, lg par = -1, lg path = 0) { path ^= val[src]; // cout << "SRC: " << src << ' ' << path << '\n'; // for(auto it : v) // { // cout << it << ' '; // } // cout << '\n'; lg ans = 0; if(!vis[min(x, src)][max(x, src)]) ans += path; vis[min(x, src)][max(x, src)] = true; // for(auto it : v) // { // ans += (path^it); // } for(auto it : adj[src]) { if(it == par) continue; //v.push_back(path); ans += dfs(it, src, path); //v.pop_back(); } return ans; } int main() { fastio; lg n; cin >> n; for(int i = 1; i <= n; i++) cin >> val[i]; for(int i = 0; i < n-1; i++) { lg u, v; cin >> u >> v; adj[u].push_back(v); adj[v].push_back(u); } // v.push_back(0); lg ans = 0; memset(vis, 0, sizeof(vis)); for(int i = 1; i <= n; i++) x = i, ans += dfs(i); cout << ans << '\n'; return 0; }
#Verdict Execution timeMemoryGrader output
Fetching results...