Submission #970773

#TimeUsernameProblemLanguageResultExecution timeMemory
97077312345678밀림 점프 (APIO21_jumps)C++17
23 / 100
694 ms48528 KiB
#include "jumps.h" #include <bits/stdc++.h> using namespace std; const int nx=2e5+5, kx=18; int l[nx], r[nx], deg[nx], lvl[nx], pa[kx][nx], in[nx], out[nx], rt, t; vector<int> d[nx], g[nx]; void dfs(int u) { //cout<<"dfs "<<u<<'\n'; in[u]=++t; for (auto v:g[u]) dfs(v); out[u]=t; } void init(int N, std::vector<int> H) { stack<int> s; for (int i=0; i<N; i++) l[i]=r[i]=N; for (int i=0; i<N; i++) { while (!s.empty()&&H[s.top()]<H[i]) s.pop(); if (!s.empty()) l[i]=s.top(), d[s.top()].push_back(i), deg[i]++; s.push(i); } while (!s.empty()) s.pop(); for (int i=N-1; i>=0; i--) { while (!s.empty()&&H[s.top()]<H[i]) s.pop(); if (!s.empty()) r[i]=s.top(), d[s.top()].push_back(i), deg[i]++; s.push(i); } //for (int i=0; i<N; i++) cout<<"debug "<<i<<' '<<l[i]<<' '<<r[i]<<'\n'; queue<int> q; for (int i=0; i<N; i++) if (l[i]==N&&r[i]==N) q.push(i), rt=i; while (!q.empty()) { auto u=q.front(); q.pop(); for (auto x:d[u]) if (--deg[x]==0) q.push(x); lvl[u]=max(lvl[l[u]], lvl[r[u]])+1; } for (int i=0; i<N; i++) { if (lvl[l[i]]>lvl[r[i]]) g[l[i]].push_back(i); else g[r[i]].push_back(i); } //for (int i=0; i<N; i++) cout<<"here "<<i<<' '<<g[i].size()<<'\n'; //for (int i=0; i<N; i++) cout<<"lvl "<<i<<' '<<lvl[i]<<'\n'; for (int i=0; i<kx; i++) pa[i][N]=N; for (int i=0; i<N; i++) pa[0][i]=(lvl[l[i]]<lvl[r[i]])?l[i]:r[i]; for (int i=1; i<kx; i++) for (int j=0; j<N; j++) pa[i][j]=pa[i-1][pa[i-1][j]]; dfs(rt); } int minimum_jumps(int A, int B, int C, int D) { if (!(in[D]<=in[A]&&out[A]<=out[D])) return -1; int res=0; for (int i=kx-1; i>=0; i--) if (lvl[pa[i][A]]>=lvl[D]) res+=(1<<i), A=pa[i][A]; return res+lvl[A]-lvl[D]; }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...