Codeforces 1074B Intersecting Subtrees（DFS交互题）

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;
}

感谢队友阿忠哥的点拨，世上只有阿忠好~