QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#758147 | #9570. Binary Tree | wht11 | WA | 0ms | 5888kb | C++20 | 5.7kb | 2024-11-17 16:11:52 | 2024-11-17 16:12:02 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
using PII = pair<int, int>;
#define endl "\n"
const int N = 2e5 + 5;
vector<int> G[N];
int siz[N], f[N], vis[N], cnt[N];
int lst = 0;
int mid = 0;
void dfs(int u, int fa)
{
f[u] = fa;
siz[u] = 1;
for (auto v : G[u])
{
if (v == fa || vis[v])
continue;
dfs(v, u);
siz[u] += siz[v];
}
if (vis[u])
return;
if (siz[u] == lst / 2 || lst - siz[u] == lst / 2)
mid = u;
}
void del(int u, int fa)
{
vis[u] = 1;
cnt[u]--;
cnt[fa]--;
siz[u] = 1;
for (auto v : G[u])
{
if (v == fa || vis[v])
continue;
siz[u] += siz[v];
del(v, u);
}
}
void solve()
{
int n;
cin >> n;
for (int i = 1; i <= n; i++)
{
G[i].clear();
vis[i] = 0;
f[i] = 0;
siz[i] = 0;
cnt[i]=0;
}
for (int i = 1; i <= n; i++)
{
int u, v;
cin >> u >> v;
if (v)
{
G[i].push_back(v);
G[v].push_back(i);
cnt[i]++;
cnt[v]++;
}
if (u)
{
G[i].push_back(u);
G[u].push_back(i);
cnt[i]++;
cnt[u]++;
}
}
lst = n;
int rt = 1;
mid = 0;
while (1)
{
int x;
if (lst == 3)
{
int num = 0;
if (cnt[rt] == 2)
num = rt;
for (auto v : G[rt])
{
if (vis[v])
continue;
if (cnt[v] == 2)
num = v;
}
vector<int> ve;
for (auto v : G[num])
{
if (vis[v])
continue;
ve.push_back(v);
}
cout << "? " << ve[0] << " " << ve[1] << endl;
cout.flush();
cin >> x;
if (x == 0)
{
cout << "! " << num << endl;
cout.flush();
return;
}
else if (x == 1)
{
cout << "! " << ve[0] << endl;
cout.flush();
return;
}
else
{
cout << "! " << ve[1] << endl;
cout.flush();
return;
}
}
mid = rt;
dfs(rt, 0);
if (cnt[mid] == 3)
{
vector<pair<int, int>> ve;
for (auto v : G[mid])
{
if (vis[v])
continue;
if (v == f[mid])
ve.emplace_back(lst - siz[mid], v);
else
ve.emplace_back(siz[v], v);
}
sort(ve.begin(), ve.end(), greater<pair<int, int>>());
int mx1 = 0, mx2 = 0;
int mx11 = 0, mx22 = 0;
for (auto tmp : ve)
{
int val = tmp.first;
int v = tmp.second;
if (val > mx1)
{
mx2 = mx1;
mx1 = val;
mx22 = mx11;
mx11 = v;
}
else if (val > mx2)
{
mx2 = val;
mx22 = v;
}
}
cout << "? " << mx11 << " " << mx22 << endl;
cout.flush();
cin >> x;
if (x == 0)
{
del(mid, mx11);
lst -= siz[mid];
rt = mx11;
}
else if (x == 1)
{
del(mx22, mid);
lst -= siz[mx22];
del(mx11, mid);
lst -= siz[mx11];
rt = mid;
}
else
{
del(mid, mx22);
lst -= siz[mid];
rt = mx22;
}
}
else if (cnt[mid] == 2)
{
vector<int> ve;
for (auto v : G[mid])
{
if (vis[v])
continue;
ve.push_back(v);
}
cout << "? " << ve[0] << " " << ve[1] << endl;
cout.flush();
cin >> x;
if (x == 0)
{
del(mid, ve[0]);
lst -= siz[mid];
rt = ve[0];
}
else if (x == 1)
{
cout << "! " << mid << endl;
return;
}
else
{
del(mid, ve[1]);
lst -= siz[mid];
rt = ve[1];
}
}
else if (cnt[mid] == 1)
{
int xx;
for (auto v : G[mid])
{
if (vis[v])
continue;
xx=v;
}
cout << "? " << mid <<" " << xx << endl;
cout.flush();
cin >>x;
if(x==1)
{
cout << "! " << mid << endl;
cout.flush();
}
else
{
cout << "! " <<xx<< endl;
cout.flush();
}
return;
}
else
{
cout << "! " << rt << endl;
cout.flush();
return;
}
}
}
signed main()
{
IOS;
int t = 1;
cin >> t;
while (t--)
{
solve();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 0ms
memory: 5888kb
input:
2 5 0 0 1 5 2 4 0 0 0 0 2 2 2 0 2 0 0 0
output:
? 4 2 ? 5 1 ! 1 ? 2 1 ! 1
result:
wrong answer Expecting 2 but found 1. (test case 2)