QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#758063 | #9570. Binary Tree | wht11 | WA | 1ms | 5848kb | C++20 | 4.6kb | 2024-11-17 15:27:33 | 2024-11-17 15:27:34 |
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;
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)
{
cout << "! " << mid << endl;
return;
}
else if (x == 1)
{
del(ve[1], mid);
lst -= siz[ve[1]];
rt = ve[0];
}
else
{
del(ve[0], mid);
lst -= ve[0];
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;
}
详细
Test #1:
score: 0
Wrong Answer
time: 1ms
memory: 5848kb
input:
2 5 0 0 1 5 2 4 0 0 0 0 2 2
output:
? 4 2 ? 3 2 ! 2
result:
wrong answer There are 3 candidates. (test case 1)