HDU 6031(CCPC女生赛1009)

题目大意是给一颗树,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
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值