[GXOI/GZOI2019]旧词(树上差分+树剖)

时间:2023-01-02 15:19:14

前置芝士:[LNOI2014]LCA

要是这题放HNOI就好了

原题:\(\sum_{l≤i≤r}dep[LCA(i,z)]\)

这题:\(\sum_{i≤r}dep[LCA(i,z)]^k\)

对于原题,我们需要把每个询问拆成1~l-1 & 1r再进行差分(所以这题帮我们省去了一个步骤~)

先考虑\(k=1\)原题

我们先转化题意

\(dep[lca]\)$\ \(==\)\ $$dis[1][lca]+1$$\ \(==\)\ $$lca->1$的点数

所以我们每一个点(x)对答案的贡献(\(dep[lca(x, z)]\)),就是他们到根节点的公共路径的点数

于是,对于每一个点,我们只需要把1->x的链上加1即可

对于每一个询问,我们只需要求出1->z的链上的和即可

这一点我们可以利用树剖\(/LCT\)解决

但是直接做是\(O(N^2*\)树剖\(\LCT)\)的,我们考虑莫队

这样复杂度变成了\(O(N\sqrt{N}*\)树剖\(\LCT)\)

什么?你觉得这个算法还不够优秀?所以我们来考虑优化莫队

莫队的\(\sqrt{N}\)是怎么来的?不停的移动左右端点

但是这道题的左端点是固定的\((1)\),所以只需要移动右端点即可,而右端点不需要动来动去,只需要往后扫一遍即可,复杂度是\(O(N*\)树剖\(\LCT)\)

代码的话可以参考[LNOI2014]LCA

考虑k!=1

我们为什么k=1的时候对于每个点是\(1->x\)路径上+1?

这个1的本质是树上差分,即:\((dep[x]+1)^1-dep[x]^1 = 1\)

所以我们只需要把1改成k即可

所以现在问题变成了:给定一个序列,每一个点有两个权值\((a, b)\),每一个点的点权为\(a*b\),支持a权值区间加1和区间查询

因为b不会改变,所以我们考虑线段树

把线段树的每一个节点新弄一个权值,为\(\sum_{l≤i≤r} b\),每次更新区间的时候用这个权值*sum即可

#include<bits/stdc++.h>
using namespace std;
#define il inline
#define re register
#define debug printf("Now is Line : %d\n",__LINE__)
#define file(a) freopen(#a".in","r",stdin);freopen(#a".out","w",stdout)
#define int long long
#define inf 123456789
#define mod 998244353
il int read() {
re int x = 0, f = 1; re char c = getchar();
while(c < '0' || c > '9') { if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') x = x * 10 + c - 48, c = getchar();
return x * f;
}
#define rep(i, s, t) for(re int i = s; i <= t; ++ i)
#define Next(i, u) for(re int i = head[u]; i; i = e[i].next)
#define mem(k, p) memset(k, p, sizeof(k))
#define ls k * 2
#define rs k * 2 + 1
#define _ 500005
struct edge {int v, next;}e[_];
struct ques {
int u, z, id;
il bool operator < (const ques x) const {return u < x.u;}
}q[_];
int n, m, k, val[_ << 2], sum[_ << 2], mi[_], head[_], cnt, ans[_], tag[_ << 2], now;
int fa[_], dep[_], size[_], son[_], top[_], seg[_], col, rev[_];
il void add(int u, int v) {
e[++ cnt] = (edge) {v, head[u]}, head[u] = cnt;
}
il int qpow(int a, int b) {
int r = 1;
while(b) {
if(b & 1) r = 1ll * r * a % mod;
b >>= 1, a = 1ll * a * a % mod;
}
return r;
}
il void dfs1(int u) {
dep[u] = dep[fa[u]] + 1, size[u] = 1;
Next(i, u) {
if(e[i].v == fa[u]) continue;
dfs1(e[i].v), size[u] += size[e[i].v];
if(size[son[u]] < size[e[i].v]) son[u] = e[i].v;
}
}
il void dfs2(int u, int fr) {
top[u] = fr, seg[u] = ++ col, rev[col] = u;
if(son[u]) dfs2(son[u], fr);
Next(i, u) if(e[i].v != son[u] && e[i].v != fa[u]) dfs2(e[i].v, e[i].v);
}
il void build(int k, int l, int r) {
if(l == r) return (void)(val[k] = (mi[dep[rev[l]]] - mi[dep[rev[l]] - 1] + mod) % mod);
int mid = (l + r) >> 1;
build(ls, l, mid), build(rs, mid + 1, r), val[k] = (val[ls] + val[rs]) % mod;
}
il void pushdown(int k) {
if(!tag[k]) return;
sum[ls] = (sum[ls] + ((tag[k] * val[ls]) % mod)) % mod;
sum[rs] = (sum[rs] + ((tag[k] * val[rs]) % mod)) % mod;
tag[ls] += tag[k], tag[rs] += tag[k], tag[k] = 0;
}
il void change(int k, int l, int r, int ll, int rr) {
if(l > rr || ll > r) return;
if(l >= ll && r <= rr) {sum[k] = (sum[k] + val[k]) % mod, ++ tag[k]; return;}
int mid = (l + r) >> 1;
pushdown(k), change(ls, l, mid, ll, rr), change(rs, mid + 1, r, ll, rr);
sum[k] = (sum[ls] + sum[rs]) % mod;
}
il int query(int k, int l, int r, int ll, int rr) {
if(l > rr || ll > r) return 0;
if(l >= ll && r <= rr) return sum[k];
int mid = (l + r) >> 1;
pushdown(k);
return (query(ls, l, mid, ll, rr) + query(rs, mid + 1, r, ll, rr)) % mod;
}
il int query(int u) {
int ans = 0;
while(top[u]) ans = (ans + query(1, 1, n, seg[top[u]], seg[u])) % mod, u = fa[top[u]];
return ans;
}
il void change(int u) {
while(top[u]) change(1, 1, n, seg[top[u]], seg[u]), u = fa[top[u]];
}
signed main() {
n = read(), m = read(), k = read() % (mod - 1), now = 1;
rep(i, 1, n) mi[i] = qpow(i, k);
rep(i, 2, n) fa[i] = read(), add(fa[i], i);
rep(i, 1, m) q[i].id = i, q[i].u = read(), q[i].z = read();
sort(q + 1, q + m + 1), dfs1(1), dfs2(1, 1), build(1, 1, n);
rep(i, 1, n) {
change(i);
while(i == q[now].u) ans[q[now].id] = query(q[now].z), ++ now;
}
rep(i, 1, m) printf("%lld\n", ans[i]);
return 0;
}

[GXOI/GZOI2019]旧词(树上差分+树剖)的更多相关文章

  1. 【BZOJ5507】&lbrack;GXOI&sol;GZOI2019&rsqb;旧词(树链剖分,线段树)

    [BZOJ5507][GXOI/GZOI2019]旧词(树链剖分,线段树) 题面 BZOJ 洛谷 题解 如果\(k=1\)就是链并裸题了... 其实\(k>1\)发现还是可以用类似链并的思想,这 ...

  2. &lbrack;LOJ3088&rsqb;&lbrack;GXOI&sol;GZOI2019&rsqb;旧词——树链剖分&plus;线段树

    题目链接: [GXOI/GZOI2019]旧词 对于$k=1$的情况,可以参见[LNOI2014]LCA,将询问离线然后从$1$号点开始对这个点到根的路径链修改,每次询问就是对询问点到根路径链查询即可 ...

  3. P5305 &lbrack;GXOI&sol;GZOI2019&rsqb;旧词

    题目地址:P5305 [GXOI/GZOI2019]旧词 这里是官方题解 \[\sum_{i \leq x}^{}\ depth(lca(i,y))^k\] \(k = 1\) 求的是 \(\sum_ ...

  4. &lbrack;GX&sol;GZOI2019&rsqb;旧词(树上差分&plus;树剖&plus;线段树)

    考虑k=1的做法:这是一道原题,我还写过题解,其实挺水的,但当时我菜还是看题解的:https://www.cnblogs.com/hfctf0210/p/10187947.html.其实就是树上差分后 ...

  5. BZOJ5507 GXOI&sol;GZOI2019旧词 (树链剖分&plus;线段树)

    https://www.cnblogs.com/Gloid/p/9412357.html差分一下是一样的问题.感觉几年没写过树剖了. #include<iostream> #include ...

  6. luogu P5305 &lbrack;GXOI&sol;GZOI2019&rsqb;旧词

    传送门 先考虑\(k=1\),一个点的深度就是到根节点的路径上的点的个数,所以\(lca(x,y)\)的深度就是\(x\)和\(y\)到根路径的交集路径上的点的个数,那么对于一个询问,我们可以对每个点 ...

  7. 洛谷P3258 &lbrack;JLOI2014&rsqb;松鼠的新家(树上差分&plus;树剖)

    题目描述 松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的.天哪,他居然真的住在”树“上. 松鼠想邀请小熊维尼前 ...

  8. &lbrack;GXOI&sol;GZOI2019&rsqb;旧词

    很像LNOI 2014 LCA那道题. 同样的套路,离线以后直接扫描线. k=1的话就是原题. 考虑一般情况. 原本的做法是对x到根的这条链做一下区间+1操作,目的是为了是的在深度为i的位置得到的贡献 ...

  9. BZOJ3881 Coci2015Divljak(AC自动机&plus;树上差分&plus;树状数组)

    建出AC自动机及其fail树,每次给新加入的串在AC自动机上经过的点染色,问题即转化为子树颜色数.显然可以用dfs序转成序列问题树状数组套权值线段树解决,显然过不掉.事实上直接树上差分,按dfs序排序 ...

随机推荐

  1. winform记事本&lpar;基本功能&rpar;

    本题主要考察各种控件的应用 using System; using System.Collections.Generic; using System.ComponentModel; using Sys ...

  2. Ajax基础知识《一》

    对于网站开发人员,一定不会陌生的Ajax技术,本篇就让我们认识一下它,或许在日后的开发过程中我们就可以使用到.Ajax在那方面使用的比较多呢?答案:表单注册,传统的表单注册,有时需要填写大量的信息,当 ...

  3. 公共增删改查&lpar;MVC&plus;三层架构&rpar;

    1.建立数据访问层:新建一个项目,选择类库,命名为XXXDAL,然后把新生成的类删除,重新建一个类BaseDal,代码如下: public class BaseDal<T> where T ...

  4. swift学习笔记之-下标脚本

    //下标脚本subscript import UIKit /*下标脚本(Subscripts) 下标脚本: 1.可以定义在类(Class).结构体(structure)和枚举(enumeration) ...

  5. VisualSVN SERVER的安装和使用

    SVN Server安装 Subversion是优秀的版本控制工具,其具体的的优点和详细介绍,这里就不再多说.下载的网址是:http://subversion.apache.org/packages. ...

  6. MySQL【第二篇】基本命令

    一.连接MySQL 登录 mysql 有两种方式: 远程主机:mysql -h主机地址 -u用户名 -p密码 -P端口号 本机:mysql -h主机地址 -u用户名 -p密码 -P端口号 如果端口号是 ...

  7. java编程排错技巧

    一.Eclipse提示错误The type java.lang.CharSequence cannot be resolved. It is indirectly referenced from re ...

  8. Spring中的&commat;Transactional&lpar;rollbackFor &equals; Exception&period;class&rpar;属性详解

    序言 今天我在写代码的时候,看到了.一个注解@Transactional(rollbackFor = Exception.class),今天就和大家分享一下,这个注解的用法: 异常 如下图所示,我们都 ...

  9. 解题:LNOI 2014 LCA

    题面 这题有点意思 转化问题,我们把询问区间的点到根链加,再查询询问点到根的权值和就是每个询问的答案. 然后如果你数据结构没学傻只需要差分一下就可以扫一遍出解了 #include<cstdio& ...

  10. eclipse - 新建jsp页面默认模板设置

    有时候我们自己如果没有现成的JSP模板时,系统一般会自动生成如下页面: 这个页面显然并不是我们所需要的,所以我们需要修改默认模板 进入 修改 <%@ page language="ja ...