洛谷P3252 [JLOI2012]树

本文提供了一种解决洛谷OJ 3252问题的方法,利用树状数组进行树上前缀和计算,通过离散化处理路径长度查询问题。文章详细介绍了如何将路径长度查询转化为树状数组上的查询,并给出完整代码。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

链接

  https://www.luogu.org/problem/show?pid=3252

题解

  如果权值都是正数的话,那很容易,但这道题的后6个点权值有负数出现(我被坑了)。
  设sum[x]表示从根节点到节点x的权值和,那么一条路径的长度等于S,就等价于sum的差等于S,即sum[x]-sum[y]=S指的是点x和y的儿子构成长度为S的链。
  变形一下,得到sum[x]=sum[y]+x。
  从根节点开始dfs,那么当前正在访问的点恰好拉成一条链,如果我当前在点x,你要看当前链上有多少点y,其sum[y]+S恰好等于sum[x],计入答案就好。
  和很显然就可以用权值树状数组了,离散然后做就行了。时间复杂度虽然是 O(NlogN) ,但在bzoj上依然484ms过。(这可能因为是树状数组常数小)

代码

//树上前缀和、树状数组 
#include <cstdio>
#include <algorithm>
#define ll long long
#define maxn 300000
#define lowbit(x) (x&-x)
using namespace std;
ll N, S, w[maxn], ans, head[maxn], to[maxn], s[maxn], nex[maxn], tot, tmp[maxn],
    ta[maxn];
void adde(ll a, ll b){to[++tot]=b;nex[tot]=head[a];head[a]=tot;}
void add(ll pos, ll v){for(;pos<=*tmp;pos+=lowbit(pos))ta[pos]+=v;}
ll read(ll x=0)
{
    char c=getchar();
    while(c<48 or c>57)c=getchar();
    while(c>=48 and c<=57)x=(x<<3)+(x<<1)+c-48,c=getchar();
    return x;
}
ll sum(ll pos)
{
    ll ans=0;
    for(;pos;pos-=lowbit(pos))ans+=ta[pos];
    return ans;
}
void init()
{
    ll i, x, y;
    N=read(), S=read();
    for(i=1;i<=N;i++)w[i]=read();
    for(i=1;i<N;i++)x=read(), y=read(), adde(x,y);
}
void lisan()
{
    ll i, x=0;
    tmp[++*tmp]=S;
    for(i=1;i<=N;i++)tmp[++*tmp]=s[i],tmp[++*tmp]=s[i]+S;
    sort(tmp+1,tmp+*tmp+1);
    for(i=1;i<=*tmp;i++)if(tmp[i]!=tmp[i-1])tmp[++x]=tmp[i];
    *tmp=x;
}
ll find(ll x){return lower_bound(tmp+1,tmp+tmp[0]+1,x)-tmp;}
void dfs1(ll pos, ll sum)
{
    ll p;
    s[pos]=sum+w[pos];
    for(p=head[pos];p;p=nex[p])dfs1(to[p],s[pos]);
}
void dfs2(ll pos)
{
    ll p, t=find(s[pos]);
    ans+=sum(t)-sum(t-1);
    add(find(s[pos]+S),1);
    for(p=head[pos];p;p=nex[p])dfs2(to[p]);
    add(find(s[pos]+S),-1);
}
int main()
{
    init();
    dfs1(1,0);
    lisan();
    add(find(S),1);
    dfs2(1);
    printf("%lld",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值