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
,b
;
int vis
;
ll n,k;
const ll inf=0x3f3f3f3f3f3f3f3f;
vector<ll> vec
;
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
;
void add(int u,int v)
{
edge[++tot
4000
].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;
}