Submission #370194

#TimeUsernameProblemLanguageResultExecution timeMemory
370194soroushZagrade (COI17_zagrade)C++11
100 / 100
1423 ms43904 KiB
/* #pragma GCC optimize("O2") #pragma GCC optimize("Ofast") #pragma GCC optimize("unroll-loops") #pragma GCC target("avx,avx2,sse,sse2,fma") */ #include <bits/stdc++.h> using namespace std; typedef long long ll; typedef long double ld; typedef pair<int , int> pii; mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); const int maxn = 3e5 + 10; const ll mod = 1e9+7; const ld PI = acos((ld)-1); #define pb push_back #define endl '\n' #define dokme(x) cout << x , exit(0) #define migmig ios::sync_with_stdio(false),cin.tie(0),cout.tie(0) #define ms(x , y) memset(x , y , sizeof x) ll pw(ll a, ll b, ll md = mod){ll res = 1;while(b){if(b&1){res=(a*res)%md;}a=(a*a)%md;b>>=1;}return(res);} int n , op[maxn] , sz[maxn] , cnt[maxn]; vector < int > adj[maxn]; string s; bool hide[maxn]; ll ans = 0; void plant(int v ,int p = 0){ sz[v] = 1; for(auto u : adj[v])if(!hide[u] and u^p) plant(u , v) , sz[v] += sz[u]; } int cen(int v , int n , int par = 0 , bool found = 0){ while(!found){ found = 1; for(auto u : adj[v])if(u ^ par and !hide[u] and sz[u] * 2 > n){ par = v , v = u , found = 0; break; } } return(v); } void add(int v , int h , int mx){ h += op[v]; if(h >= mx)cnt[h] ++ , mx = h; hide[v] = 1; for(auto u : adj[v])if(!hide[u]) add(u , h , mx); hide[v] = 0; } void rm(int v , int h , int mx){ h += op[v]; if(h >= mx)cnt[h] -- , mx = h; hide[v] = 1; for(auto u : adj[v])if(!hide[u]) rm(u , h , mx); hide[v] = 0; } ll chk(int v , int h = 0 , int mn = 0){ h += op[v]; ll ans = 0; if(h <= mn)ans += cnt[-h] , mn = h; hide[v] = 1; for(auto u : adj[v])if(!hide[u]) ans += chk(u , h , mn); hide[v] = 0; return ans; } ll solve(int v){ ll ans = 0; if(op[v] == 1){ cnt[1] ++; for(auto u : adj[v])if(!hide[u]) add(u , 1 , 1); for(auto u : adj[v])if(!hide[u]) rm(u , 1 , 1), ans += chk(u) , add(u , 1 , 1); for(auto u : adj[v])if(!hide[u]) rm(u , 1 , 1); cnt[1] --; } else{ for(auto u : adj[v])if(!hide[u]) add(u , -1 , 0); ans += cnt[0]; for(auto u : adj[v])if(!hide[u]) rm(u , -1 , 0), ans += chk(u) , add(u , -1 , 0); for(auto u : adj[v])if(!hide[u]) rm(u , -1 , 0); } return(ans); } void decomp(int v){ plant(v); v = cen(v , sz[v]); hide[v] = 1; ans += solve(v); for(auto u : adj[v]) if(!hide[u])decomp(u); } int32_t main(){ migmig; cin >> n >> s; for(int i = 1 ; i <= n ; i ++)op[i] = (s[i - 1] == '(' ? 1 : -1); for(int i = 1 ; i < n ; i ++){ int u , v; cin >> u >> v; adj[u].pb(v) , adj[v].pb(u); } decomp(1); cout << ans; return(0); }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...