HDU 5877 Weak Pair（树状数组+dfs+离散化）

http://acm.hdu.edu.cn/showproblem.php?pid=5877

题意：

给出一棵树，每个顶点都有权值，现在要你找出满足要求的点对(u,v)数，u是v的祖先并且a[u]*a[v]<=k。

思路：

转化一下，a[v]<=k/a[u]，k/a[u]的最大值也就是k/a[v]，也就是寻找<=k/a[v]的个数，到这儿，是不是很像树状数组？

我们只需要从根开始dfs，插入到树状数组中，并且查询即可。注意这道题目需要离散化一下。

 #include <iostream>
 #include <cstdio>
 #include <vector>
 #include <cstring>
 #include <algorithm>
 using namespace std;
 const int INF=0x3f3f3f3f;
 typedef long long ll;
 const int maxn=1e6+;

 ll n,k,m;
 ll ans;
 ll a[maxn];
 ll b[maxn];
 ll c[maxn];
 int degree[maxn];

 vector<int> G[maxn];

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

 int sum(int x)
 {
     int ret = ;
     while(x>)
     {
         ret+=c[x]; x-=lowbit(x);
     }
     return ret;
 }

 void add(int x, int d)
 {
     while(x<=m)
     {
         c[x]+=d;
         x+=lowbit(x);
     }
 }

 void dfs(int u, int fa)
 {
     int update_pos=lower_bound(b+,b+m+,a[u])-b;
     int query_pos;
     if(a[u]) query_pos=lower_bound(b+,b+m+,k/a[u])-b;
     else query_pos=m;
     ans+=sum(query_pos);
     add(update_pos,);
     for(int i=;i<G[u].size();i++)
     {
         int v=G[u][i];
         if(v==fa) continue;
         dfs(v,u);
     }
     add(update_pos,-);
 }

 int main()
 {
     //freopen("in.txt","r",stdin);
     int T;
     scanf("%d",&T);
     while(T--)
     {
         memset(c,,sizeof(c));
         memset(degree,,sizeof(degree));
         scanf("%lld%lld",&n,&k);
         for(int i=;i<=n;i++)  {G[i].clear();scanf("%lld",&a[i]);b[i]=a[i];}

         m=n;
         for(int i=;i<=n;i++)
         {
             if(a[i])  b[++m]=k/a[i];
         }

         sort(b+,b+m+);
         m=unique(b+,b+m+)-(b+);

         for(int i=;i<=n-;i++)
         {
             int u,v;
             scanf("%d%d",&u,&v);
             G[u].push_back(v);
             G[v].push_back(u);
             degree[v]++;
         }

         ans=;
         for(int i=;i<=n;i++)
             if(degree[i]==)  dfs(i,-);
         printf("%lld\n",ans);
     }
     return ;
 }