博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
[GX/GZOI2019]旧词(树上差分+树剖+线段树)
阅读量:4961 次
发布时间:2019-06-12

本文共 2353 字,大约阅读时间需要 7 分钟。

考虑k=1的做法:这是一道原题,我还写过题解,其实挺水的,但当时我菜还是看题解的:。其实就是树上差分后值为1。

考虑k>1的做法:其实可以再次树上差分,给每个点i赋值v[i]=dep[i]k-dep[i-1]k,然后还是和原来一样开一棵线段树,记录一个val[rt]表示当前节点内区间v值的和,以及sum[rt]表示区间值。修改时打标记,只需要将sum[rt]+=v*val[rt],lazy[rt]+=v即可。树剖一下即可。

#include
#define lson l,mid,rt<<1#define rson mid+1,r,rt<<1|1using namespace std;typedef pair
pii;const int N=51000,mod=998244353;int n,Q,k,cnt,dep[N],fa[N],pw[N],sz[N],son[N],top[N],id[N],dfx[N];int ans[N],sum[N<<2],val[N<<2],lazy[N<<2];vector
G[N];vector
vec[N];int qpow(int a,int b){ int ret=1; while(b) { if(b&1)ret=1ll*ret*a%mod; a=1ll*a*a%mod,b>>=1; } return ret;}void dfs(int u){ dep[u]=dep[fa[u]]+1,sz[u]=1; for(int i=0;i
sz[son[u]])son[u]=G[u][i]; }}void dfs2(int u,int tp){ top[u]=tp,id[u]=++cnt,dfx[cnt]=u; if(son[u])dfs2(son[u],tp); for(int i=0;i
>1; build(lson),build(rson); val[rt]=(val[rt<<1]+val[rt<<1|1])%mod;}void pushdown(int rt){ if(!lazy[rt])return; int v=lazy[rt];lazy[rt]=0; lazy[rt<<1]=(lazy[rt<<1]+v)%mod,sum[rt<<1]=(sum[rt<<1]+1ll*v*val[rt<<1])%mod; lazy[rt<<1|1]=(lazy[rt<<1|1]+v)%mod,sum[rt<<1|1]=(sum[rt<<1|1]+1ll*v*val[rt<<1|1])%mod;}void update(int L,int R,int l,int r,int rt){ if(L<=l&&r<=R){lazy[rt]++,sum[rt]=(sum[rt]+val[rt])%mod;return;} pushdown(rt); int mid=l+r>>1; if(L<=mid)update(L,R,lson); if(R>mid)update(L,R,rson); sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%mod;}void Update(int u){ while(top[u]!=1)update(id[top[u]],id[u],1,n,1),u=fa[top[u]]; update(1,id[u],1,n,1);}int query(int L,int R,int l,int r,int rt){ if(L<=l&&r<=R)return sum[rt]; pushdown(rt); int mid=l+r>>1,ret=0; if(L<=mid)ret=(ret+query(L,R,lson))%mod; if(R>mid)ret=(ret+query(L,R,rson))%mod; return ret;}int Query(int u){ int ret=0; while(top[u]!=1)ret=(ret+query(id[top[u]],id[u],1,n,1))%mod,u=fa[top[u]]; ret=(ret+query(1,id[u],1,n,1))%mod; return ret;}int main(){ scanf("%d%d%d",&n,&Q,&k); for(int i=1;i<=n;i++)pw[i]=(qpow(i,k)-qpow(i-1,k)+mod)%mod; for(int i=2;i<=n;i++)scanf("%d",&fa[i]),G[fa[i]].push_back(i); dfs(1),dfs2(1,1); build(1,n,1); for(int i=1,x,y;i<=Q;i++)scanf("%d%d",&x,&y),vec[x].push_back(pii(y,i)); for(int i=1;i<=n;i++) { Update(i); for(int j=0;j
View Code

 

转载于:https://www.cnblogs.com/hfctf0210/p/10859304.html

你可能感兴趣的文章
Python模块调用
查看>>
委托的调用
查看>>
c#中从string数组转换到int数组
查看>>
数据模型(LP32 ILP32 LP64 LLP64 ILP64 )
查看>>
java小技巧
查看>>
POJ 3204 Ikki's Story I - Road Reconstruction
查看>>
【BZOJ】2959: 长跑(lct+缩点)(暂时弃坑)
查看>>
iOS 加载图片选择imageNamed 方法还是 imageWithContentsOfFile?
查看>>
toad for oracle中文显示乱码
查看>>
SQL中Group By的使用
查看>>
错误org/aopalliance/intercept/MethodInterceptor解决方法
查看>>
Pylint在项目中的使用
查看>>
使用nginx做反向代理和负载均衡效果图
查看>>
access remote libvirtd
查看>>
(4) Orchard 开发之 Page 的信息存在哪?
查看>>
ASP.NET中 GridView(网格视图)的使用前台绑定
查看>>
深入了解Oracle ASM(二):ASM File number 1 文件目录
查看>>
Boosting(提升方法)之AdaBoost
查看>>
链接元素<a>
查看>>
Binding object to winForm controller through VS2010 Designer(通过VS2010设计器将对象绑定到winForm控件上)...
查看>>