UVALive - 6712 lca+dfs序线段树

题意:一棵树q次查询,每次查询给三个不同的点,要求计算到这三个点的比其他两个距离都要小的点数

题解:很明显的lca,倍增的找中点,关键是两个点的中点很好找,但是三个点不好找,我刚开始还准备分类讨论,后来发现巨麻烦,其实可以用线段树来维护算a的答案其实就是a在b下的答案和a在c下的答案的交集,可以用线段树区间求和区间查询做,每次更新完之后复原就不用memset线段树了

//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pi acos(-1.0)
#define ll long long
#define vi vector<int>
#define mod 1000000007
#define C 0.5772156649
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define pil pair<int,ll>
#define pli pair<ll,int>
#define pii pair<int,int>
#define cd complex<double>
#define ull unsigned long long
#define base 1000000000000000000
#define fio ios::sync_with_stdio(false);cin.tie(0)

using namespace std;

const double g=10.0,eps=1e-;
const int N=+,maxn=+,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f;

vi v[N];
int dep[N],n,sz[N],fa[][N];
int le[N],ri[N],id[N],cnt;
void dfs(int u,int f)
{
    le[u]=++cnt;
    id[cnt]=u;
    fa[][u]=f;
    sz[u]=;
    for(int i=;i<v[u].size();i++)
    {
        int x=v[u][i];
        if(x!=f)dep[x]=dep[u]+,dfs(x,u),sz[u]+=sz[x];
    }
    ri[u]=cnt;
}
int lazy[N<<],val[N<<];
void pushdown(int l,int r,int rt)
{
    if(lazy[rt]!=)
    {
        int m=(l+r)>>;
        val[rt<<]+=(m-l+)*lazy[rt];
        val[rt<<|]+=(r-m)*lazy[rt];
        lazy[rt<<]+=lazy[rt];
        lazy[rt<<|]+=lazy[rt];
        lazy[rt]=;
    }
}
void pushup(int rt)
{
    val[rt]=val[rt<<]+val[rt<<|];
}
void build(int l,int r,int rt)
{
    lazy[rt]=val[rt]=;
    if(l==r)return ;
    int m=(l+r)>>;
    build(ls);build(rs);
}
void update(int c,int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        val[rt]+=(r-l+)*c;
        lazy[rt]+=c;
        return ;
    }
    pushdown(l,r,rt);
    int m=(l+r)>>;
    if(L<=m)update(c,L,R,ls);
    if(m<R)update(c,L,R,rs);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)return val[rt];
    pushdown(l,r,rt);
    int m=(l+r)>>,ans=;
    if(L<=m)ans+=query(L,R,ls);
    if(m<R)ans+=query(L,R,rs);
    return ans;
}
void init()
{
    dep[]=;
    cnt=;
    dfs(,-);
    build(,cnt,);
    for(int i=;i<;i++)
        for(int j=;j<=n;j++)
            fa[i][j]=fa[i-][fa[i-][j]];
}
int lca(int x,int y)
{
    if(dep[x]>dep[y])swap(x,y);
    for(int i=;i<;i++)
        if((dep[y]-dep[x])>>i&)
            y=fa[i][y];
    if(x==y)return x;
    for(int i=;i>=;i--)
    {
        if(fa[i][x]!=fa[i][y])
        {
            x=fa[i][x];
            y=fa[i][y];
        }
    }
    return fa[][x];
}
int go(int u,int dis)
{
    for(int i=;i>=;i--)
        if(dis>=(<<i))
            dis-=(<<i),u=fa[i][u];
    return u;
}
int solve(int a,int b,int c)
{
    int tle,tri,ans=;
    if(dep[a]>=dep[b])
    {
        int dis=dep[a]+dep[b]-*dep[lca(a,b)];
        int x=go(a,dis/);
        if(dis%==)x=go(a,dis/-);
        update(,le[x],ri[x],,cnt,);
        tle=le[x],tri=ri[x];
    }
    else
    {
        int dis=dep[a]+dep[b]-*dep[lca(a,b)];
        int x=go(b,dis/);
        update(,,cnt,,cnt,);
        update(-,le[x],ri[x],,cnt,);
        tle=le[x],tri=ri[x];
    }
    if(dep[a]>=dep[c])
    {
        int dis=dep[a]+dep[c]-*dep[lca(a,c)];
        int x=go(a,dis/);
        if(dis%==)x=go(a,dis/-);
        ans=query(le[x],ri[x],,cnt,);
    }
    else
    {
        int dis=dep[a]+dep[c]-*dep[lca(a,c)];
        int x=go(c,dis/);
        ans=query(,cnt,,cnt,);
        ans-=query(le[x],ri[x],,cnt,);
    }
    if(dep[a]>=dep[b])update(-,tle,tri,,cnt,);
    else update(-,,cnt,,cnt,),update(,tle,tri,,cnt,);
    return ans;
}
int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=;i<=n;i++)v[i].clear();
        for(int i=;i<n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            v[a].pb(b),v[b].pb(a);
        }
        init();
        int q;scanf("%d",&q);
        while(q--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
//            printf("%d\n",solve(b,a,c));
            printf("%d %d %d\n",solve(a,b,c),solve(b,a,c),solve(c,a,b));
        }
    }
    return ;
}
/***********************
1
9
1 2
1 3
1 4
2 5
2 6
2 7
6 8
6 9
2
1 2 8
2 1 4
***********************/