HDU 6031(CCPC女生赛1009)

来源：互联网发布：标书软件破解版编辑：程序博客网时间：2024/05/19 08:40

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

#include<stdio.h>#include<string.h>#include<stdlib.h>#include<math.h>#include<algorithm>#include<set>#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;i<n;i++){        anc[i][0]=fa[i];        for(int j=1;(1<<j)<n;j++)            anc[i][j]=-1;    }    for(int j=1;(1<<j)<n;j++){        for(int i=0;i<n;i++)        if(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<<i)){        p=anc[p][i];        depth-=(1<<i);    }    return p;}int K1, K2;bool check(int mid){    set<int> S;    for(int j=0;j<K1;j++){        int dis=dep[s1[j]]-mid;        int pre=query(s1[j], dis);        if(pre!=-1) S.insert(pre);    }    int flag=0;    for(int j=0;j<K2;j++){        int 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;i<n;i++){            scanf("%d%d",&u, &v);            u--, v--;            addedge(u,v);            addedge(v,u);        }        dfs(0,-1,1);        process();        for(int i=0;i<m;i++){            scanf("%d",&K1);            int L=1, R=1;            for(int j=0;j<K1;j++){                scanf("%d",&s1[j]);                s1[j]--;                R=max(R, dep[s1[j]]);            }            scanf("%d",&K2);            for(int j=0;j<K2;j++){                scanf("%d",&s2[j]);                s2[j]--;            }            while(L+1<R){                int mid=(L+R)>>1;                if(check(mid)) L=mid;                else R=mid-1;            }            printf("%d\n",check(R)?R:L);        }    }    return 0;}

0 1