Weak Pair HDU - 5877 (dfs+线段树！！有根树！！)

题意：给出了一棵有根树!! 注意有根树！！ 询问满足条件的一个节点与其祖先节点的乘积小于K的总和

思路：

因为是有根树，所有不能造双向边，因为这个和这题玩了一下午。。

对于每个节点深搜下去，每次更新线段树，注意要离散化！ 因为数字很大，但节点不多

每次查找的思路是 查找他祖先节点的 满足条件的区间。在回溯时注意将该节点的值删除掉

#include<stdio.h>
#include<string.h>
#include<iostream>
#include <set>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=200005;
struct node
{
    ll sum,num;
} tree[N*4];
ll a[N],b[N];
int vis[N];
ll n,k;
const ll inf=0x3f3f3f3f3f3f3f3f;
vector<ll> vec[N];
int rnu;
int tot;
ll ans ;
void build(int i,int left,int right)
{
    tree[i].sum=0;
    if(left==right)
    {
        return ;
    }
    int mid=(left+right)>>1;
    build(i<<1,left,mid);
    build(i<<1|1,mid+1,right);
    return ;
}
void update(int i,int left,int right,int pos,int val)
{
    tree[i].sum+=val;
    if(left==right&&left==pos)
    {
        return ;
    }
    int mid=(left+right)>>1;
    if(pos<=mid)
        update(i<<1,left,mid,pos,val);
    else update(i<<1|1,mid+1,right,pos,val);
    return ;
}
int query(int i,int left,int right,int nl,int nr)
{
    if(nl==left&&nr==right)
    {
        return tree[i].sum;
    }
    int mid=(left+right)>>1;
    int res=0;
    if(nr<=mid)
        res+=query(i<<1,left,mid,nl,nr);
    else if(nl>mid)
        res+=query(i<<1|1,mid+1,right,nl,nr);
    else {
        res+=query(i<<1,left,mid,nl,mid);
        res+=query(i<<1|1,mid+1,right,mid+1,nr);
    }
    return res;
}
struct eg{
    int v,next;
}edge[N*2];
int head[N];
void add(int u,int v)
{
    edge[++tot].v=v,edge[tot].next=head[u];
    head[u]=tot;
}
void dfs(int u )
{
    ll num;
    if(a[u]!=0)
        num=k/a[u];
    else num=inf;
    int pos=lower_bound(b+1,b+1+rnu,num)-b;
    ans+=query(1,1,rnu,1,pos);
    pos=lower_bound(b+1,b+1+rnu,a[u])-b;
    update(1,1,rnu,pos,1);

    for(int i=head[u]; i!=0; i=edge[i].next)
    {
        int v=edge[i].v;
        dfs(v);
    }
    update(1,1,rnu,pos,-1);
}

int main()
{
    int t;
    scanf("%d",&t);
        while(t--)
        {
            memset(head,0,sizeof(head));
            memset(vis,0,sizeof(vis));
            tot=ans=0;
            scanf("%lld%lld",&n,&k);
            for(int i=1; i<=n; i++)
            {
                scanf("%lld",&a[i]);
                b[i]=a[i];
            }
            for(int i=1;i<=n;i++)
            {
                if(a[i]==0)
                    b[i+n]=inf;
                else b[i+n]=k/a[i];
            }
            sort(b+1,b+1+n*2);
            rnu=unique(b+1,b+1+2*n)-b-1;
            build(1,1,rnu);
            for(int i=1; i<n; i++)
            {
                int u,v;
                scanf("%d%d",&u,&v);
                add(u,v);
                vis[v]++;
            }
            for(int i=1;i<=n;i++)
                if(!vis[i])
                {
                    dfs(i);
                    break;
                }
            printf("%lld\n",ans);
        }

    return 0;
}