洛谷 P3384树链剖分 题解
挺好的一道树剖模板;
首先要学会最模板的树剖;
然后这道题要注意几个细节:
初始化时,seg[0]=1,seg[root]=1,top[root]=root,rev[1]=root;
在线段树上进行操作时,要使用lazy标记;
对于一个以x为根的子树,它子树中所有的元素一定时在线段树上连续的区间,且以seg[x]开始,以seg[x]+size[x]-1结束;
然后写码的时候注意不要手残(比如说预处理时写成了dep[u]=dep[u]+1);
#include <bits/stdc++.h>
using namespace std;
int n,m,r,p;
int head[2000010],cnt;
class littlestar{
public:
int to;
int nxt;
void add(int u,int v){
to=v;
nxt=head[u];
head[u]=cnt;
}
}star[2000010];
int a[100010];
int f[100010],dep[100010],son[100010],seg[100010],rev[100010],size[100010],top[100010];
void dfs1(int u,int fa)
{
size[u]=1;
f[u]=fa;
dep[u]=dep[fa]+1;
for(int i=head[u];i;i=star[i].nxt){
int v=star[i].to;
if(v==fa) continue;
dfs1(v,u);
size[u]+=size[v];
if(size[v]>size[son[u]]) son[u]=v;
}
}
void dfs2(int u,int fa)
{
if(son[u]){
seg[son[u]]=++seg[0];
rev[seg[0]]=son[u];
top[son[u]]=top[u];
dfs2(son[u],u);
}
for(int i=head[u];i;i=star[i].nxt){
int v=star[i].to;
if(v==fa) continue;
if(!top[v]){
seg[v]=++seg[0];
rev[seg[0]]=v;
top[v]=v;
dfs2(v,u);
}
}
}
struct ss{
int sum;
int lazy;
}tree[1000010];
void build(int k,int l,int r)
{
if(l==r){
tree[k].sum=a[rev[l]]%p;
return;
}
int mid=(l+r)/2;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}
void pre()
{
dfs1(r,0);
seg[0]=seg[r]=1;
top[r]=r;
rev[1]=r;
dfs2(r,0);
build(1,1,seg[0]);
}
void pushdown(int k,int l,int r)
{
int mid=(l+r)/2;
tree[k<<1].lazy=(tree[k<<1].lazy+tree[k].lazy)%p;
tree[k<<1].sum=(tree[k<<1].sum+tree[k].lazy*(mid-l+1))%p;
tree[k<<1|1].lazy=(tree[k<<1|1].lazy+tree[k].lazy)%p;
tree[k<<1|1].sum=(tree[k<<1|1].sum+tree[k].lazy*(r-mid))%p;
tree[k].lazy=0;
}
int query(int k,int l,int r,int x,int y)
{
if(r<x||l>y){
return 0;
}
if(l>=x&&r<=y){
return tree[k].sum%p;
}
int mid=(l+r)/2;
pushdown(k,l,r);
return (query(k<<1,l,mid,x,y)+query(k<<1|1,mid+1,r,x,y))%p;
}
void change(int k,int l,int r,int x,int y,int goal)
{
if(r<x||l>y) return;
if(l>=x&&r<=y){
tree[k].sum=(tree[k].sum+(r-l+1)*goal)%p;
tree[k].lazy=(tree[k].lazy+goal)%p;
return;
}
pushdown(k,l,r);
int mid=(l+r)/2;
change(k<<1,l,mid,x,y,goal);
change(k<<1|1,mid+1,r,x,y,goal);
tree[k].sum=(tree[k<<1].sum+tree[k<<1|1].sum)%p;
}
void changeroad(int x,int y,int z)
{
int fx=top[x],fy=top[y];
while(fx!=fy){
if(dep[fx]<dep[fy]) swap(fx,fy),swap(x,y);
change(1,1,seg[0],seg[fx],seg[x],z);
x=f[fx];
fx=top[x];
}
if(dep[x]>dep[y]) swap(x,y);
change(1,1,seg[0],seg[x],seg[y],z);
}
int queryroad(int x,int y)
{
long long ans=0;
int fx=top[x],fy=top[y];
while(fx!=fy){
if(dep[fx]<dep[fy]) swap(fx,fy),swap(x,y);
ans=(ans+query(1,1,seg[0],seg[fx],seg[x]))%p;
x=f[fx];
fx=top[x];
}
if(dep[y]<dep[x]) swap(x,y);
ans=(ans+query(1,1,seg[0],seg[x],seg[y]))%p;
return ans%p;
}
void changetree(int x,int goal)
{
change(1,1,seg[0],seg[x],seg[x]+size[x]-1,goal);
return;
}
long long querytree(int x)
{
long long res=0;
res=(res+query(1,1,seg[0],seg[x],seg[x]+size[x]-1))%p;
return res;
}
int main(){
cin>>n>>m>>r>>p;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
for(int i=1;i<=n-1;i++){
int u,v;
scanf("%d%d",&u,&v);
star[++cnt].add(u,v);
star[++cnt].add(v,u);
}
pre();
for(int i=1;i<=m;i++){
int type;
scanf("%d",&type);
if(type==1){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
changeroad(x,y,z);
}
else if(type==2){
int x,y;
scanf("%d%d",&x,&y);
cout<<queryroad(x,y)%p<<endl;
}
else if(type==3){
int x,z;
scanf("%d%d",&x,&z);
changetree(x,z);
}
else{
int x;
scanf("%d",&x);
cout<<querytree(x)%p<<endl;
}
}
}
/*
5 5 2 30000
7 3 7 8 0
1 2
1 5
3 1
4 1
3 4 2
3 2 2
4 5
1 5 1 3
2 1 3
*/
洛谷 P3384树链剖分 题解的更多相关文章
- 洛谷P3384 树链剖分
如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式: 2 x ...
- 洛谷 P3384 树链剖分(模板题)
题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式 ...
- 洛谷 [P3384] 树链剖分 模版
支持各种数据结构上树,注意取膜. #include <iostream> #include <cstring> #include <algorithm> #incl ...
- 【算法学习】【洛谷】树链剖分 & P3384 【模板】树链剖分 P2146 软件包管理器
刚学的好玩算法,AC2题,非常开心. 其实很早就有教过,以前以为很难就没有学,现在发现其实很简单也很有用. 更重要的是我很好调试,两题都是几乎一遍过的. 介绍树链剖分前,先确保已经学会以下基本技巧: ...
- 【树链剖分】洛谷P3379 树链剖分求LCA
题目描述 如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先. 输入输出格式 输入格式: 第一行包含三个正整数N.M.S,分别表示树的结点个数.询问的个数和树根结点的序号. 接下来N-1行每 ...
- 洛谷P2146 树链剖分
题意 思路:直接树链剖分,用线段树维护即可,算是树剖的经典题目吧. 代码: #include <bits/stdc++.h> #define ls(x) (x << 1) #d ...
- 洛谷p3384【模板】树链剖分题解
洛谷p3384 [模板]树链剖分错误记录 首先感谢\(lfd\)在课上调了出来\(Orz\) \(1\).以后少写全局变量 \(2\).线段树递归的时候最好把左右区间一起传 \(3\).写\(dfs\ ...
- P3384 【模板】树链剖分 题解&&树链剖分详解
题外话: 一道至今为止做题时间最长的题: begin at 8.30A.M 然后求助_yjk dalao后 最后一次搞取模: awsl. 正解开始: 题目链接. 树链剖分,指的是将一棵树通过两次遍历后 ...
- 【树链剖分】洛谷P3384树剖模板
题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式 ...
随机推荐
- 【原创】LUOGU P1808 单词分类
STL大法好!!! 使用sort()将string排序,map去重并统计即可. 最短代码如下: #include<bits/stdc++.h> using namespace std; s ...
- java浮点数精度问题解决方法
基础知识回顾: BigDecimal.setScale()方法用于格式化小数点setScale(1)表示保留一位小数,默认用四舍五入方式 setScale(1,BigDecimal.ROUND_DOW ...
- SRS之RTMP handshake
1. SrsRtmpServer::handshake 位于 srs_rtmp_stack.cpp. int SrsRtmpServer::handshake() { int ret = ERROR_ ...
- phpmyadmin 导入sql报错(sql为phpstudy内置数据库导出来)
解决方法 1.打开sql,把头部注释去掉
- koa 基础(十四)cookie 的基本使用
1.app.js /** * cookie的简介: * 1.cookie保存在浏览器客户端 * 2.可以让我们用同一个浏览器访问同一个域名的时候共享数据 * * cookie的作用: * 1.保存用户 ...
- html添加注释怎么弄?
HTML 注释: <!--这是一段注释.注释不会在浏览器中显示.--> 这是一段普通的段落. 快捷键: 我用的是 Notpad++ 添加行注释 Ctrl+K 取消行注释 Ctrl+Shif ...
- Win10删除文件显示删除确认对话框
1.右键单击“回收站”图标:2.在弹出属性窗口中,点击“属性”选项:3.在“回收站”窗口中,在选项“显示删除确认对话框”前面打钩,并单击“确定”按钮:
- 搭建Git服务器及本机克隆提交
前文 Git是什么? Git是目前世界上最先进的分布式版本控制系统. SVN与Git的最主要的区别? SVN是集中式版本控制系统,版本库是集中放在中央服务器的,而干活的时候,用的都是自己的电脑,所以首 ...
- spring 过滤器- 过滤登陆请求路径(过滤静态资源跳转到登陆页面)
public class LoginedFilter implements Filter { /** * 排除的地址 */ private Map<String, Boolean> ign ...
- kvm的使用(2)
一.远程管理kvm虚机 (2)有些情况下,有一个要配置的地方. 因为 KVM(准确说是 Libvirt)默认不接受远程管理,需要按下面的内容配置被管理宿主机中的两个文件: vim /etc/defau ...