HDU 6031(CCPC女生赛1009)

题目大意是给一颗树,n个点,1为根,m次询问,每次询问给出两个集合A,B。求集合A和集合B中的点的深度最大的LCA。
大致思路是深搜,处理出每个点的父节点,倍增法处理出anc数组。
每次询问时,二分目标深度,找出集合A中每个点在目标深度的祖先,并插入集合S,再找出集合B中每个点在目标深度的祖先,如果祖先在集合S中,说明这个深度A,B存在公共祖先。
T不大,sumK也不大,可以水过去。

#include
#include
#include
#include
#include
#include
#define mm(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=1e5+5;
const int maxe=2e5+50;
struct Node{
    int v,next;
}edge[maxe];
int head[maxn],cnt;
int dep[maxn],fa[maxn];
int n,m;
int anc[maxn][20];
void init(){
    mm(head,-1);cnt=0;
}
void addedge(int u,int v){
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs(int u,int pre,int depth){
    fa[u]=pre;
    dep[u]=depth;
    for(int k=head[u];k!=-1;k=edge[k].next){
        Node& e=edge[k];
        if(e.v==pre) continue;
        dfs(e.v,u,depth+1);
    }
}
int s1[maxn],s2[maxn];
void process(){
    for(int i=0;i0]=fa[i];
        for(int j=1;(1<1;
    }
    for(int j=1;(1<for(int i=0;iif(anc[i][j-1]!=-1){
            int a=anc[i][j-1];
            anc[i][j]=anc[a][j-1];
        }
    }
}
int query(int p,int depth){
    if(depth<0) return -1;
    if(depth==0) return p;
    for(int i=20;i>=0;i--)
    if(depth>=(1<1<return p;
}
int K1, K2;
bool check(int mid){
    set<int> S;
    for(int j=0;jint dis=dep[s1[j]]-mid;
        int pre=query(s1[j], dis);
        if(pre!=-1) S.insert(pre);
    }
    int flag=0;
    for(int j=0;jint dis=dep[s2[j]]-mid;
        int pre=query(s2[j], dis);
        if(S.count(pre)) return true;
    }
    return false;
}
int main(){

    int u, v;
    while(~scanf("%d%d",&n,&m)){
        init();
        for(int i=1;iscanf("%d%d",&u, &v);
            u--, v--;
            addedge(u,v);
            addedge(v,u);
        }
        dfs(0,-1,1);

        process();
        for(int i=0;iscanf("%d",&K1);
            int L=1, R=1;
            for(int j=0;jscanf("%d",&s1[j]);
                s1[j]--;
                R=max(R, dep[s1[j]]);
            }
            scanf("%d",&K2);
            for(int j=0;jscanf("%d",&s2[j]);
                s2[j]--;
            }
            while(L+1int mid=(L+R)>>1;
                if(check(mid)) L=mid;
                else R=mid-1;
            }
            printf("%d\n",check(R)?R:L);
        }
    }
    return 0;
}

你可能感兴趣的:(------OJ------,HDU)