hdu5877 Weak Pair(dfs+线段树+离散化)

给定一棵有根树（注意跟不一定是1.。。），求有序对(u,v)满足的个数：
1.u是v的祖先
2.a[u]*a[v]<=k。
我们在对树进行dfs时，做到v点，此时在我们的dfs栈中的点正是v的祖先，一个不多，一个不少，我们只需在这些点中找到小于等于k/a[v]的点。显然可以把权值离散化了，用线段树来维护

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;#define ll long long#define inf 0x3f3f3f3f#define N 100010inline ll read(){    ll x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();    return x*f;}int tst,n,a[N],aa[N],m,h[N],num,in[N];ll kk,ans;struct edge{    int to,next;}data[N];struct node{    int x;}tree[N<<2];void build(int p,int l,int r){    tree[p].x=0;    if(l==r) return;int mid=l+r>>1;    build(p<<1,l,mid);build(p<<1|1,mid+1,r);}void add(int p,int l,int r,int x,int val){    tree[p].x+=val;    if(l==r) return;    int mid=l+r>>1;    if(x<=mid) add(p<<1,l,mid,x,val);    else add(p<<1|1,mid+1,r,x,val);}int qsum(int p,int l,int r,int x,int y){    if(x>y) return 0;    if(x<=l&&r<=y) return tree[p].x;    int mid=l+r>>1,res=0;    if(x<=mid) res+=qsum(p<<1,l,mid,x,y);    if(y>mid) res+=qsum(p<<1|1,mid+1,r,x,y);    return res;}void dfs(int x){    ans+=qsum(1,1,m,1,upper_bound(aa+1,aa+m+1,kk/a[x])-aa-1);    add(1,1,m,lower_bound(aa+1,aa+m+1,a[x])-aa,1);    for(int i=h[x];i;i=data[i].next) dfs(data[i].to);    add(1,1,m,lower_bound(aa+1,aa+m+1,a[x])-aa,-1);}int main(){//  freopen("a.in","r",stdin);    tst=read();    while(tst--){        n=read();kk=read();num=0;memset(h,0,sizeof(h));ans=0;memset(in,0,sizeof(in));        for(int i=1;i<=n;++i) a[i]=aa[i]=read();        sort(aa+1,aa+n+1);m=unique(aa+1,aa+n+1)-aa-1;        for(int i=1;i<n;++i){            int x=read(),y=read();data[++num].to=y;data[num].next=h[x];h[x]=num;in[y]++;        }build(1,1,m);        for(int i=1;i<=n;++i) if(!in[i]) dfs(i);//人家可没说根是1.。。         printf("%lld\n",ans);    }    return 0;}