Codeforces 1074B Intersecting Subtrees(DFS交互题)
程序员文章站
2022-03-03 11:35:18
...
Codeforces 1074B Intersecting Subtrees(DFS交互题)
题意
给定一棵树,两个人给这棵树的节点标号不一样,每个人都选一个子树(分别标上黄色和绿色),如果有交点,那么输出第一个人给随便一个交点的编号,如果没有交点,输出-1.因为第二个人给数的标号我们不知道,所以我们需要查询对应的顶点信息,用A x查询x节点在对应的第二个人那里标了数字几,用B y查询第二个人那边的y节点在我方这边的编号是多少。用C ans输出答案。限制查询最多不超过5次。
思路
首先我们从B的子树上随便选一个点,查询对应的A节点,如果刚好重合,那么直接输出。
如果不是A的子树上一点,因为是树,所以没有环,那么只可能在其中一个方向上去找子树才有可能找到,否则就会出现这棵树被这一个节点分割成两半变成森林的情况,所以我们只需要去找最近的在A子树内的节点,找到以后再返回来查它在不在B子树上即可。如果在就输出答案,否则就输出-1。
代码
#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
struct edge{
int from;
int to;
int nxt;
edge(int f,int t,int n):from(f),to(t),nxt(n){}
};
vector<edge> edges;
int egs[1005];
void addedge(int f,int t){
edges.emplace_back(f,t,egs[f]);
egs[f]=edges.size()-1;
}
bool aisspecial[1005];
bool bisspecial[1005];
int dfs(int num,int fa){
if(aisspecial[num]){
return num;
}
for(int i=egs[num];i!=-1;i=edges[i].nxt){
if(edges[i].to==fa){
continue;
}
int temp=dfs(edges[i].to,num);
if(temp!=-1){
return temp;
}
}
return -1;
}
int main() {
int t;
cin>>t;
while(t--){
int n;
memset(egs,-1,sizeof egs);
edges.clear();
memset(aisspecial,0,sizeof aisspecial);
memset(bisspecial,0,sizeof bisspecial);
cin>>n;
for(int i=0;i<n-1;i++){
int a,b;
cin>>a>>b;
addedge(a,b);
addedge(b,a);
}
int k1,k2;
cin>>k1;
for(int i=0;i<k1;i++){
int temp;
cin>>temp;
aisspecial[temp]=true;
}
cin>>k2;
int starter=-1;
for(int i=0;i<k2;i++){
int temp;
cin>>temp;
if(starter==-1){
starter=temp;
}
bisspecial[temp]=true;
}
cout<<"B "<<starter<<endl;
cout.flush();
cin>>starter;
if(aisspecial[starter]){
cout<<"C "<<starter<<endl;
cout.flush();
}
else{
int ans=dfs(starter,0);
if(ans==-1){
cout<<"C -1\n";
cout.flush();
}
else{
cout<<"A "<<ans<<endl;
cout.flush();
int fakeans;
cin>>fakeans;
if(bisspecial[fakeans]){
cout<<"C "<<ans<<endl;
cout.flush();
}
else {
cout<<"C -1\n";
cout.flush();
}
}
}
}
return 0;
}
感谢队友阿忠哥的点拨,世上只有阿忠好~