# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
253244 | 2020-07-27T14:19:52 Z | Autoratch | Factories (JOI14_factories) | C++14 | 7317 ms | 111692 KB |
#include "factories.h" #include <bits/stdc++.h> using namespace std; const int N = 5e5 + 1; const int K = 19; vector<pair<int,int> > adj[N]; int sz[N],lv[N],dp[K][N],pa[N],dis[N]; bool blocked[N]; int res[N][2],ans; vector<int> rev; void dfs(int u,int p) { dp[0][u] = p,lv[u] = lv[p]+1; for(auto [d,v] : adj[u]) if(v!=p) dis[v] = dis[u]+d,dfs(v,u); } void getsz(int u,int p){ sz[u] = 1; for(auto [d,v] : adj[u]) if(v!=p and !blocked[v]) getsz(v,u),sz[u]+=sz[v]; } void build(int u,int cp) { getsz(u,cp); int c = u,prev = 0; while(true) { int mx = -1,r = 0; for(auto [d,v] : adj[c]) if(v!=prev and !blocked[v] and sz[v]>mx) mx = sz[v],r = v; if(mx*2>sz[u]) prev = c,c = r; else break; } pa[c] = cp,blocked[c] = true; for(auto [d,v] : adj[c]) if(!blocked[v]) build(v,c); } void Init(int n,int a[],int b[],int d[]) { for(int i = 0;i < n-1;i++) a[i]++,b[i]++; for(int i = 0;i < n-1;i++) { adj[a[i]].push_back({d[i],b[i]}); adj[b[i]].push_back({d[i],a[i]}); } dfs(1,0); build(1,0); for(int i = 1;i < K;i++) for(int j = 1;j <= n;j++) dp[i][j] = dp[i-1][dp[i-1][j]]; for(int i = 1;i <= n;i++) res[i][0] = res[i][1] = INT_MAX; } int lca(int a,int b) { if(lv[a]<lv[b]) swap(a,b); for(int i = K-1;i >= 0;i--) if(lv[dp[i][a]]>=lv[b]) a = dp[i][a]; if(a==b) return a; for(int i = K-1;i >= 0;i--) if(dp[i][a]!=dp[i][b]) a = dp[i][a],b = dp[i][b]; return dp[0][a]; } int dist(int a,int b) { int l = lca(a,b); return dis[a]+dis[b]-dis[l]-dis[l]; } void upd(int t,int x) { for(int u = x;u;u = pa[u]) { rev.push_back(u); res[u][t] = min(res[u][t],dist(u,x)); if(res[u][0]!=INT_MAX and res[u][1]!=INT_MAX) ans = min(ans,res[u][0]+res[u][1]); } } long long Query(int s,int x[],int t,int y[]) { for(int i = 0;i < s;i++) x[i]++; for(int i = 0;i < t;i++) y[i]++; ans = INT_MAX; for(int i = 0;i < s;i++) upd(0,x[i]); for(int i = 0;i < t;i++) upd(1,y[i]); for(int x : rev) res[x][0] = res[x][1] = INT_MAX; rev.clear(); return ans; }
Compilation message
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 46 ms | 12800 KB | Output is correct |
2 | Correct | 2084 ms | 30972 KB | Output is correct |
3 | Incorrect | 3622 ms | 30800 KB | Output isn't correct |
4 | Halted | 0 ms | 0 KB | - |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 13 ms | 12416 KB | Output is correct |
2 | Incorrect | 7317 ms | 111692 KB | Output isn't correct |
3 | Halted | 0 ms | 0 KB | - |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 46 ms | 12800 KB | Output is correct |
2 | Correct | 2084 ms | 30972 KB | Output is correct |
3 | Incorrect | 3622 ms | 30800 KB | Output isn't correct |
4 | Halted | 0 ms | 0 KB | - |