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+5;

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 = 0;
    while(x>0)
    {
        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+1,b+m+1,a[u])-b;
    int query_pos;
    if(a[u]) query_pos=lower_bound(b+1,b+m+1,k/a[u])-b;
    else query_pos=m;
    ans+=sum(query_pos);
    add(update_pos,1);
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
    }
    add(update_pos,-1);
}

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

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

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

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

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