hdu5877dfs,树状数组离散化

第一个要求u 是v 的祖先，那么可以dfs，遍历到v时，所有他上方的都是满足第一条件的u，做多了树就很容易想到在退出某个子树的时候消除这个影响，这样就能保证所有有影响的都是祖先。要求w[u]*w[v]<=k，那么到v的时候，所有小于等于k/w[v]的u都满足，可以想到树状数组。结点的值最大10亿，肯定要离散化，离散化的时候要把k/w[v]加进去一起离散。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <vector>
#include <map>
using namespace std;
const long long maxn=200003;
long long root;
long long sum,k;
long long in[100005];
vector<long long>g[100005];
long long a[100005];
long long b[200005];

long long c[200005];
map<long long,long long>q;

long long Lowbit(long long t)
{
    return t&(t^(t-1));
}
void add(long long x,long long t)
{
    while(x > 0)
    {
        c[x]+=t;
        x -= Lowbit(x);
    }
}
long long Sum(long long li)
{
    long long s=0;
    while(li<200005)
    {
        s+=c[li];
        li=li+Lowbit(li);
    }
    return s;
}

void dfs(long long t)
{
    long long n=g[t].size();
    for(long long i=0; i<n; i++)
    {
        long long v=g[t][i];
        sum+=(long long)Sum(q[a[v]]);
        add(q[k/a[v]],1);
        dfs(v);
        add(q[k/a[v]],-1);
    }
}

int main()
{
    int t,n;
    scanf("%lld",&t);
    while(t--)
    {
        q.clear();
        memset(in,0,sizeof(in));
        memset(c,0,sizeof(c));
        memset(b,0,sizeof(b));
        scanf("%d%lld",&n,&k);
        for(long long i=1; i<=n; i++)
        {
            scanf("%lld",&a[i]);
            b[2*i-2]=a[i];
            if(a[i]!=0)
                b[2*i-1]=k/a[i];
            g[i].clear();
        }
        sort(b,b+2*n);
        long long tot=0,pre=-1;
        for(int i=0; i<2*n; i++)
        {
            if(b[i]!=pre)
            {
                pre=b[i];
                q[pre]=++tot;
            }
        }
//        for(int i=0;i<100;i++)
//        cout<<q[i]<<endl;
        for(int i=0; i<n-1; i++)
        {
            long long aa,bb;
            scanf("%lld%lld",&aa,&bb);
            g[aa].push_back(bb);
            in[bb]++;
        }
        for(int i=1; i<=n; i++)
            if(in[i]==0)
            {
                root=i;
                break;
            }
        sum=0;
        add(q[k/a[root]],1);
        dfs(root);
        printf("%lld\n",sum);
    }
    return 0;
}