题意:求一棵树上每个点的子树中经过这个点的所有路径中边数在给定范围内的最长路径
早上8:05开考,8:10分看完题就秒了
然而有个地方没有开longlong
而且线段树维护长链的方向反了
所以连部分分都没了
裸的长链剖分,需要用线段树维护区间最大值,最好是zkw线段树(偷懒不想写)
Code:
#include<bits/stdc++.h>
#define mod 998244353
#define ll long long
#define pb push_back
#define mp make_pair
#define db double
#define ri register
#define fi first
#define se second
using namespace std;
inline int read(){
int res=0,f=1;char ch=getchar();
while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
return res*f;
}
const int N=1e6+5,bs=23333;
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;if(x<0) x+=mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
namespace segtree{
struct seg{int l,r;ll add,mx;}tr[N<<2];
#define ls tr[k].l
#define rs tr[k].r
#define mid (ls+rs>>1)
inline void pushup(int k){tr[k].mx=max(tr[k<<1].mx,tr[k<<1|1].mx);}
inline void pushadd(int k,ll v){tr[k].add+=v,tr[k].mx+=v;}
inline void pushdown(int k){
if(tr[k].add){
pushadd(k<<1,tr[k].add);pushadd(k<<1|1,tr[k].add);
tr[k].add=0;
}
}
void build(int k,int l,int r){
ls=l,rs=r,tr[k].add=0;
if(l==r) return;
build(k<<1,l,mid);build(k<<1|1,mid+1,r);
}
void modify(int k,int pos,ll x){
if(ls==rs) {tr[k].mx=max(tr[k].mx,x);return;}
pushdown(k);
if(pos<=mid) modify(k<<1,pos,x);
else modify(k<<1|1,pos,x);
pushup(k);
}
void modify_seq(int k,int ql,int qr,ll v){
if(ql<=ls && rs<=qr) return pushadd(k,v);
pushdown(k);
if(qr<=mid) modify_seq(k<<1,ql,qr,v);
else if(ql>mid) modify_seq(k<<1|1,ql,qr,v);
else modify_seq(k<<1,ql,mid,v),modify_seq(k<<1|1,mid+1,qr,v);
pushup(k);
}
ll query(int k,int ql,int qr){
if(ql<=ls && rs<=qr) return tr[k].mx;
pushdown(k);
if(qr<=mid) return query(k<<1,ql,qr);
else if(ql>mid) return query(k<<1|1,ql,qr);
else return max(query(k<<1,ql,mid),query(k<<1|1,mid+1,qr));
}
}
using namespace segtree;
int lson[N],mxdep[N],dep[N],len[N];
int vis[N<<1],head[N],nxt[N<<1],c[N<<1],tot=0;
inline void add(int x,int y,int z){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;c[tot]=z;}
void dfs1(int v){
mxdep[v]=dep[v];
for(int i=head[v];i;i=nxt[i]){
int y=vis[i];
dep[y]=dep[v]+1;
dfs1(y);
mxdep[v]=max(mxdep[v],mxdep[y]);
if(mxdep[v]==mxdep[y]) lson[v]=y;
}
len[v]=mxdep[v]-dep[v];
}
int dfn[N],sign=0;
void dfs2(int v){
dfn[v]=++sign;
if(!lson[v]) return;
dfs2(lson[v]);
for(int i=head[v];i;i=nxt[i]) if(vis[i]!=lson[v]) dfs2(vis[i]);
}
ll ans[N];
int L[N],R[N];
void dp(int v){
if(!lson[v]) return;
dp(lson[v]);
for(int i=head[v];i;i=nxt[i])
if(vis[i]==lson[v]) modify_seq(1,dfn[v]+1,dfn[v]+len[v],c[i]);
ll tmp;
for(int i=head[v];i;i=nxt[i]){
int y=vis[i];
if(y==lson[v]) continue;
dp(y);
for(int j=0;j<=len[y];j++){
tmp=query(1,dfn[y]+j,dfn[y]+j)+c[i];
if(L[v]<=j+1+len[v] && j+1<=R[v])
ans[v]=max(ans[v],tmp+query(1,dfn[v]+max(0,L[v]-j-1),dfn[v]+min(len[v],R[v]-j-1)));
}
for(int j=0;j<=len[y];j++){
tmp=query(1,dfn[y]+j,dfn[y]+j)+c[i];
modify(1,dfn[v]+j+1,tmp);
}
}
if(len[v]>=L[v]) ans[v]=max(ans[v],query(1,dfn[v]+L[v],dfn[v]+min(len[v],R[v])));
}
inline void file(){freopen("watchdog.in","r",stdin);freopen("watchdog.out","w",stdout);}
int main(){
int size=100<<20;//40M
__asm__ ("movq %0,%%rsp\n"::"r"((char*)malloc(size)+size));//提交用这个
int n=read();build(1,1,n);
for(int i=1;i<=n;i++) L[i]=read(),R[i]=read();
for(int x,y,i=2;i<=n;i++){x=read(),y=read();add(x,i,y);}
memset(ans,-1,sizeof(ans));
dep[1]=1;dfs1(1);dfs2(1);dp(1);
int res=0;
for(int i=n,mt=1;i;--i,Mul(mt,23333)) inc(res,mul(mt,(ans[i]%mod+mod)%mod));
cout<<res;
exit(0);
return 0;
}