結果
問題 |
No.3189 Semifinal Stage
|
ユーザー |
👑 ![]() |
提出日時 | 2025-06-20 22:11:30 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,505 ms / 4,000 ms |
コード長 | 10,171 bytes |
コンパイル時間 | 2,780 ms |
コンパイル使用メモリ | 209,372 KB |
実行使用メモリ | 27,660 KB |
最終ジャッジ日時 | 2025-06-20 22:12:06 |
合計ジャッジ時間 | 27,844 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
//#include <bits/stdc++.h> //using namespace std; //using ll=long long; //const ll ILL=2167167167167167167; //const int INF=2100000000; //#define rep(i,a,b) for (int i=(int)(a);i<(int)(b);i++) //#define all(p) p.begin(),p.end() //template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>; //template<class T> int LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();} //template<class T> int UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();} //template<class T> bool chmin(T &a,T b){if(b<a){a=b;return 1;}else return 0;} //template<class T> bool chmax(T &a,T b){if(a<b){a=b;return 1;}else return 0;} //template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());} //template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});} //bool yneos(bool a,bool upp=false){if(a){cout<<(upp?"YES\n":"Yes\n");}else{cout<<(upp?"NO\n":"No\n");}return a;} //template<class T> void vec_out(vector<T> &p,int ty=0){ // if(ty==2){cout<<'{';for(int i=0;i<(int)p.size();i++){if(i){cout<<",";}cout<<'"'<<p[i]<<'"';}cout<<"}\n";} // else{if(ty==1){cout<<p.size()<<"\n";}for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";}} //template<class T> T vec_min(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmin(ans,x);return ans;} //template<class T> T vec_max(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmax(ans,x);return ans;} //template<class T> T vec_sum(vector<T> &a){T ans=T(0);for(auto &x:a) ans+=x;return ans;} //int pop_count(long long a){int res=0;while(a){res+=(a&1),a>>=1;}return res;} //template<class T> T square(T a){return a * a;} // //#include "po167_library/graph/tree/LCA.hpp" // //void solve(); //// POP'N ROLL MUSIC / TOMOO //int main() { // ios::sync_with_stdio(false); // cin.tie(nullptr); // // int t = 1; // // cin >> t; // rep(i, 0, t) solve(); //} // //void solve(){ // int N; // cin >> N; // const int B = 320; // vector<vector<int>> G(N); // rep(i, 0, N - 1){ // int a, b; // cin >> a >> b; // a--, b--; // G[a].push_back(b); // G[b].push_back(a); // } // int Q; // cin >> Q; // vector<int> T(Q), V(Q); // rep(i, 0, Q) cin >> T[i] >> V[i], V[i]--; // po167::LCA L(G); // vector<int> ans(N); // vector<int> col(N); // vector<int> rem; // rep(i, 0, Q){ // if (i % B == 0){ // rem.clear(); // vector<int> C(N, 1); // int r = min(i + B, Q); // rep(j, i, r) if (T[j] == 1) C[V[j]] = 0; // rep(j, 0, N) ans[j] = INF; // vector<int> order; // rep(j, 0, N) if (C[j] == 0) rem.push_back(j); // rep(j, 0, N) if (C[j] == 1 && col[j] == 1){ // ans[j] = 0; // order.push_back(j); // } // rep(rp, 0, order.size()){ // int a = order[rp]; // for (auto x : G[a]) if (ans[x] == INF){ // ans[x] = ans[a] + 1; // order.push_back(x); // } // } // } // if (T[i] == 1) col[V[i]] ^= 1; // else{ // int a = ans[V[i]]; // for (auto x : rem) if (col[x] == 1) chmin(a, L.dist(V[i], x)); // cout << a << "\n"; // } // } //} // #line 1 "h.cpp" #include <bits/stdc++.h> using namespace std; using ll=long long; const ll ILL=2167167167167167167; const int INF=2100000000; #define rep(i,a,b) for (int i=(int)(a);i<(int)(b);i++) #define all(p) p.begin(),p.end() template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>; template<class T> int LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();} template<class T> int UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();} template<class T> bool chmin(T &a,T b){if(b<a){a=b;return 1;}else return 0;} template<class T> bool chmax(T &a,T b){if(a<b){a=b;return 1;}else return 0;} template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());} template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});} bool yneos(bool a,bool upp=false){if(a){cout<<(upp?"YES\n":"Yes\n");}else{cout<<(upp?"NO\n":"No\n");}return a;} template<class T> void vec_out(vector<T> &p,int ty=0){ if(ty==2){cout<<'{';for(int i=0;i<(int)p.size();i++){if(i){cout<<",";}cout<<'"'<<p[i]<<'"';}cout<<"}\n";} else{if(ty==1){cout<<p.size()<<"\n";}for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";}} template<class T> T vec_min(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmin(ans,x);return ans;} template<class T> T vec_max(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmax(ans,x);return ans;} template<class T> T vec_sum(vector<T> &a){T ans=T(0);for(auto &x:a) ans+=x;return ans;} int pop_count(long long a){int res=0;while(a){res+=(a&1),a>>=1;}return res;} template<class T> T square(T a){return a * a;} #line 4 "/Users/Shared/po167_library/ds/Sparse_table.hpp" namespace po167{ template<class T, T(*op)(T, T)> struct Sparse_table{ int n; int depth; std::vector<std::vector<T>> val; void init(std::vector<T> &v){ depth = 1; n = v.size(); while ((1 << depth) <= n) depth++; val.resize(depth); val[0] = v; for (int i = 1; i < depth; i++){ val[i].resize(n); for (int j = 0; j <= n - (1 << i); j++){ val[i][j] = op(val[i - 1][j], val[i - 1][j + (1 << (i - 1))]); } } } Sparse_table(std::vector<T> v){ init(v); } Sparse_table(){} // 0 <= l < r <= n // if l == r : assert T prod(int l, int r){ assert(0 <= l && l < r && r <= n); int z=31-__builtin_clz(r-l); return op(val[z][l], val[z][r - (1 << z)]); } }; } #line 6 "/Users/Shared/po167_library/graph/tree/LCA.hpp" namespace po167{ int op(int a, int b){ return std::min(a, b); } struct LCA{ Sparse_table<int, op> table; std::vector<int> depth; std::vector<int> E; std::vector<int> order; int var_num; void init(std::vector<std::vector<int>> &g, int root = 0){ var_num = g.size(); assert(0 <= root && root < var_num); std::vector<int> val; depth.assign(var_num, -1); depth[root] = 0; E.resize(var_num); std::vector<int> tmp; order.clear(); tmp.reserve(var_num); order.reserve(var_num); int c = 0; auto dfs = [&](auto self, int var, int pare) -> void { E[var] = c++; if (var != root) tmp.push_back(E[pare]); order.push_back(var); for (auto x : g[var]) if (depth[x] == -1){ depth[x] = depth[var] + 1; self(self, x, var); } }; dfs(dfs, root, -1); assert(c == var_num); table.init(tmp); } void init(std::vector<int> &pare){ int root = -1; int n = pare.size(); std::vector<std::vector<int>> g(n); for (int i = 0; i < n; i++){ if (pare[i] < 0){ assert(root == -1); root = i; } else{ assert(0 <= pare[i] && pare[i] < n); g[pare[i]].push_back(i); } } assert(root != -1); init(g, root); } LCA (std::vector<std::vector<int>> g, int root = 0){ init(g, root); } LCA (std::vector<int> pare){ init(pare); } LCA(){ } int lca(int a, int b){ assert(0 <= std::min(a, b) && std::max(a, b) < var_num); if (a == b) return a; if (E[a] > E[b]) std::swap(a, b); return order[table.prod(E[a], E[b])]; } int dist(int a, int b){ assert(0 <= std::min(a, b) && std::max(a, b) < var_num); return depth[a] + depth[b] - 2 * depth[lca(a, b)]; } int back(int var, int len){ assert(len <= depth[var]); if (len == 0) return var; int l = 0, r = E[var]; while (r - l > 1){ int m = (l + r) / 2; if (depth[var] - depth[order[table.prod(m, E[var])]] < len){ r = m; } else l = m; } return order[table.prod(l, E[var])]; } // a -> b int jump(int a, int b, int len){ int c = lca(a, b); if (len <= depth[a] - depth[c]) return back(a, len); len -= depth[a] - depth[c]; if (len <= depth[b] - depth[c]) return back(b, depth[b] - depth[c] - len); return -1; } }; } #line 26 "h.cpp" void solve(); // POP'N ROLL MUSIC / TOMOO int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int t = 1; // cin >> t; rep(i, 0, t) solve(); } void solve(){ int N; cin >> N; const int B = 320; vector<vector<int>> G(N); rep(i, 0, N - 1){ int a, b; cin >> a >> b; a--, b--; G[a].push_back(b); G[b].push_back(a); } int Q; cin >> Q; vector<int> T(Q), V(Q); rep(i, 0, Q) cin >> T[i] >> V[i], V[i]--; po167::LCA L(G); vector<int> ans(N); vector<int> col(N); vector<int> rem; rep(i, 0, Q){ if (i % B == 0){ rem.clear(); vector<int> C(N, 1); int r = min(i + B, Q); rep(j, i, r) if (T[j] == 1) C[V[j]] = 0; rep(j, 0, N) ans[j] = INF; vector<int> order; rep(j, 0, N) if (C[j] == 0) rem.push_back(j); rep(j, 0, N) if (C[j] == 1 && col[j] == 1){ ans[j] = 0; order.push_back(j); } rep(rp, 0, order.size()){ int a = order[rp]; for (auto x : G[a]) if (ans[x] == INF){ ans[x] = ans[a] + 1; order.push_back(x); } } } if (T[i] == 1) col[V[i]] ^= 1; else{ int a = ans[V[i]]; for (auto x : rem) if (col[x] == 1) chmin(a, L.dist(V[i], x)); cout << a << "\n"; } } }