数据结构例题2-CSDN博客

题目
我们开一棵线段树记录某个深度中最大的\(dis\)。
同时利用dfs序来让每条长链开一个线段树并且保证不重复。
遍历到某个点时，先处理重儿子，然后把自己加入线段树，计算一端为自己另一端在重儿子子树中的答案。
然后处理每个轻儿子，暴力计算一端在该轻儿子所在子树中，另一端在该点已遍历过的子树中的答案，然后把该轻儿子子树中的所有线段树合并到该点来。

#include<bits/stdc++.h>
#define LL long long
#define pb push_back
#define mp make_pair
#define P pair<int,int>
#define fir first
#define sec second
#define mid ((l+r)>>1)
#define ls p<<1
#define rs p<<1|1
using namespace std;
namespace IO
{
    char ibuf[(1<<21)+1],*iS,*iT;
    char Get(){return (iS==iT? (iT=(iS=ibuf)+fread(ibuf,1,(1<<21)+1,stdin),(iS==iT? EOF:*iS++)):*iS++);}
    int read(){int x=0;char ch=Get();while(ch>57||ch<48)ch=Get();while(ch>=48&&ch<=57)x=x*10+(ch^48),ch=Get();return x;}
}
using namespace IO;
const int N=1000007,M=998244353;
vector<P>E[N];
int len[N],son[N],L[N],R[N],dfn[N],T,n;
LL dis[N],t[N],ans[N],mx[N<<2];
void modify(int p,int l,int r,int x,LL v)
{
    if(l==r) return mx[p]=max(mx[p],v),void();
    (x<=mid? modify(ls,l,mid,x,v):modify(rs,mid+1,r,x,v)),mx[p]=max(mx[ls],mx[rs]);
}
LL query(int p,int l,int r,int L,int R)
{
    if(L<=l&&r<=R) return mx[p];
    return max((L<=mid? query(ls,l,mid,L,R):0),(R>mid? query(rs,mid+1,r,L,R):0));
}
void dfs(int u)
{
    int v,w;
    for(auto x:E[u]) v=x.fir,w=x.sec,dis[v]=dis[u]+w,dfs(v),len[v]+1>len[u]? len[u]=len[son[u]=v]+1:0;
}
void modify(int u,int x,LL v){modify(1,1,n,dfn[u]+x,v);}
LL query(int u,int l,int r){return l=max(l,0),r=min(r,len[u]),(l>r? -1ll<<60:query(1,1,n,dfn[u]+l,dfn[u]+r));}
void dfs2(int u)
{
    dfn[u]=++T;
    if(son[u]) dfs2(son[u]);
    modify(u,0,dis[u]),ans[u]=max(ans[u],query(u,L[u],R[u])-dis[u]);
    int i,v;
    for(auto x:E[u])
    if((v=x.fir)^son[u])
    {
        dfs2(v);
        for(i=0;i<=len[v];++i) ans[u]=max(ans[u],query(u,L[u]-i-1,R[u]-i-1)+(t[i]=query(v,i,i))-(dis[u]<<1));
        for(i=0;i<=len[v];++i) modify(u,i+1,t[i]);
    }
}
int main()
{
    freopen("1.in","r",stdin);
    n=read();int i,u,v,sum=0;
    for(i=1;i<=n;++i) L[i]=read(),R[i]=read(),ans[i]=-1;
    for(v=2;v<=n;++v) u=read(),i=read(),E[u].pb(mp(v,i));
    dfs(1),dfs2(1);
    for(i=1;i<=n;++i) sum=(23333ll*sum+ans[i]+M)%M;
    return !printf("%d",sum);
}

转载于:https://www.cnblogs.com/cjoierShiina-Mashiro/p/11561571.html