题目描述
如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作:

操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z

操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和

操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z

操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和

输入格式
第一行包含4个正整数N、M、R、P,分别表示树的结点个数、操作个数、根节点序号和取模数(即所有的输出结果均对此取模)。

接下来一行包含N个非负整数,分别依次表示各个节点上初始的数值。

接下来N-1行每行包含两个整数x、y,表示点x和点y之间连有一条边(保证无环且连通)

接下来M行每行包含若干个正整数,每行表示一个操作,格式如下:

操作1: 1 x y z

操作2: 2 x y

操作3: 3 x z

操作4: 4 x

输出格式
输出包含若干行,分别依次表示每个操作2或操作4所得的结果(对P取模)

输入输出样例
输入 #1 复制
5 5 2 24
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
输出 #1 复制
2
21
说明/提示
时空限制:1s,128M

数据规模:

对于30%的数据: N \leq 10, M \leq 10 N≤10,M≤10

对于70%的数据: N \leq {10}^3, M \leq {10}^3 N≤10
3
,M≤10
3

对于100%的数据: N \leq {10}^5, M \leq {10}^5 N≤10
5
,M≤10
5

( 其实,纯随机生成的树LCA+暴力是能过的,可是,你觉得可能是纯随机的么233 )

样例说明:

树的结构如下:

各个操作如下:

故输出应依次为2、21(重要的事情说三遍:记得取模)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <iomanip>
#define ALL(x) (x).begin(), (x).end()
#define sz(a) int(a.size())
#define all(a) a.begin(), a.end()
#define rep(i,x,n) for(int i=x;i<n;i++)
#define repd(i,x,n) for(int i=x;i<=n;i++)
#define pii pair<int,int>
#define pll pair<long long ,long long>
#define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define MS0(X) memset((X), 0, sizeof((X)))
#define MSC0(X) memset((X), '\0', sizeof((X)))
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define eps 1e-6
#define gg(x) getInt(&x)
#define chu(x) cout<<"["<<#x<<" "<<(x)<<"]"<<endl
using namespace std;
typedef long long ll;
ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;}
ll lcm(ll a, ll b) {return a / gcd(a, b) * b;}
ll powmod(ll a, ll b, ll MOD) {ll ans = 1; while (b) {if (b % 2)ans = ans * a % MOD; a = a * a % MOD; b /= 2;} return ans;}
inline void getInt(int* p);
const int maxn = 100010;
const int inf = 0x3f3f3f3f;
/*** TEMPLATE CODE * * STARTS HERE ***/
int n, m;
ll p;
int root;
ll a[maxn];// 初始点权
ll wt[maxn];// 新建编号点权。
int cnt;// 编号用的变量
int top[maxn];// 所在重链的顶点编号
int id[maxn];//节点的新编号。
std::vector<int> son[maxn];
int SZ[maxn];// 子数大小
int wson[maxn];// 重儿子
int fa[maxn];// 父节点
int dep[maxn];// 节点的深度void dfs1(int id, int pre, int step) // 维护出sz,wson,fa,dep
{dep[id] = step;fa[id] = pre;SZ[id] = 1;int  maxson = -1;for (auto x : son[id]){if (x != pre){dfs1(x, id, step + 1);SZ[id] += SZ[x];if (SZ[x] > maxson){maxson = SZ[x];wson[id] = x;}}}}//处理出top[],wt[],id[]
void dfs2(int u,int topf)
{id[u] = ++cnt;wt[cnt]=a[u];top[u]=topf;if(!wson[u]) // 没儿子时直接结束{return ;}dfs2(wson[u],topf);// 先处理重儿子for(auto x:son[u]){if(x==wson[u]||x==fa[u])//只处理轻儿子{continue;}dfs2(x,x);// 每个轻儿子以自己为top}
}struct node
{int l,r;ll sum;ll laze;
}segment_tree[maxn<<2];void pushup(int rt)
{segment_tree[rt].sum=(segment_tree[rt<<1].sum+segment_tree[rt<<1|1].sum)%p;
}
void build(int rt,int l,int r)
{segment_tree[rt].l=l;segment_tree[rt].r=r;segment_tree[rt].laze=0;if(l==r){segment_tree[rt].sum=wt[l];segment_tree[rt].sum%=p;return;}int mid=(l+r)>>1;build(rt<<1,l,mid);build(rt<<1|1,mid+1,r);pushup(rt);
}void pushdown(int rt)
{if(segment_tree[rt].laze){ll val=segment_tree[rt].laze;segment_tree[rt].laze=0ll;segment_tree[rt<<1].sum+=(val)*(segment_tree[rt<<1].r-segment_tree[rt<<1].l+1)%p;segment_tree[rt<<1|1].sum+=(val)*(segment_tree[rt<<1|1].r-segment_tree[rt<<1|1].l+1)%p;segment_tree[rt<<1].laze+=val;segment_tree[rt<<1|1].laze+=val;segment_tree[rt<<1].sum%=p;segment_tree[rt<<1|1].sum%=p;segment_tree[rt<<1].laze%=p;segment_tree[rt<<1|1].laze%=p;}
}
void update(int rt,int l,int r,int val)
{val%=p;if(segment_tree[rt].l>=l&&segment_tree[rt].r<=r){int wei=segment_tree[rt].r-segment_tree[rt].l+1;segment_tree[rt].sum+=(wei)*val%p;segment_tree[rt].sum%=p;segment_tree[rt].laze+=val;segment_tree[rt].laze%=p;return ;}pushdown(rt);int mid=(segment_tree[rt].l+segment_tree[rt].r)>>1;if(mid>=l){update(rt<<1,l,r,val);}if(mid<r){update(rt<<1|1,l,r,val);}pushup(rt);
}
ll query(int rt,int l,int r)
{if(segment_tree[rt].l>=l&&segment_tree[rt].r<=r){ll res=0ll;res+=segment_tree[rt].sum;res%=p;return res;}pushdown(rt);int mid=(segment_tree[rt].l+segment_tree[rt].r)>>1;ll res=0ll;if(mid>=l){res+=query(rt<<1,l,r);res%=p;}if(mid<r){res+=query(rt<<1|1,l,r);res%=p;}return res;}void uprange(int x,int y,int k)
{k%=p;while(top[x]!=top[y]){if(dep[top[x]]<dep[top[y]])// 使x的top深度较大{swap(x,y);}update(1,id[top[x]],id[x],k);// 处理x到top[x] 那段链x=fa[top[x]];// x向上爬到top[x]的父节点}if(dep[x]>dep[y])使x的深度较大swap(x,y);update(1,id[x],id[y],k);//更新x到y这段链,根据上面的处理,他们一定是在同一条链上
}ll qrange(int x,int y)
{ll ans=0ll;while(top[x]!=top[y]){if(dep[top[x]]<dep[top[y]]){swap(x,y);}ans+=query(1,id[top[x]],id[x]);ans%=p;x=fa[top[x]];}if(dep[x]>dep[y])swap(x,y);ans+=query(1,id[x],id[y]);ans%=p;return ans;
}
void upson(int x,ll val)
{val%=p;update(1,id[x],id[x]+SZ[x]-1,val);//子树区间右端点为id[x]+siz[x]-1
}
ll qson(int x)
{ll res=0ll;res+=query(1,id[x],id[x]+SZ[x]-1);res%=p;return res;
}
int main()
{
//    freopen("D:\\common_text\\code_stream\\in.txt","r",stdin);
//    freopen("D:\\common_text\\code_stream\\out.txt","w",stdout);gbtb;cin >> n >> m >> root >> p;repd(i, 1, n){cin >> a[i];}int u, v;repd(i, 2, n){cin >> u >> v;son[u].pb(v);son[v].pb(u);}dfs1(root,0,1);dfs2(root,root);build(1,1,n);int op,x,y,z;
//  操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z
//
//  操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和
//
//  操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z
//
//  操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和while(m--){cin>>op;if(op==1){cin>>x>>y>>z;uprange(x,y,z);}else if(op==2){cin>>x>>y;cout<<qrange(x,y)<<endl;}else if(op==3){cin>>x>>z;upson(x,z);}else if(op==4){cin>>x;cout<<qson(x)<<endl;}}return 0;
}inline void getInt(int* p) {char ch;do {ch = getchar();} while (ch == ' ' || ch == '\n');if (ch == '-') {*p = -(getchar() - '0');while ((ch = getchar()) >= '0' && ch <= '9') {*p = *p * 10 - ch + '0';}}else {*p = ch - '0';while ((ch = getchar()) >= '0' && ch <= '9') {*p = *p * 10 + ch - '0';}}
}

转载于:https://www.cnblogs.com/qieqiemin/p/11297835.html

洛谷 P3384 【模板】树链剖分相关推荐

  1. 洛谷3384(树链剖分模板题)

    题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式 ...

  2. 【模板】树链剖分 P3384

    题目链接 //部分转自:https://www.luogu.org/problemnew/solution/P3384 初学树链剖分,感觉这个模板题还是容易理解的,但是实在是码量很大的. 知识点: 重 ...

  3. 洛谷 P3384 【模板】树链剖分-树链剖分(点权)(路径节点更新、路径求和、子树节点更新、子树求和)模板-备注结合一下以前写的题目,懒得写很详细的注释...

    P3384 [模板]树链剖分 题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节 ...

  4. 洛谷P3384 - 树链剖分(树链剖分模板题)

    题目链接 https://www.luogu.org/problemnew/show/P3384 [描述] 树链剖分模板题,记一下板子 #include<bits/stdc++.h> #d ...

  5. 专题·树链剖分【including 洛谷·【模板】树链剖分

    初见安~~~终于学会了树剖~~~ [兴奋]当初机房的大佬在学树剖的时候我反复强调过:"学树剖没有前途的!!!" 恩.真香. 一.重链与重儿子 所谓树剖--树链剖分,就是赋予一个链的 ...

  6. 洛谷3384:【模板】树链剖分——题解

    https://www.luogu.org/problemnew/show/P3384 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 ...

  7. ⌈洛谷1505⌋⌈BZOJ2157⌋⌈国家集训队⌋旅游【树链剖分】

    题目链接 [洛谷] [BZOJ] 题目描述 Ray 乐忠于旅游,这次他来到了T 城.T 城是一个水上城市,一共有 N 个景点,有些景点之间会用一座桥连接.为了方便游客到达每个景点但又为了节约成本,T ...

  8. 树链剖分代码(洛谷3384)

    题目链接 题意:树链剖分模板题 (树链剖分+线段树) 准备工作: 第一步: 定义声明 // 链式前向星存图 (**** 记得开2倍空间 ****) struct EDGE {int to;int ne ...

  9. 【树链剖分】洛谷树(P3401)

    正题 P3401 题目大意 给你一棵树,让你进行以下操作 修改一条边的边权 查询一条路径的所有子路径异或值的和 解题思路 记下所有点到根节点的路径亦或值,那么查询就是所有点对的异或值之和 因为边权&l ...

  10. 洛谷P2486 [SDOI2011]染色(树链剖分+线段树判断边界)

    [题目链接] [思路]: 涉及到树上区间修改操作,所以使用树链剖分,涉及到区间查询,所以使用线段树. update操作时,就正常操作,难点在于query操作的计数. 因为树链剖分的dfs序只能保证一条 ...

最新文章

  1. 机器学习博士在获得学位之前需要掌握的九种工具!
  2. 2019年美国国家人工智能战略报告(中文翻译版)
  3. 四、唱歌不如跳舞(上)
  4. php的filter input,记一个php://filter和php://input的CTF题
  5. Design Pattern: Singleton 模式
  6. 从HelloWorld看iphone程序的生命周期
  7. href脱离iframe显示
  8. 服务器精益改善系列,精益生产改善的内容是什么?
  9. LeetCode LCP 29. 乐团站位(数学 等差数列)
  10. 黑客攻防技术宝典Web实战篇第2版—第11章 攻击应用程序逻辑
  11. js for in遍历对象_JS中轻松遍历对象属性的几种方式
  12. 十二款世界顶级杀毒软件下载,有序列号
  13. php做APP接口开发,接口的安全性
  14. 现在Web前端培训,哪个机构比较好?
  15. 截获HTTP/HTTPS请求实现头脑王者辅助
  16. 导数的零点定理与达布定理
  17. 计算机专业术语,收藏用
  18. 猜数字小游戏html,猜数字游戏.html
  19. 凸集 凸函数 判定凸函数
  20. 利用手机作为渗透工具的一些思路

热门文章

  1. SAP CRM one order框架到了S/4HANA后是如何重构的
  2. SAP CRM Opportunity订单的文档流Document Flow的一些变体variant
  3. SAP C4C Opportunity和SAP ERP Sales流程的集成
  4. 使用SAP云平台 + JNDI访问Internet Service
  5. c++ websocket客户端_python测试开发django81.dwebsocket实现websocket
  6. python选择题题库for、if_Python题目1:猜年龄(for、if else和where)
  7. c语言用链表编写简单程序,C语言单链表简单实现(简单程序复杂化)
  8. 传热学环肋肋效率matlab程序,传热学 第二章第四节 通过肋片的导热.pdf
  9. 数据包如何正确的发到目的地_江苏工业交换机厂家浅谈交换机如何应用
  10. java for 循环执行顺序