hdu 5877

题意：有根树 1e5 个节点。 u 为v 的祖先,求树中有多少对 a[u]*a[v]<=k k题目中给出。 a[i]为i节点的点权值。

容易想到在深搜过程中维护一个有序序列，在到达节点v时，看序列中有多少<=k的值。但这个序列当时想的是可以用vector 维护，后来发现不行，不知道到底行不行，还有一中方法就是用线段树，或者树状数组。先离散在通过点修改，求和的方法统计每个几点的贡献就行了

#include<stdio.h>
#include<algorithm>
#include<vector>
#include<string.h>
#include<iostream>
using namespace std;
long long  a[100010*2];
long long  sum[100010*8];
long long  b[200010];
long long k;
int  in[100010];
int m;
vector<int>v[100010];
long long res;
void push_up(int rt)
{
    sum[rt] = sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=0;
        return ;
    }
    int mid = (l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    push_up(rt);
}

void update(int pos,int val, int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]+=val;
        return ;
    }
    int mid = (l+r)>>1;
    if(pos<=mid)
        update(pos,val,l,mid,rt<<1);
    else
        update(pos,val,mid+1,r,rt<<1|1);
    push_up(rt);
}

long long query(int L ,int R,int l ,int r ,int rt)
{
    if(L<=l&&r<=R)return sum[rt];
    int mid = (l+r)>>1;
    long long ans = 0;
    if(L<=mid)
        ans+=query(L,R,l,mid,rt<<1);
    if(R>mid)
        ans+=query(L,R,mid+1,r,rt<<1|1);
    return ans ;
}
void dfs(int u)
{
    int  x = lower_bound(a+1,a+1+m,k/b[u])- (a);
    int y = lower_bound(a+1,a+1+m,b[u])-(a);
    res+=query(1,x,1,m,1);
    update(y,1,1,m,1);
    for(int i =0; i<v[u].size(); i++)
    {
        int to=v[u][i];
        dfs(to);
    }
    update(y,-1,1,m,1);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        res =0;
        memset(in,0,sizeof(in));
        int n;
        scanf("%d%lld",&n,&k);
        for(int i =1; i<=n; i++)v[i].clear();
        int cot = 1 ;
        for(int i = 1; i<=n; i++)
        {
            scanf("%lld",&b[i]);
            a[cot++] =b[i];
            a[cot++] = k/a[cot-1];
        }
        for(int i = 1; i<n; i++)
        {
            int u,vg;
            scanf("%d%d",&u,&vg);
            v[u].push_back(vg);
            in[vg]++;
        }
        sort(a+1,a+1+2*n);
        m = unique(a+1,a+1+2*n)-(a+1);
        build(1,m,1);
        for(int i = 1; i<=n; i++)
        {
            if(in[i]==0)
            {
                dfs(i);
                break;
            }
        }
        cout<<res<<endl;
    }
    return 0;
}

猜你喜欢