hdu6162 Ch’s gift(LCA)

题意：

给你一棵树，每个节点有一个价值，要执行m次询问，每次询问输入四个数，s,t,a,b，求解s到t的最短路，并把路径上节点的价值在[a,b]之间的价值求和输出。

思路：

对每组询问，求LCA（开始询问前先以1为根节点dfs一遍，确定每个节点的高度，开始每组询问时，根据高度差迭代求lca，详情看代码），并在求解过程判断价值符不符合要求。

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int maxn = 1e5+7;
ll ans[maxn],arr[maxn];
int height[maxn],fa[maxn];
vector<int> e[maxn];
int n,m;
void dfs(int u,int f,int h)
{
    height[u] = h;
    fa[u] = f;
    int len = e[u].size();
    for(int i = 0;i<len;i++)
    {
        int v= e[u][i];
        if(v!=f)
        {
            dfs(v,u,h+1);
        }
    }
}

ll solve(int s,int t,int a,int b)
{
    ll res = 0;
    if(t==1)
    {
        while(s!=0)
        {
            if(arr[s]>=a&&arr[s]<=b)
                res+=arr[s];
            s = fa[s];
        }
        return res;
    }
    if(s==1)
    {
        while(t!=0)
        {
            if(arr[t]>=a&&arr[t]<=b)
                res+=arr[t];
            t = fa[t];
        }
        return res;
    }
    while(1)
    {
        if(height[t]==height[s])
        {
            if(t==s)
            {
                if(arr[s]>=a&&arr[s]<=b)
                    res+=arr[s];
                return res;
            }
            else
            {
                if(arr[s]>=a&&arr[s]<=b)
                    res+=arr[s];
                s = fa[s];
            }
        }
        else if(height[s]>height[t])
        {
            if(arr[s]>=a&&arr[s]<=b)
                    res+=arr[s];
                s = fa[s];
        }
        else
        {
            if(arr[t]>=a&&arr[t]<=b)
                res+=arr[t];
            t = fa[t];
        }
    }
    return res;
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i = 1;i<=n;i++)
            e[i].clear();
        for(int i = 1;i<=n;i++)
            scanf("%lld",arr+i);
        for(int i = 0;i<n-1;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            e[u].push_back(v);
            e[v].push_back(u);
        }
        dfs(1,0,1);
        for(int i = 0;i<m;i++)
        {
            int s,t,a,b;
            scanf("%d%d%d%d",&s,&t,&a,&b);
            ans[i] = solve(s,t,a,b);
        }
        for(int i= 0;i<m;i++)
        {
            if(i!=m-1)
                printf("%lld ",ans[i]);
            else
                printf("%lld\n",ans[i]);
        }
    }
    return 0;
}