Codeforces Round #746 (Div. 2) D. Hemose in ICPC ?(交互,dfs序)

Posted 2021-11-01 issue是fw

LINK

给一棵$n$个节点、边权未知的树

定义两个节点之间的距离为路径上所有边权的$gcd$。每次你可以询问一个节点集合

然后交互程序会返回所给集合中最大的距离,求在不超过$12$次询问的前提下

距离最大的两个节点的编号。($n<=10^3$)

---

先询问一次所有节点得到$mx$

然后注意到$2^{10}>=1000$,那么肯定与二分有关

我们处理出这棵树的$dfs$序

直接在$dfs$序上二分可以找到最大值的那个点$u$

于是输出$u$和$u$的父亲即可

#include <bits/stdc++.h>
using namespace std;
const int maxn = 3e5+10;
int n,id,topf[maxn],dfn[maxn],father[maxn],mx;
vector<int>vec[maxn];
void dfs(int u,int fa)
{
	dfn[u] = ++id; topf[id] = u; father[u] = fa;
	for(auto v:vec[u] )
	{
		if( v==fa )	continue;
		dfs( v,u );
	}
}
bool isok(int x)
{
	cout << "? " << x;
	for(int i=1;i<=x;i++)	cout << " " << topf[i];	
	cout << endl;
	int z; cin >> z;
	return z==mx;
}
int main()
{
	cin >> n;
	for(int i=1;i<n;i++)
	{
		int l,r; cin >> l >> r;
		vec[l].push_back( r ); vec[r].push_back( l );	
	}
	cout << "? " << n;
	for(int i=1;i<=n;i++)	cout << " " << i;	
	cout << endl;
	cin >> mx;	
	
	dfs( 1,0 );
	int l = 2, r = n, ans = 0;
	while( r>=l )
	{
		int mid = l+r>>1;
		if( isok(mid) )	r = mid-1, ans = mid;	
		else	l = mid+1;
	}
	cout << "! ";
	cout << topf[ans] << " " << father[topf[ans]] << endl;
}