GDUT 排位赛2.19 E

题目：Milk Visits

题目链接：https://codeforces.ml/group/5yyKg9gx7m/contest/269717/problem/E

题目描述；
给出树状图，每个点上有对应的牛奶类型；
问给出的路中有没有对应的的牛奶；

题目分析：
把1设为树的根，算出1到各点对应俩种牛奶的含点数（每点两种点相加便是深度）。每条路上的牛奶对应点数为h[x]+h[y]-h[fa]- h[fafa] (x,y为起点和终点，fa为它们最近的父结点，fafa为父结点的父结点)；

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>

using namespace std;
char s[2];
int g[100005],h[100005],fa[100005][18];
vector<int>q[100005];
void build(int x,int y) //建树
{
    int i,k;
    fa[x][0]=y; //找直接父结点
    h[x]+=h[y]; //利用差分数点数
    g[x]+=g[y];
    for(i=0,k=q[x].size();i<k;i++)
    {
        if(q[x][i]!=y) build(q[x][i],x);
    }
}
int dp(int x) //算深度
{
    return h[x]+g[x];
}
int fg(int x,int y) //找最近父节点
{
    if(x==y) return x;
    int z,x0,y0,i;
    if(dp(x)>dp(y)) {z=x;x=y;y=z;}
    for(i=17,z=dp(x);z!=dp(y);i--) //把深度调到一样
    {
        if(dp(fa[y][i])>=z) y=fa[y][i];
    }
    if(x==y) return x;
    for(i=17;i>=0;i--) //找最近对应父结点
    {
        x0=fa[x][i];
        y0=fa[y][i];
        if(x0!=y0)
        {
            x=x0;
            y=y0;
        }
    }
    return fa[x][0];    
}
int main()
{
    int n,m,i,r,l,ge,j;
    scanf("%d%d",&n,&m);
    getchar();
    for(i=1;i<=n;i++)
    {
        scanf("%c",s);
        if(s[0]=='H') h[i]=1;
        else g[i]=1;
    }

    for(ge=n-1;ge;ge--)
    {
        scanf("%d%d",&r,&l);
        q[r].push_back(l);
        q[l].push_back(r);
    }
    build(1,0);

    for(i=1;i<18;i++)  //找对应1，2，4.。。的父结点，方变找；
    for(j=1;j<=n;j++)
    fa[j][i]=fa[fa[j][i-1]][i-1];

    while(m--)
    {
        scanf("%d%d%s",&r,&l,s);
        ge=fg(r,l);
        if(s[0]=='H')
        {
            if(h[r]+h[l]-h[ge]-h[fa[ge][0]]) printf("1");
            else printf("0");
        }
        else
        {
            if(g[r]+g[l]-g[ge]-g[fa[ge][0]]) printf("1");
            else printf("0");
        }
    }
    return 0;
}