2019河南省CCPC省赛:大小接近的点对(数状数组+dfs序)

题目链接:http://acm.zzuli.edu.cn/problem.php?id=2520
题目大意:在这里插入图片描述
在这里插入图片描述

思路:先把可能出现的数组离散化,直接dfs,用数状数组维护每个数出现的次数,a[i]=i出现的次数,利用dfs序的思想,访问一个点的时候,统计现在A=[ a[i]-k, a[i]+k ]的元素个数。把自己的值加入数状数组,继续访问它的子树,当返回这个点时,统计现在B=[ a[i]-k, a[i]+k ]的元素个数。B-A就是子树对自己的贡献。再加上所有子树的 sum[] 就是这个点的 sum[] 。时间复杂度:O(n*log(n))

#include<bits/stdc++.h>
#define LL long long
using namespace std;

LL n, sum[500010]={0}, a[500010], b[500010], s[100010]={0};

void add(LL p, LL y)
{
    while(p<=n)
    {
        sum[p]+=y;
        p+=(p&-p);
    }
}

LL cx(LL p)
{
    LL ans=0;
    while(p)
    {
        ans+=sum[p];
        p-=(p&-p);
    }
    return ans;
}

vector<LL> v;
vector<LL> e[100005];
LL k, cut, vis[100010]={0};
LL dfs(LL u)
{
    vis[u]=1;
    LL T=lower_bound(v.begin(), v.begin()+cut,a[u])-v.begin()+1;
    LL L=lower_bound(v.begin(), v.begin()+cut,a[u]-k)-v.begin()+1;
    LL R=lower_bound(v.begin(), v.begin()+cut,a[u]+k)-v.begin()+1;
    s[u]-=(cx(R)-cx(L-1));//A
    add(T, 1);//把自己加入数状数组
    for(LL i=0;i<e[u].size();i++)
    {
        LL to=e[u][i];
        if(!vis[to])
        {
            dfs(to);
            s[u]+=s[to];//加上子树的sum
        }
    }
    s[u]+=(cx(R)-cx(L-1));//B
}

int main()
{
    LL n;
    scanf("%lld%lld",&n,&k);
    for(LL i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);

        v.push_back(a[i]);
        v.push_back(a[i]-k);
        v.push_back(a[i]+k);
    }

    cut=unique(v.begin(), v.end())-v.begin();
    ::n=cut;
    sort(v.begin(), v.begin()+cut);
    for(LL i=2;i<=n;i++)
    {
        LL x;
        scanf("%lld",&x);
        e[x].push_back(i);
        e[i].push_back(x);
    }
    dfs(1);
    for(LL i=1;i<=n;i++)
    {
        printf("%lld\n",s[i]);
    }

	return 0;
}

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值