HDU-5877 Weak Pair

首先对所有的a[i]，外加k/a[i]离散化

然后从根节点dfs，用树状数组维护路径上的数，对于经过的每个节点，查询小于等于k/a[i]的数的个数

全部加起来就是答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=1e5+10;
struct Edge
{
    int to,next;
}edge1[N]; //邻接表
int tt; //初始化为0
int first[N];//初始化为-1
void add(int a,int b) //添加边
{
    edge1[tt].to=b;
    edge1[tt].next=first[a];
    first[a]=tt++;
}
ll c[2*N],a[N],k,b[2*N];
int n,cnt;
inline int lowbit(int x) { return x & -x;}
inline int getid(ll v){return lower_bound(b,b+cnt,v)-b+1;}
inline ll getk(ll x){return x==0?1e10+1:k/x;}
void update(int idx, int v)
{
    while (idx <= cnt) 
    {
        c[idx] += v;
        idx += lowbit(idx);
    }
}
ll query(int x) // The sum of 1 to x
{
    ll ans = 0;
    while(x > 0) 
    {
        ans += c[x];
        x -= lowbit(x);
    }
    return ans;
}
ll ans;
void dfs(int u)
{
	for(int i=first[u];~i;i=edge1[i].next)
	{
		int v=edge1[i].to;
		ans+=query(getid(getk(a[v])));
		update(getid(a[v]),1);
		dfs(v);
		update(getid(a[v]),-1);
	}
}
bool vis[N];
int main()
{
	int T,u,v;
	while(~scanf("%d",&T))
	while(T--)
	{
		scanf("%d%lld",&n,&k);
		for(int i=1;i<=n;i++)
		{
			scanf("%lld",&a[i]);
			b[i+n-1]=getk(a[i]);
			b[i-1]=a[i];
		}
		sort(b,b+2*n);
		cnt=1;
		for(int i=1;i<2*n;i++)
			if(b[i]!=b[i-1]) b[cnt++]=b[i];
		memset(vis,false,sizeof(vis));
		tt=0;
		memset(first,-1,sizeof(first));
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&u,&v);
			add(u,v);
			vis[v]=true;
		}
		int root=0;
		for(int i=1;i<=n;i++)
			if(!vis[i])
			{
				root=i;
				break;
			}
		memset(c,0,sizeof(c));
		ans=0;
		update(getid(a[root]),1);
		dfs(root);
		printf("%lld\n",ans);
	}
	return 0;
}