hdu 5877 (dfs+树状数组) Weak Pair

题目：这里

题意：

给出一个n个结点的树和一个数k，每个结点都有一个权值，问有多少对点(u，v)满足u是v的祖先结点且二者的权值之积小于等于k、

从根结点开始dfs，假设搜的的点的权值是v，我们需要的是在此之前搜的点中小于等于k/v的数的个数，于是用树状数组查询，查完后将v加入树状数组以供下个

点查询，回溯的时候再一个个的删除将其树状数组。由于这个权值和k的范围比较大，所以得先离散化后在加入树状数组。

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #include<vector>
 using namespace std;

 typedef long long ll;
 const int M = 1e5 + ;
 int head[M],cas,du[M],has[M],len;
 ll a[M],b[M],k,sum;
 bool vis[M];

 int max(int x,int y){return x>y?x:y;}
 struct Edge{
     int to,next;
 }edge[M*];

 void add(int u,int v)
 {
     edge[++cas].next=head[u];
     edge[cas].to=v;
     head[u]=cas;
 }

 int lowbit(int x) {return x&(-x);}

 void shuadd(int x,int y)
 {
     while (x<=len){
         has[x]+=y;
         x+=lowbit(x);
     }
 }

 int getsum(int x){
     int ans=;
     while (x>){
         ans+=has[x];
         x-=lowbit(x);
     }
     return ans;
 }

 int sreach(ll x)
 {
     int l=,r=len,ans=;
     while (l<=r){
         int mid=(l+r)/;
         if (b[mid]<=x) ans=max(ans,mid),l=mid+;
         else r=mid-;
     }
     return ans;
 }

 void dfs(int u)
 {
     int pos;
     if (a[u]!=) pos=sreach(k/a[u]);
     else pos=len;
     sum+=getsum(pos);
     //cout<<sum<<endl;
     shuadd(sreach(a[u]),);
     for (int i=head[u] ; i ; i=edge[i].next)
     {
         int v=edge[i].to;
         if (vis[v]) continue;
         vis[v]=true;
         dfs(v);vis[v]=false;
         shuadd(sreach(a[v]),-);
     }
 }

 int main()
 {
     int t;
     scanf("%d",&t);
     while (t--){
         int n,j=;cas=;
         scanf("%d%I64d",&n,&k);
         for (int i= ; i<=n ; i++) {
             scanf("%I64d",&a[i]);
             if (a[i]==) continue;
             b[i]=a[i];
         }
         sort(b+,b+n+);
         len=unique(b+,b+n+)-b;
         sort(b+,b+len);len--;
        // cout<<len<<endl;
         memset(du,,sizeof(du));
         memset(vis,false,sizeof(vis));
         memset(head,,sizeof(head));
         memset(has,,sizeof(has));
         for (int i= ; i<n ; i++){
             int u,v;
             scanf("%d%d",&u,&v);
             add(u,v);
             du[v]++;
         }
         //cout<<len<<endl;
         sum=;
         for (int i= ; i<=n ; i++)
         {
             if (du[i]) continue;
             dfs(i);
         }
         printf("%I64d\n",sum);
     }
     return ;
 }