HDU 5877 Weak Pair

$dfs$序，线段树。

可以统计每一个节点作为$root$的子树上对答案的贡献，可以将树转换成序列。问题就变成了一段区间上求小于等于某个值的数有几个。用线段树记录排好序之后的区间序列，询问的时候，属于询问区间的每个节点二分一下统计答案即可。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<iostream>
using namespace std;
typedef long long LL;
const double pi=acos(-1.0),eps=1e-;
void File()
{
    freopen("D:\\in.txt","r",stdin);
    freopen("D:\\out.txt","w",stdout);
}
template <class T>
inline void read(T &x)
{
    char c=getchar(); x=;
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) {x=x*+c-''; c=getchar();}
}

const int maxn=;
int T,n,h[maxn],sz,r[maxn],root;
LL k,v[maxn];
struct Edge { int u,v,nx; }e[maxn];
LL a[*maxn],L[*maxn],R[*maxn];
vector<int>s[*maxn];

void add(int u,int v)
{
    e[sz].u=u; e[sz].v=v;
    e[sz].nx=h[u]; h[u]=sz++;
}

void dfs(int x)
{
    sz++; a[sz]=v[x]; L[x]=sz;
    for(int i=h[x];i!=-;i=e[i].nx) dfs(e[i].v);
    sz++; a[sz]=v[x]; R[x]=sz;
}

void build(int l,int r,int rt)
{
    if(l==r) { s[rt].push_back(a[l]); return; }
    int m=(l+r)/; build(l,m,*rt); build(m+,r,*rt+);

    int sum=,p1=,p2=;
    while(sum<r-l+)
    {
        if(p1<s[*rt].size()&&p2<s[*rt+].size())
        {
            if(s[*rt][p1]<s[*rt+][p2]) s[rt].push_back(s[*rt][p1]), p1++;
            else s[rt].push_back(s[*rt+][p2]), p2++;
        }
        else if(p1<s[*rt].size()) s[rt].push_back(s[*rt][p1]), p1++;
        else s[rt].push_back(s[*rt+][p2]), p2++;
        sum++;
    }
}

int get(int L,int R,LL num,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        int left=,right=r-l,pos=-;

        while(left<=right)
        {
            int mid=(left+right)/;
            if((LL)s[rt][mid]>num) right=mid-;
            else left=mid+,pos=mid;
        }

        return pos+;
    }

    int m=(l+r)/,x1=,x2=;
    if(L<=m) x1=get(L,R,num,l,m,*rt);
    if(R>m) x2=get(L,R,num,m+,r,*rt+);
    return x1+x2;
}

int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%lld",&n,&k);
        for(int i=;i<=n;i++) scanf("%lld",&v[i]);

        memset(h,-,sizeof h);
        memset(r,sz=,sizeof r);
        for(int i=;i<*maxn;i++) s[i].clear();

        for(int i=;i<=n-;i++)
        {
            int u,v; scanf("%d%d",&u,&v);
            add(u,v); r[v]++;
        }

        for(int i=;i<=n;i++) if(r[i]==) root=i;
        sz=; dfs(root); build(,*n,);

        LL Ans=;
        for(int i=;i<=n;i++)
        {
            if(L[i]+==R[i]) continue;
            if(v[i]==) { Ans=Ans+(R[i]-L[i]-); continue; }
            Ans=Ans+get(L[i]+,R[i]-,k/v[i],,*n,);
        }
        printf("%lld\n",Ans/);
    }
    return ;
}