結果
問題 |
No.3237 Find the Treasure!
|
ユーザー |
👑 ![]() |
提出日時 | 2025-08-15 21:49:07 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 404 ms / 3,000 ms |
コード長 | 4,433 bytes |
コンパイル時間 | 2,066 ms |
コンパイル使用メモリ | 206,776 KB |
実行使用メモリ | 25,972 KB |
平均クエリ数 | 15.52 |
最終ジャッジ日時 | 2025-08-15 21:53:08 |
合計ジャッジ時間 | 10,891 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 22 |
ソースコード
#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;} namespace po167{ struct UF { using _F = int; int _n; std::vector<_F> wei; std::vector<int> q; int component; UF(int n):_n(n), wei(n), component(n), par(n){ for (int i = 0; i < n; i++){ wei[i] =1, par[i] = i; } } void intialize(){ for (auto x : q){ wei[root(x)] = 1; par[x] = x; wei[x] = 1; } component = (int)par.size(); q = {}; } //根っこを返す int root(int a){ assert(0 <= a && a < _n); if (a == par[a]) return a; return par[a] = root(par[a]); } //trueなら1,falseなら0 int same(int a, int b){ assert(0 <= a && a < _n); assert(0 <= b && b < _n); if(root(a) == root(b)) return 1; else return 0; } _F size(int a){ return wei[root(a)]; } //a,bが違う根っこの元なら結合する,結合したらtrueを返す bool unite(int a,int b){ a = root(a), b = root(b); if (a == b) return false; if (wei[a] < wei[b]) std::swap(a, b); par[b] = a; q.push_back(b); wei[a] += wei[b]; component--; return true; } private: std::vector<int> par; }; } using po167::UF; 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; vector<int> A(N - 1), B(N - 1); rep(i, 0, N - 1) cin >> A[i] >> B[i], A[i]--, B[i]--; UF T(N * 2); rep(i, 0, N - 1){ T.unite(A[i], B[i] + N); T.unite(A[i] + N, B[i]); } vector<int> ans(N, 1); auto q = [&](vector<int> v) -> void { cout << "?"; for (auto c : v) cout << " " << c + 1; cout << endl; string res; cin >> res; vector<int> seen(N); for (auto c : v) seen[c] = 1; rep(i, 0, N) if ((seen[i] == 1) ^ (res == "Yes")) ans[i] = 0; }; vector<int> ind(N, -1); { vector<int> tmp; rep(i, 0, N - 1) tmp.push_back(T.same(0, A[i]) ? A[i] : B[i]); q(tmp); int c = 0; rep(i, 0, N) if (ans[i]) ind[i] = c++; rep(i, 0, N - 1) if (ind[A[i]] == -1) swap(A[i], B[i]); } int bit = 0; while ((1 << bit) <= N){ vector<int> tmp; rep(i, 0, N - 1){ tmp.push_back((ind[A[i]] & (1 << bit)) ? A[i] : B[i]); } q(tmp); bit++; } rep(i, 0, N) if (ans[i]){ cout << "! " << i + 1 << endl; } }