欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

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

相关标签: 图论