Weak Pair HDU - 5877

点击打开链接

dfs遍历到每一个点 u 时 dfs序列中存的就是这个点的所有祖先 v 了 我们只需要知道有多少祖先的值 per[v] 小于 k/per[u] 而这个功能用线段树实现即可

还有就是需要离散化 先将所有 per[i ]与 k/per[i] 存入数组 然后排序去重 以此数组的值建立线段树(也可通过map建立映射关系) 详见代码

 

#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define N 0x3f3f3f3f3f3f3f3f

struct nodeI
{
    int v;
    int next;
};

struct nodeII
{
    ll l;
    ll r;
    int val;
    int flag;
};

nodeI edge[100010];
nodeII tree[800010];
ll per[100010],pre[200010];
int first[100010],book[100010];
ll k,ans;
int n,m,num;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
    return;
}

void pushup(int cur)
{
    tree[cur].val=tree[cur*2].val+tree[cur*2+1].val;
    return;
}

void build(int l,int r,int cur)
{
    int m;
    tree[cur].l=pre[l];
    tree[cur].r=pre[r];
    tree[cur].val=0;
    tree[cur].flag=0;
    if(l==r)
    {
        tree[cur].flag=1;
        return;
    }
    m=(l+r)/2;
    build(l,m,cur*2);
    build(m+1,r,cur*2+1);
    return;
}

void update(ll tar,int val,int cur)
{
    if(tree[cur].flag)
    {
        tree[cur].val+=val;
        return;
    }
    if(tar<=tree[cur*2].r) update(tar,val,cur*2);
    else update(tar,val,cur*2+1);
    pushup(cur);
    return;
}

ll query(ll pl,ll pr,int cur)
{
    ll res;
    if(pl<=tree[cur].l&&tree[cur].r<=pr)
    {
        return tree[cur].val;
    }
    res=0;
    if(pl<=tree[cur*2].r) res+=query(pl,pr,cur*2);
    if(pr>=tree[cur*2+1].l) res+=query(pl,pr,cur*2+1);
    return res;
}

void dfs(int u)
{
    ll p;
    int i,v,a,b;
    if(per[u]==0) p=N;
    else p=k/per[u];
    ans+=query(0,k/per[u],1);
    update(per[u],1,1);
    for(i=first[u];i!=-1;i=edge[i].next)
    {
        v=edge[i].v;
        if(!book[v])
        {
            book[v]=1;
            dfs(v);
        }
    }
    update(per[u],-1,1);
    return;
}

int main()
{
    int t,i,j,u,v;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%lld",&n,&k);
        for(i=1;i<=n;i++)
        {
            scanf("%lld",&per[i]);
            pre[i]=per[i];
        }
        for(i=n+1;i<=2*n;i++)
        {
            if(per[i-n]==0) pre[i]=N;
            else pre[i]=k/pre[i-n];
        }
        sort(pre+1,pre+2*n+1);
        pre[0]=-1;
        for(i=1,j=0;i<=2*n;i++)
        {
            if(pre[j]!=pre[i])
            {
                pre[++j]=pre[i];
            }
        }
        m=j;
        build(1,m,1);

        memset(first,-1,sizeof(first));
        memset(book,0,sizeof(book));
        num=0;
        for(i=1;i<=n-1;i++)
        {
            scanf("%d%d",&u,&v);
            addedge(u,v);
            book[v]=1;
        }
        ans=0;
        for(i=1;i<=n;i++)
        {
            if(!book[i])
            {
                memset(book,0,sizeof(book));
                book[i]=1;
                dfs(i);
                break;
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}