题意: 有根树 1e5 个节点。 u 为v 的祖先,求树中有多少对 a[u]*a[v]<=k k题目中给出。 a[i]为i节点的点权值。
容易想到在深搜过程中维护一个有序序列,在 到达节点v时 ,看序列中有多少<=k的值。 但这个序列当时想的是可以用vector 维护,后来发现不行,不知道到底行不行,还有一中方法就是用线段树,或者树状数组。先离散 在通过点修改,求和的方法统计每个几点的贡献就行了
#include<stdio.h>
#include<algorithm>
#include<vector>
#include<string.h>
#include<iostream>
using namespace std;
long long a[100010*2];
long long sum[100010*8];
long long b[200010];
long long k;
int in[100010];
int m;
vector<int>v[100010];
long long res;
void push_up(int rt)
{
sum[rt] = sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
if(l==r)
{
sum[rt]=0;
return ;
}
int mid = (l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
push_up(rt);
}
void update(int pos,int val, int l,int r,int rt)
{
if(l==r)
{
sum[rt]+=val;
return ;
}
int mid = (l+r)>>1;
if(pos<=mid)
update(pos,val,l,mid,rt<<1);
else
update(pos,val,mid+1,r,rt<<1|1);
push_up(rt);
}
long long query(int L ,int R,int l ,int r ,int rt)
{
if(L<=l&&r<=R)return sum[rt];
int mid = (l+r)>>1;
long long ans = 0;
if(L<=mid)
ans+=query(L,R,l,mid,rt<<1);
if(R>mid)
ans+=query(L,R,mid+1,r,rt<<1|1);
return ans ;
}
void dfs(int u)
{
int x = lower_bound(a+1,a+1+m,k/b[u])- (a);
int y = lower_bound(a+1,a+1+m,b[u])-(a);
res+=query(1,x,1,m,1);
update(y,1,1,m,1);
for(int i =0; i<v[u].size(); i++)
{
int to=v[u][i];
dfs(to);
}
update(y,-1,1,m,1);
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
res =0;
memset(in,0,sizeof(in));
int n;
scanf("%d%lld",&n,&k);
for(int i =1; i<=n; i++)v[i].clear();
int cot = 1 ;
for(int i = 1; i<=n; i++)
{
scanf("%lld",&b[i]);
a[cot++] =b[i];
a[cot++] = k/a[cot-1];
}
for(int i = 1; i<n; i++)
{
int u,vg;
scanf("%d%d",&u,&vg);
v[u].push_back(vg);
in[vg]++;
}
sort(a+1,a+1+2*n);
m = unique(a+1,a+1+2*n)-(a+1);
build(1,m,1);
for(int i = 1; i<=n; i++)
{
if(in[i]==0)
{
dfs(i);
break;
}
}
cout<<res<<endl;
}
return 0;
}