【BZOJ5507】[GXOI/GZOI2019]旧词(树链剖分,线段树)
【BZOJ5507】[GXOI/GZOI2019]旧词(树链剖分,线段树)
题面
题解
如果\(k=1\)就是链并裸题了。。。
其实\(k>1\)发现还是可以用类似链并的思想,这个东西本质上就是对于当前的一个\(x\),考虑对于其他所有点的贡献,而他们的\(LCA\)一定是\(x\)到根节点链上的一个点。那么对于某个\(x\)的祖先节点,除了\(x\)所在的子树内,其他的所有子树内的点全部会产生这个点的深度的\(k\)次方的贡献。\(k=1\)的时候这个东西可以直接做的原因是因为\(1\)次方的差分可以直接相减。换到\(k>1\)不过是额外维护一下要减去多少个深度\(-1\)就行了。
那么树剖+线段树或者\(LCT\)就可以很容易的解决了。
一开始WA的原因是因为在Modify更新特定节点的时候直接调用了pushup操作,所以此时线段树要开8倍。。。。
#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
#define MOD 998244353
#define MAX 50500
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
int n,Q,K;
struct Line{int v,next;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int fpow(int a,int b){int s=1;while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}return s;}
int hson[MAX],size[MAX],top[MAX],fa[MAX],dep[MAX],dfn[MAX],tim,ln[MAX];
void dfs1(int u,int ff)
{
fa[u]=ff;dep[u]=dep[ff]+1;size[u]=1;
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;dfs1(v,u);size[u]+=size[v];
if(size[hson[u]]<size[v])hson[u]=v;
}
}
void dfs2(int u,int tp)
{
top[u]=tp;dfn[u]=++tim;ln[tim]=u;
if(hson[u])dfs2(hson[u],tp);
for(int i=h[u];i;i=e[i].next)
if(e[i].v!=hson[u])dfs2(e[i].v,e[i].v);
}
#define lson (now<<1)
#define rson (now<<1|1)
struct Node{int s,w1,w2,v1,v2;}t[MAX<<3];
void pushup(int now){t[now].s=(0ll+t[lson].s+t[rson].s+1ll*t[now].w1*t[now].v1+1ll*t[now].w2*t[now].v2)%MOD;}
void Build(int now,int l,int r)
{
if(l==r){t[now].v1=fpow(dep[ln[l]],K);t[now].v2=fpow(dep[ln[l]]-1,K);return;}
int mid=(l+r)>>1;
Build(lson,l,mid);Build(rson,mid+1,r);
t[now].v1=(t[lson].v1+t[rson].v1)%MOD;
t[now].v2=(t[lson].v2+t[rson].v2)%MOD;
}
void Modify(int now,int l,int r,int L,int R,int w1,int w2)
{
if(L<=l&&r<=R){t[now].w1=(t[now].w1+w1)%MOD;t[now].w2=(t[now].w2+w2)%MOD;pushup(now);return;}
int mid=(l+r)>>1;
if(L<=mid)Modify(lson,l,mid,L,R,w1,w2);
if(R>mid)Modify(rson,mid+1,r,L,R,w1,w2);
pushup(now);
}
int Query(int now,int l,int r,int L,int R,int c1,int c2)
{
if(L==l&&r==R)return (t[now].s+1ll*c1*t[now].v1+1ll*c2*t[now].v2)%MOD;
int mid=(l+r)>>1;c1=(c1+t[now].w1)%MOD;c2=(c2+t[now].w2)%MOD;
if(R<=mid)return Query(lson,l,mid,L,R,c1,c2);
if(L>mid)return Query(rson,mid+1,r,L,R,c1,c2);
return (Query(lson,l,mid,L,mid,c1,c2)+Query(rson,mid+1,r,mid+1,R,c1,c2))%MOD;
}
int Y[MAX],ans[MAX];vector<int> Qry[MAX];
int main()
{
n=read();Q=read();K=read();
for(int i=2;i<=n;++i)Add(read(),i);
dfs1(1,0);dfs2(1,1);Build(1,1,n);
for(int i=1,x;i<=Q;++i)x=read(),Y[i]=read(),Qry[x].push_back(i);
for(int i=1;i<=n;++i)
{
int u=i;while(u)Modify(1,1,n,dfn[top[u]],dfn[u],1,(MOD-1)),u=fa[top[u]];
for(int j=0;j<Qry[i].size();++j)
{
int v=Qry[i][j],y=Y[v];
while(y)ans[v]=(ans[v]+Query(1,1,n,dfn[top[y]],dfn[y],0,0))%MOD,y=fa[top[y]];
}
}
for(int i=1;i<=Q;++i)printf("%d\n",ans[i]);
return 0;
}
【BZOJ5507】[GXOI/GZOI2019]旧词(树链剖分,线段树)的更多相关文章
- 【BZOJ-2325】道馆之战 树链剖分 + 线段树
2325: [ZJOI2011]道馆之战 Time Limit: 40 Sec Memory Limit: 256 MBSubmit: 1153 Solved: 421[Submit][Statu ...
- 【BZOJ2243】[SDOI2011]染色 树链剖分+线段树
[BZOJ2243][SDOI2011]染色 Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的 ...
- BZOJ2243 (树链剖分+线段树)
Problem 染色(BZOJ2243) 题目大意 给定一颗树,每个节点上有一种颜色. 要求支持两种操作: 操作1:将a->b上所有点染成一种颜色. 操作2:询问a->b上的颜色段数量. ...
- POJ3237 (树链剖分+线段树)
Problem Tree (POJ3237) 题目大意 给定一颗树,有边权. 要求支持三种操作: 操作一:更改某条边的权值. 操作二:将某条路径上的边权取反. 操作三:询问某条路径上的最大权值. 解题 ...
- bzoj4034 (树链剖分+线段树)
Problem T2 (bzoj4034 HAOI2015) 题目大意 给定一颗树,1为根节点,要求支持三种操作. 操作 1 :把某个节点 x 的点权增加 a . 操作 2 :把某个节点 x 为根的子 ...
- HDU4897 (树链剖分+线段树)
Problem Little Devil I (HDU4897) 题目大意 给定一棵树,每条边的颜色为黑或白,起始时均为白. 支持3种操作: 操作1:将a->b的路径中的所有边的颜色翻转. 操作 ...
- Aizu 2450 Do use segment tree 树链剖分+线段树
Do use segment tree Time Limit: 1 Sec Memory Limit: 256 MB 题目连接 http://www.bnuoj.com/v3/problem_show ...
- 【POJ3237】Tree(树链剖分+线段树)
Description You are given a tree with N nodes. The tree’s nodes are numbered 1 through N and its edg ...
- HDU 2460 Network(双连通+树链剖分+线段树)
HDU 2460 Network 题目链接 题意:给定一个无向图,问每次增加一条边,问个图中还剩多少桥 思路:先双连通缩点,然后形成一棵树,每次增加一条边,相当于询问这两点路径上有多少条边,这个用树链 ...
- bzoj2243[SDOI2011]染色 树链剖分+线段树
2243: [SDOI2011]染色 Time Limit: 20 Sec Memory Limit: 512 MBSubmit: 9012 Solved: 3375[Submit][Status ...
随机推荐
- selenium chrome在新标签页打开链接的方法
目前chrome是我在实现webdriver时运行最稳定的浏览器,如何利用webdriver打开多个标签页和链接呢,到处查找得到的往往只是如何打开标签页.打开标签页很简单,chrome浏览器打开标签页 ...
- 小米6X手机解锁(bl锁)
1. http://www.miui.com/unlock/index.html,申请解锁2. 手机:“设置 -> 更多设置 -> 开发者选项 -> 设备解锁状态”中绑定账号和设备. ...
- QT中pro文件编写的详细说明
如果用QTCreator开发的小伙伴,可能都知道.pro文件,但是里面的具体配置可能比较模糊,今天我就来给大家好好讲解下 一.名称解释 QT += :这个是添加QT需要的模块 TARGET = :生成 ...
- (办公)面试java设计模式
1.单例模式: 程序开发的时候,有些对象只能有一个.有实例,且只有一个,比如工具类. 修改构造方法为私有的. 饿汉模式: 线程安全 创建一个实例 Private Static 实例; 提供一个静态 ...
- 操作系统:修改VirtualBox for Mac的虚拟硬盘大小
我安装的是Mac版的VirtualBox,不能从GUI上修改硬盘大小,但是实在是大小不够用了. 百度后得知,可以用命令行修改. 1.打开终端,输入sudo su,取得管理员权限 $ sudo su P ...
- shell脚本批量ssh登陆主机并执行命令
shell脚本批量ssh登陆主机并执行命令 今天在客户现场遇到了这个问题,客户没有管理工具,无法批量登陆主机下发命令,几个个C段啊,让我一个一个登陆,.................. 所以写了个s ...
- CentOS.7下安装配置FTP和SFTP服务
一: FTP Centos7中默认已经安装了sshd服务(sftp), vsftpd需要手动安装 1.安装并启动FTP服务 1.1 安装vsftpd 使用 yum 安装 vsftpd yum inst ...
- Network Policy - 每天5分钟玩转 Docker 容器技术(171)
Network Policy 是 Kubernetes 的一种资源.Network Policy 通过 Label 选择 Pod,并指定其他 Pod 或外界如何与这些 Pod 通信. 默认情况下,所有 ...
- VSCode的Python扩展下程序运行的几种方式与环境变量管理
在VSCode中编写Python程序时,由于有些地方要使用环境变量,但是发现设置的环境变量有时不起作用,花了点时间研究了一下,过程不表,直接说结论. 首先,环境变量的设置,Python扩展中有三种方式 ...
- sqlserver数据库导出表结构和表数据生成创建表和insert语句
问题描述: 有时候我们只需要导出一张表和表数据到另外一个数据库,如果是备份整个库的话,就会很麻烦那样,没法满足需求. 解决方法: 以sqlserver2014为例:把MGActivity数据库的bat ...