CodeChef - ANCESTOR Ancestors in Two Trees

CodeChef - ANCESTOR Ancestors in Two Trees

数据结构，dfs序

题意

给你两颗树，都是n个节点，1是根。问你对于2~n的所有点，在两颗树中的共同祖先有几个。

思路

dfs序的应用吧。其实求出两个树的dfs序，就是公共区间覆盖的问题。

具体做法是，求出第一棵树的dfs序，然后对第二颗树dfs，每到一个点u，就在线段树或树状数组中记录这个编号在第一棵树的节点所掌握的区间。若记为L、R，就是在线段树里对L加1，对R减一。然后查询1到u的和，就是公共祖先个数。dfs完在复原。

因为如果第二课树的u节点在第一棵树里面的区间与第二颗树的区间无交集，那么前缀和的1、-1会抵消。

%潘学姐

代码

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define M(a,b) memset(a,b,sizeof(a))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int MAXN=500007;
const int oo=0x3f3f3f3f;

int sum[MAXN<<2];
void pushup(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; }
void build(int l, int r, int rt)
{
    sum[rt]=0;
    if(l==r) return;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int pos, int v, int l, int r, int rt)
{
    if(l==r)
    {
        sum[rt]+=v;
        return;
    }
    int m=(l+r)>>1;
    if(pos<=m) update(pos, v, lson);
    else update(pos, v, rson);
    pushup(rt);
}
int query(int L, int R, int l, int r, int rt)
{
    if(L<=l&&r<=R) return sum[rt];
    int m=(l+r)>>1;
    int res=0;
    if(L<=m) res+=query(L, R, lson);
    if(m<R) res+=query(L, R, rson);
    return res;
}
struct Edge
{
    int to, ne;
    Edge(int _u, int _v) { to=_u, ne=_v; }
};
vector<Edge> G1, G2;
int head1[MAXN], head2[MAXN];
int in[MAXN], ou[MAXN],res[MAXN];
int dfsnum;
void dfs1(int u, int fa)
{
    in[u]=++dfsnum;
    for(int i=head1[u];~i;i=G1[i].ne)
        if(G1[i].to!=fa)
            dfs1(G1[i].to, u);
    ou[u]=dfsnum;
}
void dfs2(int u, int fa, int n)
{
    update(in[u], 1, 1, n, 1);
    update(ou[u]+1, -1, 1, n, 1);
    res[u]=query(1, in[u], 1, n, 1)-1;
    for(int i=head2[u];~i;i=G2[i].ne)
        if(G2[i].to!=fa)
            dfs2(G2[i].to, u, n);
    update(in[u], -1, 1, n, 1);
    update(ou[u]+1, 1, 1, n, 1);
}
int main()
{
    int T;scanf("%d", &T);
    while(T--)
    {
        M(head1, -1), M(head2, -1);
        G1.clear(), G2.clear();dfsnum=0;
        int n; scanf("%d", &n);
        for(int i=1;i<n;i++)
        {
            int u, v;scanf("%d%d", &u, &v);
            G1.push_back(Edge(v, head1[u]));
            head1[u]=G1.size()-1;
            G1.push_back(Edge(u, head1[v]));
            head1[v]=G1.size()-1;
        }
        for(int i=1;i<n;i++)
        {
            int u, v;scanf("%d%d", &u, &v);
            G2.push_back(Edge(v, head2[u]));
            head2[u]=G2.size()-1;
            G2.push_back(Edge(u, head2[v]));
            head2[v]=G2.size()-1;
        }
        build(1, ++n, 1);
        dfs1(1, -1);dfs2(1, -1, n);
        for(int i=1;i<n;i++)
            printf("%d%c", res[i], i==n-1 ? '\n' : ' ');
    }
    return 0;
}