結果
問題 |
No.3237 Find the Treasure!
|
ユーザー |
![]() |
提出日時 | 2025-08-16 21:04:43 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 370 ms / 3,000 ms |
コード長 | 2,233 bytes |
コンパイル時間 | 3,675 ms |
コンパイル使用メモリ | 263,044 KB |
実行使用メモリ | 25,972 KB |
平均クエリ数 | 13.96 |
最終ジャッジ日時 | 2025-08-16 21:04:58 |
合計ジャッジ時間 | 12,975 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 22 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; #define rep(i, n) for(int i=0;i<(n);++i) #define rep1(i, n) for(int i=1;i<=(n);i++) #define ll long long using mint = modint998244353; using P = pair<ll,ll>; using lb = long double; using T = tuple<ll, ll, ll>; #ifdef LOCAL # include <debug_print.hpp> # define dbg(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__) #else # define dbg(...) (static_cast<void>(0)) #endif int main() { int n; cin >> n; vector<int> u(n-1), v(n-1); vector<vector<int>> g(n); rep(i,n-1) { cin >> u[i] >> v[i]; u[i]--;v[i]--; g[u[i]].push_back(v[i]); g[v[i]].push_back(u[i]); } vector<int> dist(n); auto dfs = [&](auto dfs, int u, int p=-1) -> void { for(int v : g[u]) { if(v==p) continue; dist[v] = dist[u]+1; dfs(dfs, v, u); } }; dfs(dfs, 0); set<int> a,b; rep(i,n) { if(dist[i]%2==0) a.insert(i); if(dist[i]%2==1) b.insert(i); } set<int> del; rep(j,15) { if(a.size()==1 && b.size()==0) break; vector<int> vs; rep(i,n-1) { if(a.count(u[i]) && !b.count(u[i])) vs.push_back(u[i]); else if(a.count(v[i]) && !b.count(v[i])) { vs.push_back(v[i]); } else{ if(del.count(u[i])) vs.push_back(u[i]); else if(del.count(v[i])) vs.push_back(v[i]); } } cout << "? "; rep(i,n-1) cout << vs[i]+1 << " "; cout << endl; string s; cin >> s; assert(s!="Invalid"); dbg(a,b); set<int> A,B; if(s=="Yes") { for(auto t : a) { if(A.size()==B.size()) A.insert(t); else B.insert(t); } for(auto d : b) del.insert(d); } else{ for(auto t : b) { if(A.size()==B.size()) A.insert(t); else B.insert(t); } for(auto d : a) del.insert(d); } a = A; b = B; dbg(a,b,del); } cout << "! " << *a.begin()+1 << endl; return 0; }