HDU 5877 Weak Pair

题意：问一棵树中有多少序数对 < u , v >满足，u是v的祖先节点，并且这两个节点的值相乘小于或等于k。
分析：感觉这个思路还是很妙的。考虑深搜的过程，当搜到一个节点的时候，所有经过的点都是它的祖先节点。所以就可以每搜到一个点就计算经过的点中有多少个点满足和它相乘<=k。这一步可以用树状数组来计算。然后再把这个点加入树状数组。

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

const int N = 2e5 + 5;
typedef long long ll;
ll a[N], s[N], ans, n, m;
int f[N], root, t;
vector<int> son[N];
struct p{
    ll k, id, sid;
}b[N];

int cmp1(p a, p b){
    return a.k < b.k;
}

int cmp2(p a, p b){
    return a.id < b.id;
}

int getId(ll p){
    int l = 0, r = n, ans = 0;
    while(l <= r){
        int mid = (l + r) / 2;
        if(b[mid].k <= p){
            ans = mid;
            l = mid + 1;
        }else r = mid - 1;
    }
    return ans;
}

void set(int k, int t){
    while(k <= n){
        s[k] += t;
        k += k & (-k);
    }
}

ll query(int k){
    ll ans = 0;
    while(k > 0){
        ans += s[k];
        k -= k & (-k);
    }
    return ans;
}

void dfs(int k){
//  cout << k << endl;
    t = b[a[k]].k == 0 ? 0 : getId(m / b[a[k]].k);
    ans += query(t);
    if(son[k].size() == 0){
        return; 
    }
    set(a[k], 1);
    for(int i = 0; i < son[k].size(); i++){
        dfs(son[k][i]);
    }
    set(a[k], -1);
}

int main(){
    int T;
    //freopen("c.txt", "r", stdin);
    scanf("%d", &T);
    for(; T--; ){
        ans = 0;
        memset(s, 0, sizeof(s));
        memset(f, 0, sizeof(f));
        scanf("%lld%lld", &n, &m);
        for(int i = 1; i <= n; i++){
            son[i].clear();
            scanf("%lld", &b[i].k);
            b[i].id = i;
        }
        sort(b + 1, b + n + 1, cmp1);
        for(int i = 1; i <= n; i++){
            a[b[i].id] = i;
        }
        for(int i = 1; i < n; i++){
            int u, v;
            scanf("%d%d", &u, &v);
            f[v]++;
            son[u].push_back(v);
        }

        for(int i = 1; i <= n; i++) if(f[i] == 0) dfs(i);;

        printf("%lld\n", ans);
    }
    return 0;
}