[校内模拟][长链剖分][线段树]看门人

题意:求一棵树上每个点的子树中经过这个点的所有路径中边数在给定范围内的最长路径

早上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;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值