BZOJ4458: GTY的OJ

最新推荐文章于 2024-03-01 08:57:26 发布

原创最新推荐文章于 2024-03-01 08:57:26 发布 · 437 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#BZOJ #贪心 #ST表 #堆

蒟蒻zxl的Blog专栏同时被 3 个专栏收录

460 篇文章

订阅专栏

BZOJ

281 篇文章

订阅专栏

其他---贪心

19 篇文章

订阅专栏

该博客主要介绍了如何利用贪心算法，结合ST表和堆的数据结构来解决BZOJ4458题目中的树上问题。文章详细阐述了树上超级钢琴问题的解决思路，强调了构建树上ST表的复杂性，并给出了相应的代码实现。

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

贪心 ST表堆

题目传送门

树上的超级钢琴，我们只需要建树上的ST表即可，稍微复杂了点。

代码：

#include<cmath>
#include<queue>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 500005
#define lg 19
#define il inline
using namespace std;
typedef long long LL;
struct edge{ int next,to; }ed[N<<1];
int n,m,L,R,fa[N][lg+5],sum[N],ST[N][lg+5],dep[N];
struct nd{
    int x,l,r,p;
    bool operator < (const nd &a) const{
        return sum[x]-sum[fa[p][0]]<sum[a.x]-sum[fa[a.p][0]];
    }
};
priority_queue <nd> que; LL ans;
il char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++; 
}
il int _read(){
    int x=0,f=1; char ch=readc();
    while (!isdigit(ch)) { if (ch=='-') f=-1; ch=readc(); }
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x*f;
}
il int Min(int x,int y){ return sum[fa[x][0]]<sum[fa[y][0]]?x:y; }
il void Make(){
    for (int j=1;j<=lg;j++)
        for (int i=1;i<=n;i++){
            fa[i][j]=fa[fa[i][j-1]][j-1];
            if (dep[i]>=(1<<j))
                ST[i][j]=Min(ST[i][j-1],ST[fa[i][j-1]][j-1]);
        }
}
il int find(int x,int y){
    for (int j=lg;j>=0;j--)
        if (y&(1<<j)) x=fa[x][j],y-=1<<j;
    return x;
}
il int srch(int x,int y){
    int ret=ST[x][0];
    for (int j=lg;j>=0;j--)
        if (dep[fa[y][j]]>=dep[x]) 
            ret=Min(ret,ST[y][j]),y=fa[y][j];
    return ret;
}
int main(){
    n=_read();
    for (int i=1;i<=n;i++) fa[i][0]=_read();
    for (int i=1;i<=n;i++){
        int x=_read(); sum[i]=sum[fa[i][0]]+x;
        dep[i]=dep[fa[i][0]]+1,ST[i][0]=i;
    }
    Make(),m=_read(),L=_read(),R=_read();
    for (int i=1;i<=n;i++)
        if (dep[i]>=L){
            int l=max(1,find(i,R-1)),r=find(i,L-1);
            que.push((nd){i,l,r,srch(l,r)});
        }
    while (m--){
        nd x=que.top(); que.pop();
        ans+=(LL)sum[x.x]-sum[fa[x.p][0]];
        if (x.p!=x.l){
            int fx=fa[x.p][0];
            que.push((nd){x.x,x.l,fx,srch(x.l,fx)});
        }
        if (x.p!=x.r){
            int fx=find(x.x,dep[x.x]-dep[x.p]-1);
            que.push((nd){x.x,fx,x.r,srch(fx,x.r)});
        }
    }
    return printf("%lld\n",ans),0;
}